tweepy-self 1.6.3__py3-none-any.whl → 1.10.0b1__py3-none-any.whl

twitter/_capsolver/fun_captcha.py

@@ -0,0 +1,260 @@
+from typing import List, Union
+
+from .core.base import BaseCaptcha
+from .core.enum import FunCaptchaTypeEnm, FunCaptchaClassificationTypeEnm
+from .core.serializer import (
+    FunCaptchaSer,
+    CaptchaResponseSer,
+    FunCaptchaClassificationOptionsSer,
+)
+
+
+class FunCaptcha(BaseCaptcha):
+    """
+    The class is used to work with Capsolver FuncaptchaTask methods.
+
+    Args:
+        api_key: Capsolver API key
+        captcha_type: Captcha type name, like ``FunCaptchaTaskProxyLess`` and etc.
+        websiteURL: Address of a webpage with Geetest.
+        websitePublicKey: Funcaptcha website key.
+
+    Examples:
+        >>> with FunCaptcha(api_key="CAI-1324...",
+        ...             captcha_type="FunCaptchaTaskProxyLess",
+        ...             websiteURL="https://api.funcaptcha.com/fc/api/nojs/",
+        ...             websitePublicKey="69A21A01-CC7B-B9C6-0F9A-E7FA06677FFC",
+        ...             funcaptchaApiJSSubdomain="https://api.funcaptcha.com/"
+        ...         ) as instance:
+        >>>     instance.captcha_handler()
+        CaptchaResponseSer(errorId=0,
+                           errorCode=None,
+                           errorDescription=None,
+                           taskId='73bdcd28-6c77-4414-8....',
+                           status=<ResponseStatusEnm.Ready: 'ready'>,
+                           solution={'token': '44795sds...'}
+                          )
+
+        >>> with FunCaptcha(api_key="CAI-1324...",
+        ...             captcha_type="FunCaptchaTaskProxyLess",
+        ...             websiteURL="https://api.funcaptcha.com/fc/api/nojs/",
+        ...             websitePublicKey="69A21A01-CC7B-B9C6-0F9A-E7FA06677FFC",
+        ...             funcaptchaApiJSSubdomain="https://api.funcaptcha.com/"
+        ...         ) as instance:
+        >>>     await instance.aio_captcha_handler()
+        CaptchaResponseSer(errorId=0,
+                           errorCode=None,
+                           errorDescription=None,
+                           taskId='73bdcd28-6c77-4414-8....',
+                           status=<ResponseStatusEnm.Ready: 'ready'>,
+                           solution={'token': '44795sds...'}
+                          )
+
+    Returns:
+        CaptchaResponseSer model with full server response
+
+    Notes:
+        https://docs.capsolver.com/guide/captcha/FunCaptcha.html
+    """
+
+    def __init__(
+        self,
+        captcha_type: Union[FunCaptchaTypeEnm, str],
+        websiteURL: str,
+        websitePublicKey: str,
+        *args,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+
+        if captcha_type in FunCaptchaTypeEnm.list():
+            self.task_params = FunCaptchaSer(**locals()).dict()
+        else:
+            raise ValueError(
+                f"""Invalid `captcha_type` parameter set for `{self.__class__.__name__}`,
+                available - {FunCaptchaTypeEnm.list()}"""
+            )
+        for key in kwargs:
+            self.task_params.update({key: kwargs[key]})
+
+    def captcha_handler(self) -> CaptchaResponseSer:
+        """
+        Sync solving method
+
+        Examples:
+            >>> FunCaptcha(api_key="CAI-BA9XXXXXXXXXXXXX2702E010",
+            ...         captcha_type="FunCaptchaTaskProxyLess",
+            ...         websiteURL="https://api.funcaptcha.com/fc/api/nojs/",
+            ...         websitePublicKey="69A21A01-CC7B-B9C6-0F9A-E7FA06677FFC",
+            ...         funcaptchaApiJSSubdomain="https://api.funcaptcha.com/"
+            ...        ).captcha_handler()
+            CaptchaResponseSer(errorId=0,
+                               errorCode=None,
+                               errorDescription=None,
+                               taskId='73bdcd28-6c77-4414-8....',
+                               status=<ResponseStatusEnm.Ready: 'ready'>,
+                               solution={'token': '44795sds...'}
+                              )
+
+            >>> FunCaptcha(api_key="CAI-BA9XXXXXXXXXXXXX2702E010",
+            ...         captcha_type="FuncaptchaTask",
+            ...         websiteURL="https://api.funcaptcha.com/fc/api/nojs/",
+            ...         websitePublicKey="69A21A01-CC7B-B9C6-0F9A-E7FA06677FFC",
+            ...         funcaptchaApiJSSubdomain="https://api.funcaptcha.com/",
+            ...         proxyType="http",
+            ...         proxyAddress="0.0.0.0",
+            ...         proxyPort=9090,
+            ...        ).captcha_handler()
+            CaptchaResponseSer(errorId=0,
+                               errorCode=None,
+                               errorDescription=None,
+                               taskId='73bdcd28-6c77-4414-8....',
+                               status=<ResponseStatusEnm.Ready: 'ready'>,
+                               solution={'token': '44795sds...'}
+                              )
+
+        Returns:
+            CaptchaResponseSer model with full service response
+
+        Notes:
+            Check class docstring for more info
+        """
+        return self._processing_captcha(create_params=self.task_params)
+
+    async def aio_captcha_handler(self) -> CaptchaResponseSer:
+        """
+        Async method for captcha solving
+
+        Examples:
+            >>> await FunCaptcha(api_key="CAI-1324...",
+            ...         captcha_type="FunCaptchaTaskProxyLess",
+            ...         websiteURL="https://api.funcaptcha.com/fc/api/nojs/",
+            ...         websitePublicKey="69A21A01-CC7B-B9C6-0F9A-E7FA06677FFC",
+            ...         funcaptchaApiJSSubdomain="https://api.funcaptcha.com/"
+            ...        ).aio_captcha_handler()
+            CaptchaResponseSer(errorId=0,
+                               errorCode=None,
+                               errorDescription=None,
+                               taskId='73bdcd28-6c77-4414-8....',
+                               status=<ResponseStatusEnm.Ready: 'ready'>,
+                               solution={'token': '44795sds...'}
+                              )
+
+            >>> await FunCaptcha(api_key="CAI-1324...",
+            ...         captcha_type="FuncaptchaTask",
+            ...         websiteURL="https://api.funcaptcha.com/fc/api/nojs/",
+            ...         websitePublicKey="69A21A01-CC7B-B9C6-0F9A-E7FA06677FFC",
+            ...         funcaptchaApiJSSubdomain="https://api.funcaptcha.com/",
+            ...         proxyType="http",
+            ...         proxyAddress="0.0.0.0",
+            ...         proxyPort=9090,
+            ...        ).aio_captcha_handler()
+            CaptchaResponseSer(errorId=0,
+                               errorCode=None,
+                               errorDescription=None,
+                               taskId='73bdcd28-6c77-4414-8....',
+                               status=<ResponseStatusEnm.Ready: 'ready'>,
+                               solution={'token': '44795sds...'}
+                              )
+
+            >>> with open('some_image.jpeg', 'rb') as img_file:
+            ...    img_data = img_file.read()
+            >>> body = base64.b64encode(img_data).decode("utf-8")
+            >>> await FunCaptcha(api_key="CAI-1324...",
+            ...         captcha_type="FunCaptchaClassification"
+            ...        ).aio_captcha_handler(
+            ...                     image=body,
+            ...                     question="Ask your question")
+            CaptchaResponseSer(errorId=0,
+                               errorCode=None,
+                               errorDescription=None,
+                               taskId='73bdcd28-6c77-4414-8....',
+                               status=<ResponseStatusEnm.Ready: 'ready'>,
+                               solution={'token': '44795sds...'}
+                              )
+
+        Returns:
+            CaptchaResponseSer model with full service response
+
+        Notes:
+            Check class docstring for more info
+        """
+        return await self._aio_processing_captcha(create_params=self.task_params)
+
+
+class FunCaptchaClassification(BaseCaptcha):
+    """
+    The class is used to work with Capsolver FunCaptchaClassification methods.
+
+    Args:
+        api_key: Capsolver API key
+        captcha_type: Captcha type name, like ``FunCaptchaClassification`` and etc.
+        images: Base64 encoded image, can be a screenshot (pass only the hexagonal image, do not pass the rest of the content)
+        question: Question name. this param value from API response game_variant field. Exmaple: maze,maze2,flockCompass,3d_rollball_animals
+
+    Examples:
+        >>> FunCaptchaClassification(api_key="CAI-1324...",
+        ...             captcha_type="FunCaptchaClassification",
+        ...             images=["image payload"],
+        ...             question="maze2",
+        ...         ).captcha_handler()
+        CaptchaResponseSer(errorId=0,
+                           errorCode=None,
+                           errorDescription=None,
+                           taskId='73bdcd28-6c77-4414-8....',
+                           status=<ResponseStatusEnm.Ready: 'ready'>,
+                           solution={"objects": [ 4 ]}
+                          )
+
+    Returns:
+        CaptchaResponseSer model with full server response
+
+    Notes:
+        https://docs.capsolver.com/guide/recognition/FunCaptchaClassification.html
+    """
+
+    def __init__(
+        self,
+        images: List[str],
+        question: str,
+        captcha_type: Union[
+            FunCaptchaClassificationTypeEnm, str
+        ] = FunCaptchaClassificationTypeEnm.FunCaptchaClassification,
+        *args,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+
+        if captcha_type in FunCaptchaClassificationTypeEnm.list():
+            self.task_params = FunCaptchaClassificationOptionsSer(**locals()).dict()
+        else:
+            raise ValueError(
+                f"""Invalid `captcha_type` parameter set for `{self.__class__.__name__}`,
+                available - {FunCaptchaClassificationTypeEnm.list()}"""
+            )
+        for key in kwargs:
+            self.task_params.update({key: kwargs[key]})
+
+    def captcha_handler(self) -> CaptchaResponseSer:
+        """
+        Sync solving method
+
+        Returns:
+            CaptchaResponseSer model with full service response
+
+        Notes:
+            Check class docstring for more info
+        """
+        return self._processing_captcha(create_params=self.task_params)
+
+    async def aio_captcha_handler(self) -> CaptchaResponseSer:
+        """
+        Async method for captcha solving
+
+        Returns:
+            CaptchaResponseSer model with full service response
+
+        Notes:
+            Check class docstring for more info
+        """
+        return await self._aio_processing_captcha(create_params=self.task_params)

twitter/account.py

@@ -1,24 +1,22 @@
 from pathlib import Path
 from typing import Sequence, Iterable
 
-from pydantic import BaseModel, Field
+from pydantic import Field
 import pyotp
 
 from .utils import hidden_value, load_lines, write_lines
 from .enums import AccountStatus
+from .models import User
 
 
-class Account(BaseModel):
+class Account(User):
     # fmt: off
     auth_token:  str | None = Field(default=None, pattern=r"^[a-f0-9]{40}$")
-    ct0:         str | None = None
-    id:          int | None = None
-    name:        str | None = None
-    username:    str | None = None
-    password:    str | None = None
-    email:       str | None = None
-    totp_secret: str | None = None
-    backup_code: str | None = None
+    ct0:         str | None = None  # 160
+    password:    str | None = None  # 128
+    email:       str | None = None  # 254
+    totp_secret: str | None = None  # 16
+    backup_code: str | None = None  # 12
     status: AccountStatus = AccountStatus.UNKNOWN
     # fmt: on
 
@@ -38,12 +36,18 @@ class Account(BaseModel):
     def hidden_backup_code(self) -> str | None:
         return hidden_value(self.backup_code) if self.backup_code else None
 
-    def __repr__(self):
-        return f"{self.__class__.__name__}(auth_token={self.hidden_auth_token}, username={self.username})"
-
     def __str__(self):
         return self.hidden_auth_token
 
+    def __repr__(self):
+        return f"{self.__class__.__name__}(id={self.id}, username={self.username}, auth_token={self.hidden_auth_token})"
+
+    def update(self, **data: dict):
+        update = self.dict()
+        update.update(data)
+        for k, v in self.validate(update).dict(exclude_defaults=True).items():
+            setattr(self, k, v)
+
     def get_totp_code(self) -> str | None:
         if not self.totp_secret:
             raise ValueError("No key2fa")

twitter/base/__init__.py

@@ -1,7 +1,7 @@
-from .client import BaseClient
+from .client import BaseHTTPClient
 from .session import BaseAsyncSession
 
 __all__ = [
-    "BaseClient",
+    "BaseHTTPClient",
     "BaseAsyncSession",
 ]

twitter/base/client.py

@@ -1,7 +1,7 @@
 from .session import BaseAsyncSession
 
 
-class BaseClient:
+class BaseHTTPClient:
     _DEFAULT_HEADERS = None
 
     def __init__(self, **session_kwargs):
@@ -14,7 +14,7 @@ class BaseClient:
         return self
 
     async def __aexit__(self, *args):
-        self.close()
+        await self.close()
 
-    def close(self):
-        self._session.close()
+    async def close(self):
+        await self._session.close()