2
0
mirror of https://github.com/Nick80835/microbot synced 2025-09-01 15:05:48 +00:00

add support for NSFW toggles and admin-level commands

This commit is contained in:
Nick80835
2020-06-09 21:05:53 -04:00
parent f352ad0e32
commit 252fb65aa9
3 changed files with 33 additions and 2 deletions

View File

@@ -3,7 +3,7 @@
import asyncio
from re import escape, search
from telethon import events
from telethon import events, types
class CommandHandler():
@@ -29,6 +29,9 @@ class CommandHandler():
if value["sudo"] and str(event.from_id) not in self.settings.get_config("owner_id").split(","):
print(f"Attempted sudo command ({event.text}) from ID {event.from_id}")
continue
elif value["admin"] and str(event.from_id) not in self.settings.get_config("owner_id").split(",") and not await self.check_admin(event):
print(f"Attempted admin command ({event.text}) from ID {event.from_id}")
continue
event.pattern_match = pattern_match
event.args = pattern_match.groups()[-1]
@@ -122,3 +125,10 @@ class CommandHandler():
return await coro
except:
return
async def check_admin(self, event):
async for user in event.client.iter_participants(event.chat, limit=10000, filter=types.ChannelParticipantsAdmins):
if user.id == event.from_id:
return True
return False