Source code for aiogram.webhook.security

from collections.abc import Sequence
from ipaddress import IPv4Address, IPv4Network

DEFAULT_TELEGRAM_NETWORKS = [
    IPv4Network("149.154.160.0/20"),
    IPv4Network("91.108.4.0/22"),
]




class IPFilter:


    def __init__(self, ips: Sequence[str | IPv4Network | IPv4Address] | None = None):
        self._allowed_ips: set[IPv4Address] = set()

        if ips:
            self.allow(*ips)


    def allow(self, *ips: str | IPv4Network | IPv4Address) -> None:
        for ip in ips:
            self.allow_ip(ip)

    def allow_ip(self, ip: str | IPv4Network | IPv4Address) -> None:
        if isinstance(ip, str):
            ip = IPv4Network(ip) if "/" in ip else IPv4Address(ip)
        if isinstance(ip, IPv4Address):
            self._allowed_ips.add(ip)
        elif isinstance(ip, IPv4Network):
            self._allowed_ips.update(ip.hosts())
        else:
            msg = f"Invalid type of ipaddress: {type(ip)} ('{ip}')"
            raise ValueError(msg)

    @classmethod
    def default(cls) -> "IPFilter":
        return cls(DEFAULT_TELEGRAM_NETWORKS)

    def check(self, ip: str | IPv4Address) -> bool:
        if not isinstance(ip, IPv4Address):
            ip = IPv4Address(ip)
        return ip in self._allowed_ips

    def __contains__(self, item: str | IPv4Address) -> bool:
        return self.check(item)