Spaces:
Paused
Paused
File size: 2,997 Bytes
53e18b4 |
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 |
# run_and_get_cookies.py
import subprocess
import os
import sys
import time
import requests
import json
def validate_cookies(cookies_data):
"""Validate that cf_clearance cookie is present and not empty"""
cookies = cookies_data.get('cookies', {})
return 'cf_clearance' in cookies and cookies['cf_clearance'].strip() != ''
def get_and_save_cookies(server_url, cookie_file_path, max_retries=3):
for attempt in range(max_retries):
try:
response = requests.get(server_url)
response.raise_for_status()
cookies_data = response.json()
if not validate_cookies(cookies_data):
print(f"Attempt {attempt + 1}: cf_clearance cookie not found, retrying...")
time.sleep(5)
continue
cookies_to_save = {
'cookies': cookies_data.get('cookies', {}),
'user_agent': cookies_data.get('user_agent', '')
}
os.makedirs(os.path.dirname(cookie_file_path), exist_ok=True)
with open(cookie_file_path, 'w', encoding='utf-8') as f:
json.dump(cookies_to_save, f, indent=4, ensure_ascii=False)
print("Successfully obtained and saved cookies with cf_clearance!")
return True
except requests.exceptions.ConnectionError as e:
print(f"Connection error on attempt {attempt + 1}: {str(e)}")
if attempt < max_retries - 1:
time.sleep(5)
else:
print("Max retries reached. Failed to get valid cookies.")
return False
print("Failed to obtain valid cf_clearance cookie after all attempts")
return False
def run_server_background():
script_dir = os.path.dirname(os.path.abspath(__file__))
server_script = os.path.abspath(os.path.join(script_dir, "server.py"))
server_dir = os.path.dirname(server_script)
os.makedirs(server_dir, exist_ok=True)
try:
process = subprocess.Popen(
[sys.executable, server_script],
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL,
cwd=server_dir,
start_new_session=True
)
return process
except Exception:
return None
if __name__ == "__main__":
print("Getting the cookies...")
server_process = run_server_background()
if server_process:
# Increase initial wait time to ensure server is fully started
time.sleep(10)
server_url = "http://localhost:8000/cookies?url=https://chat.deepseek.com"
cookie_file = "dsk/cookies.json"
# Increase max retries for more reliability
success = get_and_save_cookies(server_url, cookie_file, max_retries=5)
if not success:
print("Failed to obtain valid cookies.")
server_process.terminate()
sys.exit(1)
server_process.terminate()
else:
print("Failed to start server.")
sys.exit(1) |