mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 07:18:28 +01:00
Ruslan Kuprieiev
3410f8048a
Speeds up `time dvc` from 1.2 seconds to <0.2 seconds. Signed-off-by: Ruslan Kuprieiev <[email protected]>
102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
import os
|
|
import configobj
|
|
|
|
from dvc.command.common.base import CmdBase
|
|
from dvc.logger import Logger
|
|
from dvc.config import Config
|
|
|
|
|
|
class CmdConfig(CmdBase):
|
|
def __init__(self, args):
|
|
from dvc.project import Project
|
|
|
|
self.args = args
|
|
root_dir = self._find_root()
|
|
if args.local:
|
|
config = Config.CONFIG_LOCAL
|
|
else:
|
|
config = Config.CONFIG
|
|
self.config_file = os.path.join(root_dir, Project.DVC_DIR, config)
|
|
# Using configobj because it doesn't
|
|
# drop comments like configparser does.
|
|
self.configobj = configobj.ConfigObj(self.config_file)
|
|
|
|
def run_cmd(self):
|
|
return self.run()
|
|
|
|
def _get_key(self, d, name, add=False):
|
|
for k in d.keys():
|
|
if k.lower() == name.lower():
|
|
return k
|
|
|
|
if add:
|
|
d[name] = {}
|
|
return name
|
|
|
|
return None
|
|
|
|
def save(self):
|
|
try:
|
|
self.configobj.write()
|
|
except Exception as exc:
|
|
Logger.error("Failed to write config '{}'".format(self.configobj.filename), exc)
|
|
return 1
|
|
return 0
|
|
|
|
def unset(self, section, opt=None):
|
|
if section not in self.configobj.keys():
|
|
Logger.error("Section '{}' doesn't exist".format(section))
|
|
return 1
|
|
|
|
if opt in self.configobj[section].keys():
|
|
del self.configobj[section][opt]
|
|
|
|
if len(self.configobj[section]) == 0 or opt == None:
|
|
del self.configobj[section]
|
|
|
|
return self.save()
|
|
|
|
def show(self):
|
|
Logger.info(self.configobj[self.section][self.opt])
|
|
return 0
|
|
|
|
def set(self, section, opt, value):
|
|
if section not in self.configobj.keys():
|
|
self.configobj[section] = {}
|
|
|
|
self.configobj[section][opt] = value
|
|
|
|
return self.save()
|
|
|
|
def check_opt(self):
|
|
_section, _opt = self.args.name.strip().split('.', 1)
|
|
add = (self.args.value != None and self.args.unset == False)
|
|
|
|
section = self._get_key(self.configobj, _section, add)
|
|
|
|
if not section:
|
|
Logger.error('Invalid option name {}'.format(_section))
|
|
return 1
|
|
|
|
opt = self._get_key(self.configobj[section], _opt, add)
|
|
if not opt:
|
|
Logger.error('Invalid option value: {}'.format(_opt))
|
|
return 1
|
|
|
|
self.section = section
|
|
self.opt = opt
|
|
|
|
return 0
|
|
|
|
def run(self):
|
|
if self.check_opt() != 0:
|
|
return 1
|
|
|
|
if self.args.unset:
|
|
return self.unset(self.section, self.opt)
|
|
|
|
if self.args.value is None:
|
|
return self.show()
|
|
|
|
return self.set(self.section, self.opt, self.args.value)
|