Upload 2 files

api.py

@@ -2,6 +2,7 @@ import hashlib
 import os
 import asyncio
 import uuid
+import shutil
 from flask import Flask, jsonify, request, logging as flog
 from flask_limiter.util import get_remote_address
 import hcaptcha_solver
@@ -79,7 +80,11 @@ 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"], generate_uuid()))
+    dir_path=generate_uuid()
+    resp=asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"], dir_path))
+    shutil.rmtree(dir_path)
+    shutil.rmtree("tmp_dir")
+    return resp
 
 
 @app.route("/api/update", methods=["POST"])
@@ -92,4 +97,4 @@ def update_model():
     return jsonify(statue="ok"), 200
 
 
-app.run(host="0.0.0.0", port=7860)
+app.run(host="0.0.0.0", port=8081)

hcaptcha_solver.py

@@ -1,5 +1,6 @@
 from pathlib import Path
-import os
+
+import traceback
 from loguru import logger
 from playwright.async_api import BrowserContext as ASyncContext, async_playwright
 import hcaptcha_challenger as solver
@@ -17,7 +18,7 @@ tmp_dir = Path(__file__).parent.joinpath("tmp_dir")
 async def hit_challenge(context: ASyncContext, host, sitekey, user_data_dir, times: int = 8):
     await context.route('**/*', lambda route, request: route_continuation(route, request, host, sitekey))
     page = context.pages[0]
-    agent = AgentT.from_page(page=page, tmp_dir=tmp_dir)
+    agent = AgentT.from_page(page=page, tmp_dir=tmp_dir,self_supervised=True)
     await page.goto(f"https://{host}")
 
     await agent.handle_checkbox()
@@ -32,8 +33,7 @@ async def hit_challenge(context: ASyncContext, host, sitekey, user_data_dir, tim
                 await fl.locator("//div[@class='refresh button']").click()
             case agent.status.CHALLENGE_SUCCESS:
                 rqdata = agent.cr.__dict__
-                print(os.system(f"rm -rf {tmp_dir}"))
-                print(os.system(f"rm -rf {user_data_dir}"))
+                await context.close()
                 return rqdata["generated_pass_UUID"]
 
 
@@ -105,6 +105,7 @@ async def route_continuation(route, request, host, sitekey):
 
 
 async def bytedance(host, sitekey, user_data_dirs):
+    print(user_data_dirs)
     # playwright install firefox --with-deps
     try:
         async with async_playwright() as p:
@@ -117,5 +118,9 @@ async def bytedance(host, sitekey, user_data_dirs):
             token = await hit_challenge(context, host, sitekey, Path(__file__).parent.joinpath(user_data_dirs))
             return token
     except Exception as e:
+        await context.close()
+        traceback.print_exc()
         print(e)
+        return traceback.format_exc()
+