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

twitter/_capsolver/core/enum.py

@@ -0,0 +1,66 @@
+from enum import Enum
+from types import DynamicClassAttribute
+from typing import List
+
+
+class MyEnum(Enum):
+    @classmethod
+    def list(cls) -> List[Enum]:
+        return list(map(lambda c: c, cls))
+
+    @classmethod
+    def list_values(cls) -> List[str]:
+        return list(map(lambda c: c.value, cls))
+
+    @classmethod
+    def list_names(cls) -> List[str]:
+        return list(map(lambda c: c.name, cls))
+
+    @DynamicClassAttribute
+    def name(self) -> str:
+        """
+        The name of the Enum member
+        """
+        return self._name_
+
+    @DynamicClassAttribute
+    def value(self) -> str:
+        """
+        The name of the Enum member
+        """
+        return self._value_
+
+
+class EndpointPostfixEnm(str, MyEnum):
+    """
+    Enum stored URL postfixes for API endpoints
+    """
+
+    GET_BALANCE = "getBalance"
+    CREATE_TASK = "createTask"
+    GET_TASK_RESULT = "getTaskResult"
+    AKAMAI_BMP_INVOKE = "akamaibmp/invoke"
+    AKAMAI_WEB_INVOKE = "akamaiweb/invoke"
+
+
+class FunCaptchaTypeEnm(str, MyEnum):
+    FunCaptchaTask = "FunCaptchaTask"
+    FunCaptchaTaskProxyLess = "FunCaptchaTaskProxyLess"
+
+
+class FunCaptchaClassificationTypeEnm(str, MyEnum):
+    FunCaptchaClassification = "FunCaptchaClassification"
+
+
+class ResponseStatusEnm(str, MyEnum):
+    """
+    Enum store results `status` field variants
+
+    Notes:
+        https://docs.capsolver.com/guide/api-createtask.html
+    """
+
+    Idle = "idle"  # Task created
+    Processing = "processing"  # Task is not ready yet
+    Ready = "ready"  # Task completed, solution object can be found in solution property
+    Failed = "failed"  # Task failed, check the errorDescription to know why failed.

twitter/_capsolver/core/serializer.py

@@ -0,0 +1,85 @@
+from typing import Any, Dict, List, Literal, Optional
+
+from pydantic import Field, BaseModel, conint
+
+from .enum import ResponseStatusEnm
+from .config import APP_ID
+
+"""
+HTTP API Request ser
+"""
+
+
+class PostRequestSer(BaseModel):
+    clientKey: str = Field(..., description="Client account key, can be found in user account")
+    task: dict = Field(None, description="Task object")
+
+
+class TaskSer(BaseModel):
+    type: str = Field(..., description="Task type name", alias="captcha_type")
+
+
+class RequestCreateTaskSer(PostRequestSer):
+    appId: Literal[APP_ID] = APP_ID
+
+
+class RequestGetTaskResultSer(PostRequestSer):
+    taskId: Optional[str] = Field(None, description="ID created by the createTask method")
+
+
+"""
+HTTP API Response ser
+"""
+
+
+class ResponseSer(BaseModel):
+    errorId: int = Field(..., description="Error message: `False` - no error, `True` - with error")
+    # error info
+    errorCode: Optional[str] = Field(None, description="Error code")
+    errorDescription: Optional[str] = Field(None, description="Error description")
+
+
+class CaptchaResponseSer(ResponseSer):
+    taskId: Optional[str] = Field(None, description="Task ID for future use in getTaskResult method.")
+    status: ResponseStatusEnm = Field(ResponseStatusEnm.Processing, description="Task current status")
+    solution: Dict[str, Any] = Field(None, description="Task result data. Different for each type of task.")
+
+    class Config:
+        populate_by_name = True
+
+
+class ControlResponseSer(ResponseSer):
+    balance: Optional[float] = Field(0, description="Account balance value in USD")
+
+
+"""
+Other ser
+"""
+
+
+class CaptchaOptionsSer(BaseModel):
+    api_key: str
+    sleep_time: conint(ge=5) = 5
+
+
+"""
+Captcha tasks ser
+"""
+
+
+class FunCaptchaClassificationOptionsSer(TaskSer):
+    images: List[str] = Field(..., description="Base64-encoded images, do not include 'data:image/***;base64,'")
+    question: str = Field(
+        ...,
+        description="Question name. this param value from API response game_variant field. Exmaple: maze,maze2,flockCompass,3d_rollball_animals",
+    )
+
+
+class FunCaptchaSer(TaskSer):
+    websiteURL: str = Field(..., description="Address of a webpage with Funcaptcha")
+    websitePublicKey: str = Field(..., description="Funcaptcha website key.")
+    funcaptchaApiJSSubdomain: Optional[str] = Field(
+        None,
+        description="A special subdomain of funcaptcha.com, from which the JS captcha widget should be loaded."
+        "Most FunCaptcha installations work from shared domains.",
+    )

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()