yuuki/libs/config.py
2021-03-10 21:23:53 +08:00

91 lines
2.9 KiB
Python

# -*- coding: utf-8 -*-
"""
Yuuki_Libs
(c) 2020 Star Inc.
This Source Code Form is subject to the terms of the Mozilla Public
License, v. 2.0. If a copy of the MPL was not distributed with this
file, You can obtain one at http://mozilla.org/MPL/2.0/.
"""
import os
import yaml
class Config:
"""
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
!! DO NOT TOUCH DEFAULT SETTINGS !!
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
Please config the value you want to set though `config.yaml`.
It will overwrite these settings.
"""
connect_info = {
"Host": "",
"Command_Path": "",
"LongPoll_path": "",
}
connect_header = {
"X-Line-Application": "",
"X-Line-Access": "",
"User-Agent": ""
}
system_config = {
"name": "Yuuki",
"version": "v6.5.3-beta_RC0",
"version_check": True,
"project_url": "https://tinyurl.com/syb-yuuki",
"man_page": "https://tinyurl.com/yuuki-manual",
"privacy_page": "OpenSource - Licensed under MPL 2.0",
"copyright": "(c)2020 Star Inc.",
"Seq": 0,
"Admin": [],
"Advanced": False,
"WebAdmin": False, # Advanced Mode Required
"MDS_Port": 2019,
"WebAdmin_Port": 2020,
"SecurityService": False,
"Hour_KickLimit": 10,
"Hour_CancelLimit": 10,
"Default_Language": "en",
"GroupMembers_Demand": 100,
"helper_LINE_ACCESS_KEYs": [],
}
def __init__(self, config_path="config.yaml"):
assert os.path.isfile(config_path), "The config file, `config.yaml` was not found."
with open(config_path, "r") as configfile:
self.config = yaml.load(configfile, Loader=yaml.FullLoader)
self._config_yuuki()
def _config_yuuki(self):
assert self.config is not None, "Invalid config file"
if "Yuuki" in self.config:
for key in self.config["Yuuki"]:
if key in self.system_config:
self.system_config[key] = self.config["Yuuki"][key]
return self._config_server()
def _config_server(self):
if "Server" in self.config.get("LINE"):
for key in self.config["LINE"]["Server"]:
if key in self.connect_info:
self.connect_info[key] = self.config["LINE"]["Server"][key]
return self._config_account()
def _config_account(self):
if "Account" in self.config.get("LINE"):
for key in self.config["LINE"]["Account"]:
if key in ["X-Line-Application", "User-Agent"]:
self.config["LINE"]["Account"][key] = self.config["LINE"]["Account"][key].replace("\\t", "\t")
if key in self.connect_header:
self.connect_header[key] = self.config["LINE"]["Account"][key]
if "helper_LINE_ACCESS_KEYs" in self.config.get("LINE"):
self.system_config["helper_LINE_ACCESS_KEYs"] = self.config["LINE"]["helper_LINE_ACCESS_KEYs"]