apify 1.7.1b1__py3-none-any.whl → 2.2.1b1__py3-none-any.whl

apify/proxy_configuration.py

@@ -1,365 +0,0 @@
-from __future__ import annotations
-
-import inspect
-import ipaddress
-import re
-from typing import TYPE_CHECKING, Any, Awaitable, Callable, Pattern, TypedDict
-from urllib.parse import urljoin, urlparse
-
-import httpx
-from apify_shared.consts import ApifyEnvVars
-from apify_shared.utils import ignore_docs
-
-from apify.config import Configuration
-from apify.log import logger
-
-if TYPE_CHECKING:
-    from apify_client import ApifyClientAsync
-    from typing_extensions import NotRequired
-
-APIFY_PROXY_VALUE_REGEX = re.compile(r'^[\w._~]+$')
-COUNTRY_CODE_REGEX = re.compile(r'^[A-Z]{2}$')
-SESSION_ID_MAX_LENGTH = 50
-
-
-def is_url(url: str) -> bool:
-    """Check if the given string is a valid URL."""
-    try:
-        parsed_url = urlparse(urljoin(url, '/'))
-        has_all_parts = all([parsed_url.scheme, parsed_url.netloc, parsed_url.path])
-        is_domain = '.' in parsed_url.netloc
-        is_localhost = parsed_url.netloc == 'localhost'
-        try:
-            ipaddress.ip_address(parsed_url.netloc)
-            is_ip_address = True
-        except Exception:
-            is_ip_address = False
-
-        return has_all_parts and any([is_domain, is_localhost, is_ip_address])
-    except Exception:
-        return False
-
-
-def _check(
-    value: Any,
-    *,
-    label: str | None,
-    pattern: Pattern | None = None,
-    min_length: int | None = None,
-    max_length: int | None = None,
-) -> None:
-    error_str = f'Value {value}'
-    if label:
-        error_str += f' of argument {label}'
-
-    if min_length and len(value) < min_length:
-        raise ValueError(f'{error_str} is shorter than minimum allowed length {min_length}')
-
-    if max_length and len(value) > max_length:
-        raise ValueError(f'{error_str} is longer than maximum allowed length {max_length}')
-
-    if pattern and not re.fullmatch(pattern, value):
-        raise ValueError(f'{error_str} does not match pattern {pattern.pattern!r}')
-
-
-class ProxyInfo(TypedDict):
-    """Provides information about a proxy connection that is used for requests."""
-
-    url: str
-    """The URL of the proxy."""
-
-    hostname: str
-    """The hostname of the proxy."""
-
-    port: int
-    """The proxy port."""
-
-    username: NotRequired[str]
-    """The username for the proxy."""
-
-    password: str
-    """The password for the proxy."""
-
-    groups: NotRequired[list[str]]
-    """An array of proxy groups to be used by the [Apify Proxy](https://docs.apify.com/proxy).
-    If not provided, the proxy will select the groups automatically.
-    """
-
-    country_code: NotRequired[str]
-    """If set and relevant proxies are available in your Apify account, all proxied requests will
-    use IP addresses that are geolocated to the specified country. For example `GB` for IPs
-    from Great Britain. Note that online services often have their own rules for handling
-    geolocation and thus the country selection is a best attempt at geolocation, rather than
-    a guaranteed hit. This parameter is optional, by default, each proxied request is assigned
-    an IP address from a random country. The country code needs to be a two letter ISO country code.
-    See the [full list of available country codes](https://en.wikipedia.org/wiki/ISO_3166-1_alpha-2#Officially_assigned_code_elements).
-    This parameter is optional, by default, the proxy uses all available proxy servers from all countries.
-    """
-
-    session_id: NotRequired[str]
-    """The identifier of the used proxy session, if used. Using the same session ID guarantees getting the same proxy URL."""
-
-
-class ProxyConfiguration:
-    """Configures a connection to a proxy server with the provided options.
-
-    Proxy servers are used to prevent target websites from blocking your crawlers based on IP address rate limits or blacklists.
-    The default servers used by this class are managed by [Apify Proxy](https://docs.apify.com/proxy).
-    To be able to use Apify Proxy, you need an Apify account and access to the selected proxies. If you provide no configuration option,
-    the proxies will be managed automatically using a smart algorithm.
-
-    If you want to use your own proxies, use the `proxy_urls` or `new_url_function` constructor options.
-    Your list of proxy URLs will be rotated by the configuration, if this option is provided.
-    """
-
-    is_man_in_the_middle = False
-
-    _next_custom_url_index = 0
-    _proxy_urls: list[str]
-    _used_proxy_urls: dict[str, str]
-    _new_url_function: Callable[[str | None], str] | Callable[[str | None], Awaitable[str]] | None = None
-    _groups: list[str]
-    _country_code: str | None = None
-    _password: str | None = None
-    _hostname: str
-    _port: int
-    _uses_apify_proxy: bool | None = None
-    _actor_config: Configuration
-    _apify_client: ApifyClientAsync | None = None
-
-    @ignore_docs
-    def __init__(
-        self: ProxyConfiguration,
-        *,
-        password: str | None = None,
-        groups: list[str] | None = None,
-        country_code: str | None = None,
-        proxy_urls: list[str] | None = None,
-        new_url_function: Callable[[str | None], str] | Callable[[str | None], Awaitable[str]] | None = None,
-        _actor_config: Configuration | None = None,
-        _apify_client: ApifyClientAsync | None = None,
-    ) -> None:
-        """Create a ProxyConfiguration instance. It is highly recommended to use `Actor.create_proxy_configuration()` instead of this.
-
-        Args:
-            password (str, optional): Password for the Apify Proxy. If not provided, will use os.environ['APIFY_PROXY_PASSWORD'], if available.
-            groups (list of str, optional): Proxy groups which the Apify Proxy should use, if provided.
-            country_code (str, optional): Country which the Apify Proxy should use, if provided.
-            proxy_urls (list of str, optional): Custom proxy server URLs which should be rotated through.
-            new_url_function (Callable, optional): Function which returns a custom proxy URL to be used.
-        """
-        if groups:
-            groups = [str(group) for group in groups]
-            for group in groups:
-                _check(group, label='groups', pattern=APIFY_PROXY_VALUE_REGEX)
-        if country_code:
-            country_code = str(country_code)
-            _check(country_code, label='country_code', pattern=COUNTRY_CODE_REGEX)
-        if proxy_urls:
-            for i, url in enumerate(proxy_urls):
-                if not is_url(url):
-                    raise ValueError(f'proxy_urls[{i}] ("{url}") is not a valid URL')
-
-        # Validation
-        if proxy_urls and new_url_function:
-            raise ValueError('Cannot combine custom proxies in "proxy_urls" with custom generating function in "new_url_function".')
-
-        if (proxy_urls or new_url_function) and (groups or country_code):
-            raise ValueError(
-                'Cannot combine custom proxies with Apify Proxy!'
-                ' It is not allowed to set "proxy_urls" or "new_url_function" combined with'
-                ' "groups" or "country_code".'
-            )
-
-        # mypy has a bug with narrowing types for filter (https://github.com/python/mypy/issues/12682)
-        if proxy_urls and next(filter(lambda url: 'apify.com' in url, proxy_urls), None):  # type: ignore
-            logger.warning(
-                'Some Apify proxy features may work incorrectly. Please consider setting up Apify properties instead of `proxy_urls`.\n'
-                'See https://sdk.apify.com/docs/guides/proxy-management#apify-proxy-configuration'
-            )
-
-        self._actor_config = _actor_config or Configuration._get_default_instance()
-        self._apify_client = _apify_client
-
-        self._hostname = self._actor_config.proxy_hostname
-        self._port = self._actor_config.proxy_port
-        self._password = password or self._actor_config.proxy_password
-
-        self._proxy_urls = list(proxy_urls) if proxy_urls else []
-        self._used_proxy_urls = {}
-        self._new_url_function = new_url_function
-        self._groups = list(groups) if groups else []
-        self._country_code = country_code
-        self._uses_apify_proxy = not (proxy_urls or new_url_function)
-
-    async def initialize(self: ProxyConfiguration) -> None:
-        """Load the Apify Proxy password if the API token is provided and check access to Apify Proxy and provided proxy groups.
-
-        Only called if Apify Proxy configuration is used.
-        Also checks if country has access to Apify Proxy groups if the country code is provided.
-
-        You should use the Actor.create_proxy_configuration function
-        to create a pre-initialized `ProxyConfiguration` instance instead of calling this manually.
-        """
-        if self._uses_apify_proxy:
-            await self._maybe_fetch_password()
-            await self._check_access()
-
-    async def new_url(self: ProxyConfiguration, session_id: int | str | None = None) -> str:
-        """Return a new proxy URL based on provided configuration options and the `sessionId` parameter.
-
-        Args:
-            session_id (int or str, optional): Represents the identifier of a proxy session (https://docs.apify.com/proxy#sessions).
-            All the HTTP requests going through the proxy with the same session identifier
-            will use the same target proxy server (i.e. the same IP address).
-            The identifier must not be longer than 50 characters and include only the following: `0-9`, `a-z`, `A-Z`, `"."`, `"_"` and `"~"`.
-
-        Returns:
-            str: A string with a proxy URL, including authentication credentials and port number.
-                 For example, `http://bob:password123@proxy.example.com:8000`
-        """
-        if session_id is not None:
-            session_id = f'{session_id}'
-            _check(session_id, label='session_id', max_length=SESSION_ID_MAX_LENGTH, pattern=APIFY_PROXY_VALUE_REGEX)
-
-        if self._new_url_function:
-            try:
-                res = self._new_url_function(session_id)
-                if inspect.isawaitable(res):
-                    res = await res
-                return str(res)
-            except Exception as exc:
-                raise ValueError('The provided "new_url_function" did not return a valid URL') from exc
-
-        if self._proxy_urls:
-            if not session_id:
-                index = self._next_custom_url_index
-                self._next_custom_url_index = (self._next_custom_url_index + 1) % len(self._proxy_urls)
-                return self._proxy_urls[index]
-
-            if session_id not in self._used_proxy_urls:
-                index = self._next_custom_url_index
-                self._next_custom_url_index = (self._next_custom_url_index + 1) % len(self._proxy_urls)
-                self._used_proxy_urls[session_id] = self._proxy_urls[index]
-
-            return self._used_proxy_urls[session_id]
-
-        username = self._get_username(session_id)
-
-        return f'http://{username}:{self._password}@{self._hostname}:{self._port}'
-
-    async def new_proxy_info(self: ProxyConfiguration, session_id: int | str | None = None) -> ProxyInfo:
-        """Create a new ProxyInfo object.
-
-        Use it if you want to work with a rich representation of a proxy URL.
-        If you need the URL string only, use `ProxyConfiguration.new_url`.
-
-        Args:
-            session_id (int or str, optional): Represents the identifier of a proxy session (https://docs.apify.com/proxy#sessions).
-            All the HTTP requests going through the proxy with the same session identifier
-            will use the same target proxy server (i.e. the same IP address).
-            The identifier must not be longer than 50 characters and include only the following: `0-9`, `a-z`, `A-Z`, `"."`, `"_"` and `"~"`.
-
-        Returns:
-            ProxyInfo: Dictionary that represents information about the proxy and its configuration.
-        """
-        if session_id is not None:
-            session_id = f'{session_id}'
-            _check(session_id, label='session_id', max_length=SESSION_ID_MAX_LENGTH, pattern=APIFY_PROXY_VALUE_REGEX)
-
-        url = await self.new_url(session_id)
-        res: ProxyInfo
-        if self._uses_apify_proxy:
-            res = {
-                'url': url,
-                'hostname': self._hostname,
-                'port': self._port,
-                'username': self._get_username(session_id),
-                'password': self._password or '',
-                'groups': self._groups,
-            }
-            if self._country_code:
-                res['country_code'] = self._country_code
-            if session_id is not None:
-                res['session_id'] = session_id
-            return res
-
-        parsed_url = urlparse(url)
-        assert parsed_url.hostname is not None  # noqa: S101
-        assert parsed_url.port is not None  # noqa: S101
-        res = {
-            'url': url,
-            'hostname': parsed_url.hostname,
-            'port': parsed_url.port,
-            'password': parsed_url.password or '',
-        }
-        if parsed_url.username:
-            res['username'] = parsed_url.username
-        return res
-
-    async def _maybe_fetch_password(self: ProxyConfiguration) -> None:
-        token = self._actor_config.token
-
-        if token and self._apify_client:
-            user_info = await self._apify_client.user().get()
-            if user_info:
-                password = user_info['proxy']['password']
-
-                if self._password:
-                    if self._password != password:
-                        logger.warning(
-                            'The Apify Proxy password you provided belongs to'
-                            ' a different user than the Apify token you are using. Are you sure this is correct?'
-                        )
-                else:
-                    self._password = password
-
-        if not self._password:
-            raise ValueError(
-                'Apify Proxy password must be provided using the "password" constructor argument'
-                f' or the "{ApifyEnvVars.PROXY_PASSWORD}" environment variable.'
-                f' If you add the "{ApifyEnvVars.TOKEN}" environment variable, the password will be automatically inferred.'
-            )
-
-    async def _check_access(self: ProxyConfiguration) -> None:
-        proxy_status_url = f'{self._actor_config.proxy_status_url}/?format=json'
-
-        status = None
-        async with httpx.AsyncClient(proxies=await self.new_url()) as client:
-            for _ in range(2):
-                try:
-                    response = await client.get(proxy_status_url)
-                    status = response.json()
-                    break
-                except Exception:  # noqa: S110
-                    # retry on connection errors
-                    pass
-
-        if status:
-            if not status['connected']:
-                raise ConnectionError(status['connectionError'])
-
-            self.is_man_in_the_middle = status['isManInTheMiddle']
-        else:
-            logger.warning(
-                'Apify Proxy access check timed out. Watch out for errors with status code 407. '
-                "If you see some, it most likely means you don't have access to either all or some of the proxies you're trying to use."
-            )
-
-    def _get_username(self: ProxyConfiguration, session_id: int | str | None = None) -> str:
-        if session_id is not None:
-            session_id = f'{session_id}'
-
-        parts: list[str] = []
-
-        if self._groups:
-            parts.append(f'groups-{"+".join(self._groups)}')
-        if session_id is not None:
-            parts.append(f'session-{session_id}')
-        if self._country_code:
-            parts.append(f'country-{self._country_code}')
-
-        if not parts:
-            return 'auto'
-
-        return ','.join(parts)

