ubot/ubot.py

77 lines
1.9 KiB
Python

from telethon import TelegramClient, events
from telethon.tl.custom import Dialog
from telethon.tl.types import Channel, User, Chat
# Use your own values from my.telegram.org
api_id = 1234567
api_hash = '1234567890abcdef'
client = TelegramClient('ubot', api_id, api_hash)
@client.on(events.NewMessage)
async def edit(event):
msg = event.raw_text
if '.alive' in msg:
rpl = "Userbot alive and well!"
msg = event.text.replace(msg, rpl)
await event.edit(msg)
if '.shg' in msg:
rpl = f"¯\_(ツ)_/¯"
msg = event.text.replace('.shg', rpl)
await event.edit(msg)
if '.stats' in msg:
pms = bots = 0
groups = gadmin = gcreator = 0
chans = cadmin = ccreator = 0
unread = mentions = 0
dialog: Dialog
async for dialog in event.client.iter_dialogs():
entity = dialog.entity
if isinstance(entity, Channel):
if entity.broadcast:
chans += 1
if entity.creator or entity.admin_rights:
cadmin += 1
if entity.creator:
ccreator += 1
elif entity.megagroup:
groups += 1
if entity.creator or entity.admin_rights:
gadmin += 1
if entity.creator:
gcreator += 1
elif isinstance(entity, Chat):
groups += 1
if entity.creator or entity.admin_rights:
gadmin += 1
if entity.creator:
gcreator += 1
elif isinstance(entity, User):
pms += 1
if entity.bot:
bots +=1
mentions += dialog.unread_mentions_count
unread += dialog.unread_count
rpl = f'Private chats: {pms} ({pms - bots} users / {bots} bots)\n'
rpl += f'Groups: {groups} ({gadmin} admin / {gcreator} creator)\n'
rpl += f'Channels: {chans} ({cadmin} admin / {ccreator} creator)\n'
rpl += f'Unread: {unread} ({mentions} mentions)\n'
await event.edit(rpl)
if '.fire' in msg:
rpl = 'IMA FIRING MAH LAZER'
msg = event.text.replace(msg, rpl)
await event.edit(msg)
client.start()
client.run_until_disconnected()