python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added fields to ChatMember

Browse source
This commit is contained in:
lyteloli 2020-05-29 10:07:43 +03:00
parent e69aaa56ca
commit 6ec16c80b8
1 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
aiogram/api/types/chat_member.py
View file

@ -4,6 +4,7 @@ import datetime
from typing import TYPE_CHECKING, Optional, Union from typing import TYPE_CHECKING, Optional, Union
from .base import TelegramObject from .base import TelegramObject
from ...utils import helper
if TYPE_CHECKING: # pragma: no cover if TYPE_CHECKING: # pragma: no cover
from .user import User from .user import User
@ -67,7 +68,21 @@ class ChatMember(TelegramObject):
"""Restricted only. True, if the user is allowed to add web page previews to their messages""" """Restricted only. True, if the user is allowed to add web page previews to their messages"""
def is_chat_admin(self) -> bool: def is_chat_admin(self) -> bool:
return self.status in {"creator", "administrator"} return self.status in {ChatMemberStatus.CREATOR, ChatMemberStatus.ADMINISTRATOR}
def is_chat_member(self) -> bool: def is_chat_member(self) -> bool:
return self.status not in {"left", "kicked"} return self.status not in {ChatMemberStatus.LEFT, ChatMemberStatus.KICKED}
class ChatMemberStatus(helper.Helper):
"""
Chat member status
"""
mode = helper.HelperMode.lowercase
CREATOR = helper.Item() # creator
ADMINISTRATOR = helper.Item() # administrator
MEMBER = helper.Item() # member
RESTRICTED = helper.Item() # restricted
LEFT = helper.Item() # left
KICKED = helper.Item() # kicked