apify/storages/base_storage.py

@@ -1,181 +0,0 @@
-from __future__ import annotations
-
-import asyncio
-from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Generic, TypeVar, cast
-
-from apify_shared.utils import ignore_docs
-
-from apify._memory_storage import MemoryStorageClient
-from apify._memory_storage.resource_clients import BaseResourceClient, BaseResourceCollectionClient
-from apify.config import Configuration
-from apify.storages.storage_client_manager import StorageClientManager
-
-if TYPE_CHECKING:
-    from apify_client import ApifyClientAsync
-
-BaseResourceClientType = TypeVar('BaseResourceClientType', bound=BaseResourceClient)
-BaseResourceCollectionClientType = TypeVar('BaseResourceCollectionClientType', bound=BaseResourceCollectionClient)
-
-
-@ignore_docs
-class BaseStorage(ABC, Generic[BaseResourceClientType, BaseResourceCollectionClientType]):
-    """A class for managing storages."""
-
-    _id: str
-    _name: str | None
-    _storage_client: ApifyClientAsync | MemoryStorageClient
-    _config: Configuration
-
-    _cache_by_id: dict | None = None
-    _cache_by_name: dict | None = None
-    _storage_creating_lock: asyncio.Lock | None = None
-
-    def __init__(
-        self: BaseStorage,
-        id: str,  # noqa: A002
-        name: str | None,
-        client: ApifyClientAsync | MemoryStorageClient,
-        config: Configuration,
-    ) -> None:
-        """Initialize the storage.
-
-        Do not use this method directly, but use `Actor.open_<STORAGE>()` instead.
-
-        Args:
-            id (str): The storage id
-            name (str, optional): The storage name
-            client (ApifyClientAsync or MemoryStorageClient): The storage client
-            config (Configuration): The configuration
-        """
-        self._id = id
-        self._name = name
-        self._storage_client = client
-        self._config = config
-
-    @classmethod
-    @abstractmethod
-    def _get_human_friendly_label(cls: type[BaseStorage]) -> str:
-        raise NotImplementedError('You must override this method in the subclass!')
-
-    @classmethod
-    @abstractmethod
-    def _get_default_id(cls: type[BaseStorage], config: Configuration) -> str:
-        raise NotImplementedError('You must override this method in the subclass!')
-
-    @classmethod
-    @abstractmethod
-    def _get_single_storage_client(
-        cls: type[BaseStorage],
-        id: str,  # noqa: A002
-        client: ApifyClientAsync | MemoryStorageClient,
-    ) -> BaseResourceClientType:
-        raise NotImplementedError('You must override this method in the subclass!')
-
-    @classmethod
-    @abstractmethod
-    def _get_storage_collection_client(
-        cls: type[BaseStorage],
-        client: ApifyClientAsync | MemoryStorageClient,
-    ) -> BaseResourceCollectionClientType:
-        raise NotImplementedError('You must override this method in the subclass!')
-
-    @classmethod
-    def _ensure_class_initialized(cls: type[BaseStorage]) -> None:
-        if cls._cache_by_id is None:
-            cls._cache_by_id = {}
-        if cls._cache_by_name is None:
-            cls._cache_by_name = {}
-        if cls._storage_creating_lock is None:
-            cls._storage_creating_lock = asyncio.Lock()
-
-    @classmethod
-    @abstractmethod
-    async def open(
-        cls: type[BaseStorage],
-        *,
-        id: str | None = None,  # noqa: A002
-        name: str | None = None,
-        force_cloud: bool = False,
-        config: Configuration | None = None,
-    ) -> BaseStorage:
-        """Open a storage, or return a cached storage object if it was opened before.
-
-        Opens a storage with the given ID or name.
-        Returns the cached storage object if the storage was opened before.
-
-        Args:
-            id (str, optional): ID of the storage to be opened.
-                If neither `id` nor `name` are provided, the method returns the default storage associated with the actor run.
-                If the storage with the given ID does not exist, it raises an error.
-            name (str, optional): Name of the storage to be opened.
-                If neither `id` nor `name` are provided, the method returns the default storage associated with the actor run.
-                If the storage with the given name does not exist, it is created.
-            force_cloud (bool, optional): If set to True, it will open a storage on the Apify Platform even when running the actor locally.
-                Defaults to False.
-            config (Configuration, optional): A `Configuration` instance, uses global configuration if omitted.
-
-        Returns:
-            An instance of the storage.
-        """
-        cls._ensure_class_initialized()
-        assert cls._cache_by_id is not None  # noqa: S101
-        assert cls._cache_by_name is not None  # noqa: S101
-        assert not (id and name)  # noqa: S101
-
-        used_config = config or Configuration.get_global_configuration()
-        used_client = StorageClientManager.get_storage_client(force_cloud=force_cloud)
-
-        is_default_storage_on_local = False
-        # Fetch default ID if no ID or name was passed
-        if not id and not name:
-            if isinstance(used_client, MemoryStorageClient):
-                is_default_storage_on_local = True
-            id = cls._get_default_id(used_config)  # noqa: A001
-
-        # Try to get the storage instance from cache
-        cached_storage = None
-        if id:
-            cached_storage = cls._cache_by_id.get(id)
-        elif name:
-            cached_storage = cls._cache_by_name.get(name)
-
-        if cached_storage is not None:
-            # This cast is needed since MyPy doesn't understand very well that Self and Storage are the same
-            return cast(BaseStorage, cached_storage)
-
-        # Purge default storages if configured
-        if used_config.purge_on_start and isinstance(used_client, MemoryStorageClient):
-            await used_client._purge_on_start()
-
-        assert cls._storage_creating_lock is not None  # noqa: S101
-        async with cls._storage_creating_lock:
-            # Create the storage
-            if id and not is_default_storage_on_local:
-                single_storage_client = cls._get_single_storage_client(id, used_client)
-                storage_info = await single_storage_client.get()
-                if not storage_info:
-                    storage_label = cls._get_human_friendly_label()
-                    raise RuntimeError(f'{storage_label} with id "{id}" does not exist!')
-            elif is_default_storage_on_local:
-                storage_collection_client = cls._get_storage_collection_client(used_client)
-                storage_info = await storage_collection_client.get_or_create(name=name, _id=id)
-            else:
-                storage_collection_client = cls._get_storage_collection_client(used_client)
-                storage_info = await storage_collection_client.get_or_create(name=name)
-
-            storage = cls(storage_info['id'], storage_info.get('name'), used_client, used_config)
-
-            # Cache by id and name
-            cls._cache_by_id[storage._id] = storage
-            if storage._name is not None:
-                cls._cache_by_name[storage._name] = storage
-
-        return storage
-
-    def _remove_from_cache(self: BaseStorage) -> None:
-        if self.__class__._cache_by_id is not None:
-            del self.__class__._cache_by_id[self._id]
-
-        if self._name and self.__class__._cache_by_name is not None:
-            del self.__class__._cache_by_name[self._name]