DriveTest2 / app.py
Mythus's picture
Update app.py
452c2d6 verified
raw
history blame contribute delete
No virus
1.58 kB
from flask import Flask, jsonify
from generator import run # Import the generator script as a module
import asyncio
import time
from concurrent.futures import ThreadPoolExecutor
app = Flask(__name__)
# Create a ThreadPoolExecutor
executor = ThreadPoolExecutor(1)
latest_token = None
auth = None # Define auth as a global variable
def update_token_periodically():
global latest_token, auth
while True:
event_loop = asyncio.new_event_loop()
asyncio.set_event_loop(event_loop)
success = False
while not success:
try:
latest_token, auth = event_loop.run_until_complete(run())
if latest_token and auth:
success = True
print(f"cookie: {latest_token}, auth: {auth}")
else:
print("Token generation failed. Retrying...")
except Exception as e:
print(f"An error occurred: {e}. Retrying...")
if not success:
time.sleep(5) # Short delay before retrying
event_loop.close()
# Wait the established amount of time before generating new tokens
time.sleep(50 * 60)
@app.route('/api/cookie')
def cookie():
global latest_token, auth
if latest_token and auth:
return jsonify({"cookie": latest_token, "auth": auth})
else:
return "Token not found.", 500
if __name__ == '__main__':
# Start the background thread
executor.submit(update_token_periodically)
app.run(host='0.0.0.0', port=8080)