mirror of https://github.com/tp4a/teleport
125 lines
3.2 KiB
Python
125 lines
3.2 KiB
Python
# -*- coding: utf8 -*-
|
||
|
||
import os
|
||
import sys
|
||
import platform
|
||
from . import colorconsole as cc
|
||
|
||
__all__ = ['cfg']
|
||
|
||
|
||
class TpDict(dict):
|
||
"""
|
||
可以像属性一样访问字典的 Key，var.key 等同于 var['key']
|
||
"""
|
||
|
||
def __getattr__(self, name):
|
||
try:
|
||
return self[name]
|
||
except KeyError:
|
||
# print(self.__class__.__name__)
|
||
raise
|
||
|
||
def __setattr__(self, name, val):
|
||
self[name] = val
|
||
|
||
|
||
class ConfigFile(TpDict):
|
||
def __init__(self, **kwargs):
|
||
super().__init__(**kwargs)
|
||
# self.__file_name = None
|
||
# self.__save_indent = 0
|
||
# self.__loaded = False
|
||
|
||
def init(self, cfg_file):
|
||
if not self.load(cfg_file, True):
|
||
return False
|
||
|
||
self['ROOT_PATH'] = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
|
||
|
||
self['py_exec'] = sys.executable
|
||
|
||
_py_ver = platform.python_version_tuple()
|
||
self['py_ver'] = _py_ver
|
||
self['py_ver_str'] = '%s%s' % (_py_ver[0], _py_ver[1])
|
||
self['is_py2'] = sys.version_info[0] == 2
|
||
self['is_py3'] = sys.version_info[0] == 3
|
||
|
||
_bits = platform.architecture()[0]
|
||
if _bits == '64bit':
|
||
self['is_x64'] = True
|
||
self['is_x86'] = False
|
||
else:
|
||
self['is_x64'] = False
|
||
self['is_x86'] = True
|
||
|
||
_os = platform.system().lower()
|
||
|
||
self['dist'] = ''
|
||
if _os == 'windows':
|
||
self['dist'] = 'windows'
|
||
elif _os == 'linux':
|
||
self['dist'] = 'linux'
|
||
elif _os == 'darwin':
|
||
self['dist'] = 'macos'
|
||
else:
|
||
cc.e('not support this OS: {}'.format(platform.system()))
|
||
return False
|
||
|
||
return True
|
||
|
||
def load_str(self, module, code):
|
||
m = type(sys)(module)
|
||
m.__module_class__ = type(sys)
|
||
m.__file__ = module
|
||
|
||
try:
|
||
exec(compile(code, module, 'exec'), m.__dict__)
|
||
except Exception as e:
|
||
cc.e('%s\n' % str(e))
|
||
# print(str(e))
|
||
# if eom_dev_conf.debug:
|
||
# raise
|
||
return False
|
||
|
||
for y in m.__dict__:
|
||
if '__' == y[:2]:
|
||
continue
|
||
if isinstance(m.__dict__[y], dict):
|
||
self[y] = TpDict()
|
||
self._assign_dict(m.__dict__[y], self[y])
|
||
else:
|
||
self[y] = m.__dict__[y]
|
||
|
||
return True
|
||
|
||
def load(self, full_path, must_exists=True):
|
||
try:
|
||
f = open(full_path, encoding='utf8')
|
||
code = f.read()
|
||
f.close()
|
||
self.__loaded = True
|
||
except IOError:
|
||
if must_exists:
|
||
cc.e('Can not load config file: %s\n' % full_path)
|
||
return False
|
||
|
||
module = os.path.basename(full_path)
|
||
if not self.load_str(module, code):
|
||
return False
|
||
|
||
self.__file_name = full_path
|
||
return True
|
||
|
||
def _assign_dict(self, _from, _to):
|
||
for y in _from:
|
||
if isinstance(_from[y], dict):
|
||
_to[y] = TpDict()
|
||
self._assign_dict(_from[y], _to[y])
|
||
else:
|
||
_to[y] = _from[y]
|
||
|
||
|
||
cfg = ConfigFile()
|
||
del ConfigFile
|