51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from enum import unique
|
|
from enum import Enum
|
|
import configparser
|
|
|
|
|
|
@unique
|
|
class ServerEnv(Enum):
|
|
DEBUG = 1,
|
|
SERVER = 2,
|
|
|
|
|
|
class ServerConfig(object):
|
|
|
|
def set_tencent_api(self):
|
|
self._tencent_api_id = self._config['TencentCloud']['SecretId']
|
|
self._tencent_api_key = self._config['TencentCloud']['SecretKey']
|
|
|
|
def set_baidu_api(self):
|
|
self._baidu_api_key = self._config['BaiduCloud']['ApiKey']
|
|
self._baidu_api_secret_key = self._config['BaiduCloud']['SecretKey']
|
|
|
|
def tell_listen(self) -> tuple:
|
|
return self._listen, self._listen_port
|
|
|
|
def tell_tencent_api(self) -> tuple:
|
|
return self._tencent_api_id, self._tencent_api_key
|
|
|
|
def tell_baidu_api(self) -> tuple:
|
|
return self._baidu_api_key, self._baidu_api_secret_key
|
|
|
|
def tell_env(self) -> ServerEnv:
|
|
return self._env
|
|
|
|
def __init__(self, env: ServerEnv = ServerEnv.DEBUG):
|
|
super(ServerConfig, self).__init__()
|
|
self._tencent_api_id: str | None = None
|
|
self._tencent_api_key: str | None = None
|
|
self._baidu_api_key: str | None = None
|
|
self._baidu_api_secret_key: str | None = None
|
|
self._listen: str = "0.0.0.0"
|
|
self._listen_port: int = 5000
|
|
self._config = configparser.ConfigParser()
|
|
self._config.optionxform = str
|
|
self._env = env
|
|
if self._env == ServerEnv.DEBUG:
|
|
self._config.read("settings.ini", encoding="utf-8")
|
|
elif self._env == ServerEnv.SERVER:
|
|
self._config.read('config.ini', encoding='utf-8')
|
|
self.set_tencent_api()
|
|
self.set_baidu_api()
|