oadmin/macm1new
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Refactor configuration loading and logging in config.py; update .env.example formatting

Browse Source
This commit is contained in:
chengchongzhen
2025-01-13 13:20:33 +08:00
parent 7a14f62b88
commit 15ffc5e07a
2 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.env.example
View File

@@ -1,6 +1,6 @@
DOMAIN='xxxxx.me' DOMAIN=xxxxx.me
TEMP_MAIL='xxxxxx' TEMP_MAIL=xxxxxx
BROWSER_USER_AGENT='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.6723.92 Safari/537.36' BROWSER_USER_AGENT=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.6723.92 Safari/537.36
# 代理 # 代理
# BROWSER_PROXY='http://127.0.0.1:2080' # BROWSER_PROXY='http://127.0.0.1:2080'

19
config.py
View File

@@ -1,6 +1,7 @@
from dotenv import load_dotenv from dotenv import load_dotenv
import os import os
import sys import sys
from logger import logging
class Config: class Config:
@@ -22,11 +23,21 @@ class Config:
# 加载 .env 文件 # 加载 .env 文件
load_dotenv(dotenv_path) load_dotenv(dotenv_path)
self.temp_mail = os.getenv("TEMP_MAIL", "").strip()
self.domain = os.getenv("DOMAIN", "").strip()
if not self.temp_mail:
raise ValueError("临时邮箱未配置，请在 .env 文件中设置 TEMP_MAIL")
if not self.domain:
raise ValueError("域名未配置，请在 .env 文件中设置 DOMAIN")
def get_temp_mail(self): def get_temp_mail(self):
return os.getenv("TEMP_MAIL") logging.info(f"\033[32m临时邮箱: {self.temp_mail}\033[0m")
return self.temp_mail
def get_domain(self): def get_domain(self):
return os.getenv("DOMAIN") logging.info(f"\033[32m域名: {self.domain}\033[0m")
return self.domain
# 使用示例 # 使用示例
@@ -34,7 +45,7 @@ if __name__ == "__main__":
try: try:
config = Config() config = Config()
print("环境变量加载成功！") print("环境变量加载成功！")
print(f"临时邮箱: {config.get_temp_mail()}") config.get_temp_mail()
print(f"域名: {config.get_domain()}") config.get_domain()
except ValueError as e: except ValueError as e:
print(f"错误: {e}") print(f"错误: {e}")