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 .base import TelegramObject
from ...utils import helper
if TYPE_CHECKING: # pragma: no cover
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"""
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:
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