mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-12 10:11:52 +00:00
Implemented auth widget object and auth data validator.
This commit is contained in:
parent
6ee38bcad9
commit
4f2cb40aea
4 changed files with 75 additions and 0 deletions
|
|
@ -7,6 +7,7 @@ import aiohttp
|
||||||
from . import api
|
from . import api
|
||||||
from ..types import ParseMode, base
|
from ..types import ParseMode, base
|
||||||
from ..utils import json
|
from ..utils import json
|
||||||
|
from ..utils.auth_widget import check_token
|
||||||
|
|
||||||
|
|
||||||
class BaseBot:
|
class BaseBot:
|
||||||
|
|
@ -248,3 +249,6 @@ class BaseBot:
|
||||||
@parse_mode.deleter
|
@parse_mode.deleter
|
||||||
def parse_mode(self):
|
def parse_mode(self):
|
||||||
self.parse_mode = None
|
self.parse_mode = None
|
||||||
|
|
||||||
|
def check_auth_widget(self, data):
|
||||||
|
return check_token(data, self.__token)
|
||||||
|
|
|
||||||
|
|
@ -2,6 +2,7 @@ from . import base
|
||||||
from . import fields
|
from . import fields
|
||||||
from .animation import Animation
|
from .animation import Animation
|
||||||
from .audio import Audio
|
from .audio import Audio
|
||||||
|
from .auth_widget_data import AuthWidgetData
|
||||||
from .callback_game import CallbackGame
|
from .callback_game import CallbackGame
|
||||||
from .callback_query import CallbackQuery
|
from .callback_query import CallbackQuery
|
||||||
from .chat import Chat, ChatActions, ChatType
|
from .chat import Chat, ChatActions, ChatType
|
||||||
|
|
@ -56,6 +57,7 @@ __all__ = (
|
||||||
'AllowedUpdates',
|
'AllowedUpdates',
|
||||||
'Animation',
|
'Animation',
|
||||||
'Audio',
|
'Audio',
|
||||||
|
'AuthWidgetData',
|
||||||
'CallbackGame',
|
'CallbackGame',
|
||||||
'CallbackQuery',
|
'CallbackQuery',
|
||||||
'Chat',
|
'Chat',
|
||||||
|
|
|
||||||
44
aiogram/types/auth_widget_data.py
Normal file
44
aiogram/types/auth_widget_data.py
Normal file
|
|
@ -0,0 +1,44 @@
|
||||||
|
from aiohttp import web
|
||||||
|
|
||||||
|
from . import base
|
||||||
|
from . import fields
|
||||||
|
|
||||||
|
|
||||||
|
class AuthWidgetData(base.TelegramObject):
|
||||||
|
id: base.Integer = fields.Field()
|
||||||
|
first_name: base.String = fields.Field()
|
||||||
|
last_name: base.String = fields.Field()
|
||||||
|
username: base.String = fields.Field()
|
||||||
|
photo_url: base.String = fields.Field()
|
||||||
|
auth_date: base.String = fields.DateTimeField()
|
||||||
|
hash: base.String = fields.Field()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parse(cls, request: web.Request) -> 'AuthWidgetData':
|
||||||
|
"""
|
||||||
|
Parse request as Telegram auth widget data.
|
||||||
|
|
||||||
|
:param request:
|
||||||
|
:return: :obj:`AuthWidgetData`
|
||||||
|
:raise :obj:`aiohttp.web.HTTPBadRequest`
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
query = dict(request.query)
|
||||||
|
query['id'] = int(query['id'])
|
||||||
|
query['auth_date'] = int(query['auth_date'])
|
||||||
|
widget = AuthWidgetData(**query)
|
||||||
|
except (ValueError, KeyError):
|
||||||
|
raise web.HTTPBadRequest(text='Invalid auth data')
|
||||||
|
else:
|
||||||
|
return widget
|
||||||
|
|
||||||
|
def validate(self):
|
||||||
|
return self.bot.check_auth_widget(self.to_python())
|
||||||
|
|
||||||
|
@property
|
||||||
|
def full_name(self):
|
||||||
|
result = self.first_name
|
||||||
|
if self.last_name:
|
||||||
|
result += ' '
|
||||||
|
result += self.last_name
|
||||||
|
return result
|
||||||
25
aiogram/utils/auth_widget.py
Normal file
25
aiogram/utils/auth_widget.py
Normal file
|
|
@ -0,0 +1,25 @@
|
||||||
|
import collections
|
||||||
|
import hashlib
|
||||||
|
import hmac
|
||||||
|
|
||||||
|
|
||||||
|
def check_token(data, token):
|
||||||
|
"""
|
||||||
|
Validate auth token
|
||||||
|
https://core.telegram.org/widgets/login#checking-authorization
|
||||||
|
|
||||||
|
Source: https://gist.github.com/xen/e4bea72487d34caa28c762776cf655a3
|
||||||
|
|
||||||
|
:param data:
|
||||||
|
:param token:
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
secret = hashlib.sha256()
|
||||||
|
secret.update(token.encode('utf-8'))
|
||||||
|
sorted_params = collections.OrderedDict(sorted(data.items()))
|
||||||
|
param_hash = sorted_params.pop('hash', '') or ''
|
||||||
|
msg = "\n".join(["{}={}".format(k, v) for k, v in sorted_params.items()])
|
||||||
|
|
||||||
|
if param_hash == hmac.new(secret.digest(), msg.encode('utf-8'), digestmod=hashlib.sha256).hexdigest():
|
||||||
|
return True
|
||||||
|
return False
|
||||||
Loading…
Add table
Add a link
Reference in a new issue