mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-11 07:18:28 +01:00
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
"""
|
|
DVC config objects.
|
|
"""
|
|
import os
|
|
import configparser
|
|
|
|
from dvc.exceptions import DvcException
|
|
from dvc.logger import Logger
|
|
|
|
|
|
class ConfigError(DvcException):
|
|
""" DVC config exception """
|
|
def __init__(self, msg):
|
|
DvcException.__init__(self, 'Config file error: {}'.format(msg))
|
|
|
|
|
|
class ConfigI(object):
|
|
""" Basic config instance """
|
|
CONFIG_DIR = '.dvc'
|
|
TARGET_FILE_DEFAULT = 'target'
|
|
CONFIG = 'config'
|
|
STATE_DIR_NAME = 'state'
|
|
CACHE_DIR_NAME = 'cache'
|
|
|
|
CACHE_DIR = os.path.join(CONFIG_DIR, CACHE_DIR_NAME)
|
|
STATE_DIR = os.path.join(CONFIG_DIR, STATE_DIR_NAME)
|
|
|
|
def __init__(self, cloud=None, conf_parser=None):
|
|
self._conf_parser = None
|
|
self._cloud = None
|
|
self.set(cloud, conf_parser)
|
|
|
|
def set(self, cloud=None, conf_parser=None):
|
|
""" Set config params """
|
|
self._cloud = cloud
|
|
self._conf_parser = conf_parser
|
|
|
|
# @classmethod
|
|
# def CACHE_DIR(self):
|
|
# """ Directory with cached data files """
|
|
# return os.path.join(self.CONFIG_DIR, self.CACHE_DIR_NAME)
|
|
#
|
|
# @classmethod
|
|
# def STATE_DIR(self):
|
|
# """ Directory with state files """
|
|
# return os.path.join(self.CONFIG_DIR, self.STATE_DIR_NAME)
|
|
|
|
@property
|
|
def cloud(self):
|
|
""" Cloud config """
|
|
return self._cloud
|
|
|
|
@property
|
|
def conf_parser(self):
|
|
return self._conf_parser
|
|
|
|
|
|
class Config(ConfigI):
|
|
""" Parsed config object """
|
|
def __init__(self, conf_file=ConfigI.CONFIG, conf_pseudo_file=None):
|
|
"""
|
|
Params:
|
|
conf_file (String): configuration file
|
|
conf_pseudo_file (String): for unit testing, something that supports readline;
|
|
supersedes conf_file
|
|
"""
|
|
self._conf_file = conf_file
|
|
self._config = configparser.SafeConfigParser()
|
|
|
|
if conf_pseudo_file is not None:
|
|
self._config.readfp(conf_pseudo_file)
|
|
else:
|
|
fname = os.path.join(self.CONFIG_DIR, conf_file)
|
|
if not os.path.isfile(fname):
|
|
raise ConfigError('Config file "{}" does not exist {}'.format(fname, os.getcwd()))
|
|
self._config.read(fname)
|
|
|
|
level = self._config['Global']['LogLevel']
|
|
Logger.set_level(level)
|
|
|
|
super(Config, self).__init__(self._config['Global']['Cloud'],
|
|
self._config)
|
|
|
|
@property
|
|
def file(self):
|
|
""" Config file object """
|
|
return self._conf_file
|