Upload 2 files

api.py

@@ -1,10 +1,25 @@
 import os
+from pathlib import Path
 import asyncio
 from flask import Flask, jsonify, request, logging as flog
 from flask_limiter.util import get_remote_address
+from playwright.async_api import async_playwright
+
 import hcaptcha_solver
 
 app = Flask(__name__)
+user_data_dir = Path(__file__).parent.joinpath("user_data_dir")
+context_dir = user_data_dir.joinpath("context")
+
+
+async def __init__():
+    global contexts
+    async with async_playwright() as p:
+        contexts = p.firefox.launch_persistent_context(
+            user_data_dir=context_dir,
+            headless=True,
+            locale="en-US"
+        )
 
 
 def get_ipaddr():
@@ -71,7 +86,7 @@ def solver_captcha():
     data = request.get_json(force=True, silent=True)
     if not check_request(require_data, data):
         return jsonify(msg="Unauthorized Request"), 403
-    return asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"]))
+    return asyncio.run(hcaptcha_solver.bytedance(contexts, data["host"], data["site_key"]))
 
 
 app.run(host="0.0.0.0", port=7860)

hcaptcha_solver.py

@@ -14,16 +14,6 @@ tmp_dir = Path(__file__).parent.joinpath("tmp_dir")
 contexts = None
 
 
-async def __init__():
-    async with async_playwright() as p:
-        global contexts
-        contexts = p.firefox.launch_persistent_context(
-            user_data_dir=context_dir,
-            headless=True,
-            locale="en-US"
-        )
-
-
 async def hit_challenge(context: ASyncContext, host, sitekey, times: int = 8):
     await context.route('**/*', lambda route, request: route_continuation(route, request, host, sitekey))
     page = context.pages[0]
@@ -143,7 +133,7 @@ async def route_continuation(route, request, host, sitekey):
         await route.continue_()
 
 
-async def bytedance(host, sitekey):
+async def bytedance(context, host, sitekey):
     # playwright install firefox --with-deps
 
     await Malenia.apply_stealth(contexts)