-
Notifications
You must be signed in to change notification settings - Fork 0
/
filters.py
44 lines (33 loc) · 1.33 KB
/
filters.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
from aiogram import types
from aiogram.dispatcher.filters import BoundFilter
import config
class IsOwnerFilter(BoundFilter):
"""
Custom filter "is_owner".
"""
key = "is_owner"
def __init__(self, is_owner):
self.is_owner = is_owner
async def check(self, message: types.Message):
return message.from_user.id == config.BOT_OWNER
class IsAdminFilter(BoundFilter):
"""
Filter that checks for admin rights existence
"""
key = "is_admin"
def __init__(self, is_admin: bool):
self.is_admin = is_admin
async def check(self, message: types.Message):
member = await message.bot.get_chat_member(message.chat.id, message.from_user.id)
return member.is_chat_admin() == self.is_admin
class MemberCanRestrictFilter(BoundFilter):
"""
Filter that checks member ability for restricting
"""
key = 'member_can_restrict'
def __init__(self, member_can_restrict: bool):
self.member_can_restrict = member_can_restrict
async def check(self, message: types.Message):
member = await message.bot.get_chat_member(message.chat.id, message.from_user.id)
# I don't know why, but telegram thinks, if member is chat creator, he cant restrict member
return (member.is_chat_creator() or member.can_restrict_members) == self.member_can_restrict