chore: настроен обход антибот защиты

src/captcha/base.py

@@ -1,7 +1,48 @@
+import asyncio
+import json
+import hashlib
+import time
+from pathlib import Path
+
 from playwright.async_api import Page
 from twocaptcha import TwoCaptcha
 
 
+def cached_persistent(file_name=None):
+
+	def key(*args, **kwargs):
+		return hashlib.sha256(f'{str(args)}-{str(kwargs)}'.encode()).hexdigest()
+
+	def wrapper(fun):
+
+		def clear_cache():
+			cache_file.write_text(json.dumps({}))
+
+		cache_file = Path(file_name)
+		if not cache_file.exists():
+			cache_file.write_text(json.dumps({}))
+
+		async def wrapped(*args, **kwargs):
+			data = json.loads(cache_file.read_text())
+			_key = key(fun.__name__, *args, **kwargs)
+
+			if cached_value := data.get(_key):
+				cached_value, cache_time = cached_value
+				if time.time() - cache_time > 300:
+					cached_value = None
+
+			value = cached_value or await fun(*args, **kwargs)
+
+			data[_key] = [value, time.time()]
+			cache_file.write_text(json.dumps(data))
+			return value
+
+		setattr(wrapped, '_clear_cache', clear_cache)
+		return wrapped
+
+	return wrapper
+
+
 class Captcha:
 	ctype = "captcha"
 
@@ -12,9 +53,22 @@ class Captcha:
 	def get_balance(self) -> float:
 		return self._solver.balance()
 
-	async def solve(self, selector: str, page: Page):
+	async def solve(self, selector: str, page: Page) -> str:
 		raise NotImplementedError()
 
+	async def _run(self, fun, /, *args, **kwargs):
+		resp = await asyncio.to_thread(fun, *args, **kwargs)
+		return resp["code"]
+
+	async def hcaptcha(self, url, site_key, is_invisible):
+		resp = await self._run(self._solver.hcaptcha, site_key, url, invisible=is_invisible)
+		return resp
+
+	@cached_persistent('recaptcha')
+	async def recaptcha(self, url, site_key):
+		resp = await self._run(self._solver.recaptcha, site_key, url)
+		return resp
+
 
 def load_captcha(key, captcha_type) -> Captcha:
 	for cls in Captcha.__subclasses__():