File size: 2,741 Bytes
75eef29 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import requests
import schedule
import time
import os
import threading
from fastapi import FastAPI
app = FastAPI()
# 从环境变量获取基础 URL
BASE_URL = os.getenv('BASE_URL', "")
def check_available_tokens():
"""
检查可用的 Tokens 数量
"""
try:
response = requests.get(f"{BASE_URL}")
if response.status_code == 200:
content = response.text
if '当前可用 Tokens 数量：<span class="text-blue-600">0' in content:
return True
return False
except Exception as e:
print(f"检查 Tokens 时发生错误: {e}")
return False
def read_and_prepare_tokens(file_path):
"""
从文件读取tokens并使用\r\n连接
"""
try:
# 适配 Hugging Face Spaces 的文件路径
if os.path.exists(file_path):
with open(file_path, 'r') as file:
# 去除每行的空白字符，过滤掉空行
tokens = [line.strip() for line in file if line.strip()]
else:
# 如果文件不存在，尝试在 /home/user 目录下查找
alternative_path = os.path.join('/home/user', file_path)
with open(alternative_path, 'r') as file:
tokens = [line.strip() for line in file if line.strip()]
# 使用\r\n连接tokens
combined_token = '\r\n'.join(tokens)
return combined_token
except FileNotFoundError:
print(f"错误：文件 {file_path} 未找到")
return None
except Exception as e:
print(f"读取文件发生错误: {e}")
return None
def post_tokens():
"""
读取tokens并发送单次POST请求
"""
url = f"{BASE_URL}/upload"
# 读取并准备tokens
combined_token = read_and_prepare_tokens('access_token.txt')
if not combined_token:
print("没有可用的tokens")
# 触发一次at.py环境变量读取生成access_token.txt
os.system('python at.py')
return
try:
payload = {"text": combined_token}
response = requests.post(url, data=payload)
#print(f"响应状态码: {response.status_code}")
print(f"响应内容: {response.text}")
except Exception as e:
print(f"请求发生错误: {e}")
@app.get("/")
async def main():
# 首先检查是否需要发送 tokens
if not check_available_tokens():
print("当前仍有可用 Tokens，无需发送")
return
return {"message": post_tokens()}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860)
|