python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-11 01:54:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Provided customisation of allowed IPs list. // Webhook

Browse source
This commit is contained in:
Alex Root Junior 2017-11-16 18:36:49 +02:00
parent 600148e6d2
commit ca8d3cad15
1 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
aiogram/dispatcher/webhook.py
View file

@ -28,7 +28,8 @@ WEBHOOK_REQUEST = 'WEBHOOK_REQUEST'
TELEGRAM_IP_LOWER = ipaddress.IPv4Address('149.154.167.197') TELEGRAM_IP_LOWER = ipaddress.IPv4Address('149.154.167.197')
TELEGRAM_IP_UPPER = ipaddress.IPv4Address('149.154.167.233') TELEGRAM_IP_UPPER = ipaddress.IPv4Address('149.154.167.233')
LOCALHOST_IP = ipaddress.IPv4Address('127.0.0.1')
allowed_ips = set()
def _check_ip(ip: str) -> bool: def _check_ip(ip: str) -> bool:
@ -39,7 +40,21 @@ def _check_ip(ip: str) -> bool:
:return: :return:
""" """
address = ipaddress.IPv4Address(ip) address = ipaddress.IPv4Address(ip)
return TELEGRAM_IP_LOWER <= address <= TELEGRAM_IP_UPPER or address == LOCALHOST_IP return address in allowed_ips
def allow_ip(*ips: str):
"""
Allow ip address.
:param ips:
:return:
"""
allowed_ips.update(ipaddress.IPv4Address(ip) for ip in ips)
# Allow access from Telegram servers
allow_ip(*(ip for ip in range(int(TELEGRAM_IP_LOWER), int(TELEGRAM_IP_UPPER) + 1)))
class WebhookRequestHandler(web.View): class WebhookRequestHandler(web.View):