mirror of
https://github.com/lanqian528/chat2api.git
synced 2026-06-13 21:02:46 +08:00
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import asyncio
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from chatgpt.refreshToken import rt2ac
|
|
from utils.Logger import logger
|
|
from utils.config import authorization_list
|
|
import chatgpt.globals as globals
|
|
|
|
|
|
def get_req_token(req_token):
|
|
if req_token in authorization_list:
|
|
if globals.token_list:
|
|
global count
|
|
count += 1
|
|
count %= len(globals.token_list)
|
|
while globals.token_list[count] in globals.error_token_list:
|
|
count += 1
|
|
count %= len(globals.token_list)
|
|
return globals.token_list[count]
|
|
else:
|
|
return None
|
|
else:
|
|
return req_token
|
|
|
|
|
|
async def verify_token(req_token):
|
|
if not req_token:
|
|
if authorization_list:
|
|
logger.error("Unauthorized with empty token.")
|
|
raise HTTPException(status_code=401)
|
|
else:
|
|
return None
|
|
else:
|
|
if req_token.startswith("eyJhbGciOi") or req_token.startswith("fk-"):
|
|
access_token = req_token
|
|
return access_token
|
|
elif len(req_token) == 45:
|
|
try:
|
|
access_token = await rt2ac(req_token, force_refresh=False)
|
|
return access_token
|
|
except HTTPException as e:
|
|
raise HTTPException(status_code=e.status_code, detail=e.detail)
|
|
else:
|
|
return req_token
|
|
|
|
|
|
async def refresh_all_tokens(force_refresh=False):
|
|
for token in globals.token_list:
|
|
if len(token) == 45:
|
|
try:
|
|
await asyncio.sleep(2)
|
|
await rt2ac(token, force_refresh=force_refresh)
|
|
except HTTPException:
|
|
pass
|
|
logger.info("All tokens refreshed.")
|