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:23:02 +03:00
parent 6ec16c80b8
commit e25f9deebd
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -67,9 +67,11 @@ class ChatMember(TelegramObject):
can_add_web_page_previews: Optional[bool] = None
"""Restricted only. True, if the user is allowed to add web page previews to their messages"""
@property
def is_chat_admin(self) -> bool:
return self.status in {ChatMemberStatus.CREATOR, ChatMemberStatus.ADMINISTRATOR}
@property
def is_chat_member(self) -> bool:
return self.status not in {ChatMemberStatus.LEFT, ChatMemberStatus.KICKED}

4
tests/test_api/test_types/test_chat_member.py
View file

@ -16,7 +16,7 @@ class TestChatMember:
)
def test_is_chat_admin(self, status: str, result: bool):
chat_member = ChatMember(user=user, status=status)
assert chat_member.is_chat_admin() == result
assert chat_member.is_chat_admin == result
@pytest.mark.parametrize(
"status,result",
@ -31,4 +31,4 @@ class TestChatMember:
)
def test_is_chat_member(self, status: str, result: bool):
chat_member = ChatMember(user=user, status=status)
assert chat_member.is_chat_member() == result
assert chat_member.is_chat_member == result