matu
/
Heroku
mirror of https://github.com/coddrago/Heroku
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update security.py

pull/114/head
Who? 2025-03-04 13:21:36 +07:00 committed by GitHub
parent c4d978d73a
commit 5c9cab926c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
hikka/security.py
View File

@ -36,9 +36,8 @@ from herokutl.hints import EntityLike
from herokutl.tl.functions.messages import GetFullChatRequest from herokutl.tl.functions.messages import GetFullChatRequest
from herokutl.tl.types import ChatParticipantAdmin, ChatParticipantCreator, Message from herokutl.tl.types import ChatParticipantAdmin, ChatParticipantCreator, Message
from herokutl.utils import get_display_name from herokutl.utils import get_display_name
import herokutl as tl
from . import main, utils, loader from . import main, utils
from .database import Database from .database import Database
from .tl_cache import CustomTelegramClient from .tl_cache import CustomTelegramClient
from .types import Command from .types import Command
@ -390,9 +389,10 @@ class SecurityManager:
if not (config := self.get_flags(func)): if not (config := self.get_flags(func)):
return False return False
if not user_id and isinstance((await self.client.get_entity(message.peer_id)), tl.types.User): if not user_id:
user_id = message.sender_id user_id = message.sender_id
if not user_id and isinstance((await self.client.get_entity(message.peer_id)), tl.types.Channel):
if not user_id:
user_id = message.peer_id user_id = message.peer_id
is_channel = False is_channel = False