ClanStat/app/utils.py

28 lines
884 B
Python

# Импорт сторонних модулей
from pyrogram.types import Message
from pyrogram.enums import MessageEntityType
from . import client
async def get_telegram_id(username: str) -> int:
"""Возвращает Telegram ID"""
telegram_user = await client.get_users(username)
return telegram_user.id
async def get_player_data(message: Message, idx: int) -> tuple():
"""..."""
if message.entities[idx].type == MessageEntityType.TEXT_LINK:
player_url = message.entities[idx].url
player_telegram_id = await get_telegram_id(player_url.replace('http://t.me/', ''))
elif message.entities[idx].type == MessageEntityType.TEXT_MENTION:
player_url = None
player_telegram_id = message.entities[idx].user.id
else:
player_url = None
player_telegram_id = None
return (player_url, player_telegram_id)