mirror of
https://github.com/Kiritow/wg-ops.git
synced 2024-03-22 13:11:37 +08:00
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import logging
|
||
|
|
import json
|
||
|
|
import traceback
|
||
|
|
import base64
|
||
|
|
|
||
|
|
|
||
|
|
class SimpleLogger(object):
|
||
|
|
def __init__(self, name=None, filename=None, fileonly=False,
|
||
|
|
level=logging.INFO,
|
||
|
|
default_encoding='utf-8',
|
||
|
|
log_format="%(asctime)s @%(module)s [%(levelname)s] %(funcName)s: %(message)s"):
|
||
|
|
if name is None:
|
||
|
|
name = __name__
|
||
|
|
|
||
|
|
if not filename and fileonly:
|
||
|
|
raise Exception("FileOnly=True but no filename provided.")
|
||
|
|
|
||
|
|
self.logger = logging.getLogger(name)
|
||
|
|
if not getattr(self.logger, "_is_configured", None):
|
||
|
|
formatter = logging.Formatter(log_format)
|
||
|
|
if not fileonly:
|
||
|
|
console_handler = logging.StreamHandler()
|
||
|
|
console_handler.setFormatter(formatter)
|
||
|
|
self.logger.addHandler(console_handler)
|
||
|
|
if filename is not None:
|
||
|
|
file_handler = logging.FileHandler(filename, encoding=default_encoding)
|
||
|
|
file_handler.setFormatter(formatter)
|
||
|
|
self.logger.addHandler(file_handler)
|
||
|
|
self.logger.setLevel(level)
|
||
|
|
setattr(self.logger, "_is_configured", True)
|
||
|
|
|
||
|
|
# Just acts as a logger
|
||
|
|
def __getattr__(self, name):
|
||
|
|
return getattr(self.logger, name)
|
||
|
|
|
||
|
|
|
||
|
|
logger = SimpleLogger()
|
||
|
|
|
||
|
|
|
||
|
|
def load_config(filename=None):
|
||
|
|
config_filename = filename or "local/config.json"
|
||
|
|
try:
|
||
|
|
with open(config_filename) as f:
|
||
|
|
return json.loads(f.read())
|
||
|
|
except Exception:
|
||
|
|
logger.error("Unable to load config: {}".format(traceback.format_exc()))
|
||
|
|
return {}
|
||
|
|
|
||
|
|
|
||
|
|
def save_config(config, filename=None):
|
||
|
|
config_filename = filename or "local/config.json"
|
||
|
|
content = json.dumps(config, ensure_ascii=False, default=str, indent=2)
|
||
|
|
try:
|
||
|
|
with open(config_filename, "w", encoding='utf-8') as f:
|
||
|
|
f.write(content)
|
||
|
|
except Exception:
|
||
|
|
logger.error("Unable to save config: {}".format(traceback.format_exc()))
|
||
|
|
logger.info("Config: {}".format(content))
|
||
|
|
|
||
|
|
|
||
|
|
def json_to_base64(content):
|
||
|
|
return base64.b64encode(json.dumps(content, ensure_ascii=False).encode('utf-8')).decode('utf-8')
|
||
|
|
|
||
|
|
|
||
|
|
def base64_to_json(content):
|
||
|
|
return json.loads(base64.b64decode(content).decode('utf-8'))
|