mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 07:18:27 +01:00
Ruslan Kuprieiev
e853fc76aa
From now on everything is stored in filename + .dvc path, except for local_state. This allows us to be more explicit and help users see everything they need in relation to version control of their own data files. Signed-off-by: Ruslan Kuprieiev <[email protected]>
113 lines
3.1 KiB
Python
113 lines
3.1 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
from dvc.command.common.base import CmdBase
|
|
from dvc.logger import Logger
|
|
from dvc.config import Config
|
|
from dvc.exceptions import DvcException
|
|
from dvc.path.data_item import DataItem
|
|
|
|
|
|
class InitError(DvcException):
|
|
def __init__(self, msg):
|
|
DvcException.__init__(self, 'Init error: {}'.format(msg))
|
|
|
|
|
|
class CmdInit(CmdBase):
|
|
CONFIG_TEMPLATE = '''[Global]
|
|
# Default target
|
|
Target =
|
|
|
|
# Supported clouds: AWS, GCP
|
|
Cloud = AWS
|
|
|
|
# This global storage path is going to be deprecated in the next version.
|
|
# Please use StoragePath from a specific cloud instead.
|
|
#StoragePath =
|
|
|
|
# Log levels: Debug, Info, Warning and Error
|
|
LogLevel = Info
|
|
|
|
[AWS]
|
|
CredentialPath = ~/.aws/credentials
|
|
CredentialSection = default
|
|
|
|
StoragePath = dvc/tutorial
|
|
|
|
# Default settings for AWS instances:
|
|
Type = t2.nano
|
|
Image = ami-2d39803a
|
|
|
|
SpotPrice =
|
|
SpotTimeout = 300
|
|
|
|
KeyPairName = dvc-key
|
|
KeyPairDir = ~/.ssh
|
|
SecurityGroup = dvc-sg
|
|
|
|
Region = us-east-1
|
|
Zone = us-east-1a
|
|
SubnetId =
|
|
|
|
Volume = my-100gb-drive-io
|
|
|
|
Monitoring = false
|
|
EbsOptimized = false
|
|
AllDisksAsRAID0 = false
|
|
|
|
[GCP]
|
|
StoragePath =
|
|
ProjectName =
|
|
'''
|
|
|
|
def __init__(self, settings):
|
|
super(CmdInit, self).__init__(settings)
|
|
|
|
@property
|
|
def is_locker(self):
|
|
return False
|
|
|
|
def get_not_existing_path(self, *args):
|
|
path = Path(os.path.join(self.git.git_dir, *args))
|
|
if path.exists():
|
|
raise InitError('Path "{}" already exist'.format(path.name))
|
|
return path
|
|
|
|
def get_not_existing_conf_file_name(self):
|
|
file_name = os.path.join(self.git.git_dir, Config.CONFIG_DIR, Config.CONFIG)
|
|
if os.path.exists(file_name):
|
|
raise InitError('Configuration file "{}" already exist'.format(file_name))
|
|
return file_name
|
|
|
|
def run(self):
|
|
if not self.no_git_actions and not self.git.is_ready_to_go():
|
|
return 1
|
|
|
|
if os.path.realpath(os.path.curdir) != self.settings.git.git_dir_abs:
|
|
Logger.error('DVC error: initialization could be done only from git root directory {}'.format(
|
|
self.settings.git.git_dir_abs
|
|
))
|
|
return 1
|
|
|
|
config_dir_path = self.get_not_existing_path(Config.CONFIG_DIR)
|
|
cache_dir_path = self.get_not_existing_path(Config.CONFIG_DIR, Config.CACHE_DIR_NAME)
|
|
|
|
conf_file_name = self.get_not_existing_conf_file_name()
|
|
|
|
config_dir_path.mkdir()
|
|
cache_dir_path.mkdir()
|
|
Logger.info('Directories {}/, {}/ were created'.format(
|
|
config_dir_path.name,
|
|
os.path.join(config_dir_path.name, cache_dir_path.name)))
|
|
|
|
conf_file = open(conf_file_name, 'wt')
|
|
conf_file.write(self.CONFIG_TEMPLATE)
|
|
conf_file.close()
|
|
|
|
self.git.modify_gitignore([os.path.join(config_dir_path.name, cache_dir_path.name),
|
|
os.path.join(config_dir_path.name, os.path.basename(self.git.lock_file)),
|
|
'*' + DataItem.LOCAL_STATE_FILE_SUFFIX])
|
|
|
|
message = 'DVC init. cache dir {}'.format(cache_dir_path.name)
|
|
return self.commit_if_needed(message)
|