mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
5 years ago
|
from __future__ import absolute_import, unicode_literals
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from collections import OrderedDict
|
||
|
|
|
||
|
|
from virtualenv.util.six import ensure_text
|
||
|
|
|
||
|
|
|
||
|
|
class PyEnvCfg(object):
|
||
|
|
def __init__(self, content, path):
|
||
|
|
self.content = content
|
||
|
|
self.path = path
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_folder(cls, folder):
|
||
|
|
return cls.from_file(folder / "pyvenv.cfg")
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_file(cls, path):
|
||
|
|
content = cls._read_values(path) if path.exists() else OrderedDict()
|
||
|
|
return PyEnvCfg(content, path)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _read_values(path):
|
||
|
|
content = OrderedDict()
|
||
|
|
for line in path.read_text(encoding="utf-8").splitlines():
|
||
|
|
equals_at = line.index("=")
|
||
|
|
key = line[:equals_at].strip()
|
||
|
|
value = line[equals_at + 1 :].strip()
|
||
|
|
content[key] = value
|
||
|
|
return content
|
||
|
|
|
||
|
|
def write(self):
|
||
|
|
logging.debug("write %s", ensure_text(str(self.path)))
|
||
|
|
text = ""
|
||
|
|
for key, value in self.content.items():
|
||
|
|
line = "{} = {}".format(key, value)
|
||
|
|
logging.debug("\t%s", line)
|
||
|
|
text += line
|
||
|
|
text += "\n"
|
||
|
|
self.path.write_text(text, encoding="utf-8")
|
||
|
|
|
||
|
|
def refresh(self):
|
||
|
|
self.content = self._read_values(self.path)
|
||
|
|
return self.content
|
||
|
|
|
||
|
|
def __setitem__(self, key, value):
|
||
|
|
self.content[key] = value
|
||
|
|
|
||
|
|
def __getitem__(self, key):
|
||
|
|
return self.content[key]
|
||
|
|
|
||
|
|
def __contains__(self, item):
|
||
|
|
return item in self.content
|
||
|
|
|
||
|
|
def update(self, other):
|
||
|
|
self.content.update(other)
|
||
|
|
return self
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return "{}(path={})".format(self.__class__.__name__, self.path)
|