feat: 增加自定义域名的支持
This commit is contained in:
37
config.py
Normal file
37
config.py
Normal file
@@ -0,0 +1,37 @@
|
||||
from dotenv import load_dotenv
|
||||
import os
|
||||
import sys
|
||||
|
||||
|
||||
class Config:
|
||||
def __init__(self):
|
||||
# 获取应用程序的根目录路径
|
||||
if getattr(sys, "frozen", False):
|
||||
# 如果是打包后的可执行文件
|
||||
application_path = os.path.dirname(sys.executable)
|
||||
else:
|
||||
# 如果是开发环境
|
||||
application_path = os.path.dirname(os.path.abspath(__file__))
|
||||
|
||||
# 指定 .env 文件的路径
|
||||
dotenv_path = os.path.join(application_path, ".env")
|
||||
|
||||
# 加载 .env 文件
|
||||
load_dotenv(dotenv_path)
|
||||
|
||||
def get_temp_mail(self):
|
||||
return os.getenv("TEMP_MAIL")
|
||||
|
||||
def get_domain(self):
|
||||
return os.getenv("DOMAIN")
|
||||
|
||||
|
||||
# 使用示例
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
config = Config()
|
||||
print("环境变量加载成功!")
|
||||
print(f"临时邮箱: {config.get_temp_mail()}")
|
||||
print(f"域名: {config.get_domain()}")
|
||||
except ValueError as e:
|
||||
print(f"错误: {e}")
|
||||
Reference in New Issue
Block a user