lan
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
# @Time : 2023/8/15 09:51
|
|
# @Author : Lan
|
|
# @File : settings.py
|
|
# @Software: PyCharm
|
|
from pathlib import Path
|
|
|
|
data_root = Path('./data')
|
|
if not data_root.exists():
|
|
data_root.mkdir(parents=True, exist_ok=True)
|
|
env_path = data_root / '.env2'
|
|
default_value = {
|
|
'file_storage': 'local',
|
|
'name': '文件快递柜-FileCodeBox',
|
|
'description': '开箱即用的文件快传系统',
|
|
'keywords': 'FileCodeBox，文件快递柜，口令传送箱，匿名口令分享文本，文件',
|
|
's3_access_key_id': '',
|
|
's3_secret_access_key': '',
|
|
's3_bucket_name': '',
|
|
's3_endpoint_url': '',
|
|
'admin_token': 'FileCodeBox2023',
|
|
'openUpload': 1,
|
|
'uploadSize': 1024 * 1024 * 10,
|
|
'uploadMinute': 1,
|
|
'uploadCount': 10,
|
|
'errorMinute': 1,
|
|
'errorCount': 1,
|
|
}
|
|
|
|
|
|
class Settings:
|
|
__instance = None
|
|
|
|
def __init__(self):
|
|
# 读取.env
|
|
if not (env_path).exists():
|
|
with open(env_path, 'w', encoding='utf-8') as f:
|
|
for key, value in default_value.items():
|
|
f.write(f'{key}={value}\n')
|
|
# 更新default_value
|
|
with open(env_path, 'r', encoding='utf-8') as f:
|
|
for line in f.readlines():
|
|
key, value = line.strip().split('=')
|
|
# 将字符串转换为原本的类型
|
|
if value.isdigit():
|
|
value = int(value)
|
|
default_value[key] = value
|
|
|
|
# 更新self
|
|
for key, value in default_value.items():
|
|
self.__setattr__(key, value)
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
if not cls.__instance:
|
|
cls.__instance = super(Settings, cls).__new__(cls, *args, **kwargs)
|
|
return cls.__instance
|
|
|
|
def __setattr__(self, key, value):
|
|
self.__dict__[key] = value
|
|
with open(env_path, 'w', encoding='utf-8') as f:
|
|
for key, value in self.__dict__.items():
|
|
f.write(f'{key}={value}\n')
|
|
|
|
|
|
settings = Settings()
|