mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 07:18:27 +01:00
Merge pull request #222 from efiop/master
dvc: get rid of 'test' command
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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)
|
||||
Reference in New Issue
Block a user