Skip to content

Commit 2a3d0a9

Browse files
committed
feat: 重写settings,放弃env存储,改到数据库
1 parent a891fbe commit 2a3d0a9

File tree

1 file changed

+20
-35
lines changed

1 file changed

+20
-35
lines changed

core/settings.py

Lines changed: 20 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -2,20 +2,21 @@
22
# @Author : Lan
33
# @File : settings.py
44
# @Software: PyCharm
5+
import json
56
from pathlib import Path
67

78
BASE_DIR = Path(__file__).resolve().parent.parent
89
data_root = BASE_DIR / 'data'
10+
911
if not data_root.exists():
1012
data_root.mkdir(parents=True, exist_ok=True)
11-
env_path = data_root / '.env2'
12-
default_value = {
13+
14+
DEFAULT_CONFIG = {
1315
'file_storage': 'local',
1416
'name': '文件快递柜-FileCodeBox',
1517
'description': '开箱即用的文件快传系统',
1618
'page_explain': '根据《中华人民共和国网络安全法》、《中华人民共和国刑法》、《中华人民共和国治安管理处罚法》等相关规定。 传播或存储违法、违规内容,会受到相关处罚,严重者将承担刑事责任。请勿上传非法文件,本站坚决配合相关部门,确保网络内容的安全,和谐,打造绿色网络环境。',
1719
'keywords': 'FileCodeBox, 文件快递柜, 口令传送箱, 匿名口令分享文本, 文件',
18-
'max_save_seconds': 0,
1920
's3_access_key_id': '',
2021
's3_secret_access_key': '',
2122
's3_bucket_name': '',
@@ -34,6 +35,7 @@
3435
'admin_token': 'FileCodeBox2023',
3536
'openUpload': 1,
3637
'uploadSize': 1024 * 1024 * 10,
38+
'expireStyle': ['day', 'hour', 'minute', 'forever', 'count'],
3739
'uploadMinute': 1,
3840
'opacity': 0.9,
3941
'background': '',
@@ -45,42 +47,25 @@
4547

4648

4749
class Settings:
48-
__instance = None
49-
50-
def __init__(self):
51-
# 读取.env
52-
if not (env_path).exists():
53-
with open(env_path, 'w', encoding='utf-8') as f:
54-
for key, value in default_value.items():
55-
f.write(f'{key}={value}\n')
56-
# 更新default_value
57-
with open(env_path, 'r', encoding='utf-8') as f:
58-
for line in f.readlines():
59-
key, value = line.strip().split('=', maxsplit=1)
60-
# 将字符串转换为原本的类型
61-
if key and not key.startswith('opendal_') and isinstance(default_value.get(key, ''), int):
62-
value = int(value)
63-
default_value[key] = value
64-
65-
# 更新self
66-
for key, value in default_value.items():
67-
self.__setattr__(key, value)
50+
def __init__(self, defaults=None):
51+
self.default_config = defaults or {}
52+
self.user_config = {}
6853

69-
def __new__(cls, *args, **kwargs):
70-
if not cls.__instance:
71-
cls.__instance = super(Settings, cls).__new__(cls, *args, **kwargs)
72-
return cls.__instance
54+
def __getattr__(self, attr):
55+
if attr in self.user_config:
56+
return self.user_config[attr]
57+
if attr in self.default_config:
58+
return self.default_config[attr]
59+
raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{attr}'")
7360

7461
def __setattr__(self, key, value):
75-
if not key.startswith('opendal_') and type(value) == str and value.isnumeric():
76-
value = int(value)
77-
self.__dict__[key] = value
78-
with open(env_path, 'w', encoding='utf-8') as f:
79-
for key, value in self.__dict__.items():
80-
f.write(f'{key}={value}\n')
62+
if key in ['default_config', 'user_config']:
63+
super().__setattr__(key, value)
64+
else:
65+
self.user_config[key] = value
8166

8267
def items(self):
83-
return self.__dict__.items()
68+
return {**self.default_config, **self.user_config}.items()
8469

8570

86-
settings = Settings()
71+
settings = Settings(DEFAULT_CONFIG)

0 commit comments

Comments
 (0)