52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
import os
|
|
import requests
|
|
|
|
# 定义常量URL
|
|
AD_URL = "https://raw.githubusercontent.com/NianBroken/ZFCheckScores/main/data/advertise.txt"
|
|
NO_AD_USERS_LIST_URL = "https://raw.githubusercontent.com/NianBroken/ZFCheckScores/main/data/no_ad_users_list.txt"
|
|
|
|
# 定义本地数据存储路径
|
|
FOLDER_PATH = "data"
|
|
INFO_FILE_PATH = os.path.join(FOLDER_PATH, "info.txt")
|
|
|
|
# 从文件中读取当前用户
|
|
try:
|
|
with open(INFO_FILE_PATH, "r") as info_file:
|
|
CURRENT_USER = info_file.read().strip()
|
|
except Exception:
|
|
CURRENT_USER = "None"
|
|
|
|
|
|
# 发送HTTP GET请求获取指定URL的内容
|
|
def fetch_content(url):
|
|
try:
|
|
response = requests.get(url)
|
|
return response.text if response.status_code == 200 else None
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
# 获取免广告用户列表
|
|
def get_no_ad_users_list():
|
|
try:
|
|
return fetch_content(NO_AD_USERS_LIST_URL) or []
|
|
except Exception:
|
|
return []
|
|
|
|
|
|
# 获取广告内容
|
|
def get_advertise():
|
|
try:
|
|
# 获取免广告用户列表
|
|
no_ad_users_list = get_no_ad_users_list()
|
|
|
|
# 如果当前用户不在免广告用户列表中，获取广告内容
|
|
if CURRENT_USER not in no_ad_users_list:
|
|
ad_content = fetch_content(AD_URL)
|
|
return ad_content
|
|
# 当前用户在免广告用户列表中，不推送广告
|
|
else:
|
|
return None
|
|
except Exception:
|
|
return None
|