Merge pull request #222 from efiop/master

dvc: get rid of 'test' command
This commit is contained in:
Ruslan Kuprieiev
2017-09-17 23:30:03 +03:00
committed by GitHub
2 changed files with 0 additions and 49 deletions
-1
View File
@@ -15,7 +15,6 @@ from dvc.command.show_workflow import CmdShowWorkflow
from dvc.command.instance_create import CmdInstanceCreate
from dvc.command.config import CmdConfig
from dvc.command.show_pipeline import CmdShowPipeline
from dvc.command.test import CmdTest
from dvc.config import Config
from dvc import VERSION
-48
View File
@@ -1,48 +0,0 @@
import os
import sys
from google.cloud import storage
import dvc
from dvc.command.base import CmdBase
from dvc.logger import Logger
from dvc.config import Config
class CmdTest(CmdBase):
def __init__(self, settings):
super(CmdTest, self).__init__(settings)
def define_args(self, parser):
self.set_no_git_actions(parser)
def run(self):
cloud = self.settings._config.cloud
assert cloud in ['amazon', 'google'], 'unknown cloud: %s' % cloud
if cloud == 'amazon':
print('TODO')
sys.exit(-1)
elif cloud == 'google':
Logger.info('testing gcloud')
if self.settings._config.gc_project_name == '':
Logger.error('Please specify the google cloud project name in dvc.conf')
raise dvc.settings.SettingsError('must specify GC ProjectName')
try:
client = storage.Client(project=self.settings._config.gc_project_name)
except:
Logger.error('initializing storage.Client raised')
raise
sb = self.config.storage_bucket
if sb == '':
Logger.error('Please specificy a bucket for google cloud in dvc.conf')
sys.exit(-1)
bucket = client.bucket(sb)
if not bucket.exists():
Logger.error('google cloud storage: bucket %s doesn\'t exist' % sb)
sys.exit(-1)
Logger.info('bucket %s visible: google cloud seems to be configured correctly' % sb)
sys.exit(0)