🐛🐛 修复通过环境变量获取自行定义环境变量的方式获取配置信息

This commit is contained in:
echo 2024-09-17 21:37:45 +08:00
parent 279168912c
commit 81a76d3e3b

View File

@ -4,7 +4,9 @@
# @author Echo # @author Echo
# @EditTime 2024/9/14 # @EditTime 2024/9/14
import datetime import datetime
import os
import random import random
import re
import time import time
import httpx import httpx
from typing import Text, Optional, List, Dict from typing import Text, Optional, List, Dict
@ -33,25 +35,29 @@ from typing import Text, Optional, List, Dict
""" """
设置配置常量 设置配置常量
""" """
API_KEY = "" # 天聚数行密钥 CONFIG = {
APP_ID = "" # 微信公众号appid "API_KEY": "", # 天聚数行密钥
APP_SECRET = "" # 微信公众号appsecret "APP_ID": "", # 微信公众号appid
WECHAT_USER_IDS = [] # 要推送的微信用户ID "APP_SECRET": "", # 微信公众号appsecret
TEMPLATE_ID = "" # 模板ID "TEMPLATE_ID": "", # 模板ID
CITY_NAME = "" # 城市 "CITY_NAME": "", # 城市
AREA = "" # 区县 "AREA": "", # 区县
EXAMINATION_DATE = "" # 考研日期 "EXAMINATION_DATE": "", # 考研日期
USER = "" # 对象称呼 "USER": "", # 对象称呼
BIRTHDAY = "" # 对象生日 "BIRTHDAY": "", # 对象生日
STAR_SIGN = "" # 对象星座 "STAR_SIGN": "", # 对象星座
LOVE_DATE = "" # 恋爱开始日期 "LOVE_DATE": "" # 恋爱开始日期
}
# 从环境变量中获取配置,如果环境变量不存在则使用默认值
for key in CONFIG:
CONFIG[key] = os.environ.get(key, CONFIG[key])
# if 'tian_api_key' in os.environ: # 特殊处理 WECHAT_USER_IDS因为它需要被分割
# api_key = re.split("@|&", os.environ.get("tian_api_key")) WECHAT_USER_IDS = re.split("@|&", os.environ.get("WECHAT_USER_IDS", "")) if "WECHAT_USER_IDS" in os.environ else []
# else:
# api_key = [] # 使用全局变量
# print("未查找到tian_api_key变量.") globals().update(CONFIG)
def time_diff(time1: Text, time2: Text, format) -> int: def time_diff(time1: Text, time2: Text, format) -> int: