99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
"""
|
|
Show information about the chat or replied user
|
|
"""
|
|
import datetime
|
|
import html
|
|
import time
|
|
|
|
from telethon import events
|
|
from telethon import utils
|
|
from telethon.tl import types, functions
|
|
|
|
|
|
def get_who_string(who, rank=None):
|
|
who_string = html.escape(utils.get_display_name(who))
|
|
if rank is not None:
|
|
who_string += f' <i>"{html.escape(rank)}"</i>'
|
|
if isinstance(who, (types.User, types.Channel)) and who.username:
|
|
who_string += f" <i>(@{who.username})</i>"
|
|
who_string += f", <a href='tg://user?id={who.id}'>#{who.id}</a>"
|
|
return who_string
|
|
|
|
|
|
@borg.on(events.NewMessage(pattern=r"\.who", outgoing=True))
|
|
async def _(event):
|
|
rank = None
|
|
if not event.message.is_reply:
|
|
who = await event.get_chat()
|
|
else:
|
|
msg = await event.message.get_reply_message()
|
|
if msg.forward:
|
|
# FIXME forward privacy memes
|
|
who = await borg.get_entity(
|
|
msg.forward.from_id or msg.forward.channel_id)
|
|
else:
|
|
who = await msg.get_sender()
|
|
ic = await event.get_input_chat()
|
|
if isinstance(ic, types.InputPeerChannel):
|
|
rank = getattr((await borg(functions.channels.GetParticipantRequest(
|
|
ic,
|
|
who
|
|
))).participant, 'rank', None)
|
|
|
|
await event.edit(get_who_string(who, rank), parse_mode='html')
|
|
|
|
|
|
@borg.on(events.NewMessage(pattern=r"\.members", outgoing=True))
|
|
async def _(event):
|
|
last = 0
|
|
index = 0
|
|
members = []
|
|
|
|
it = borg.iter_participants(event.chat_id)
|
|
async for member in it:
|
|
index += 1
|
|
now = time.time()
|
|
if now - last > 0.5:
|
|
last = now
|
|
await event.edit(f'counting member stats ({index / it.total:.2%})…')
|
|
|
|
messages = await borg.get_messages(
|
|
event.chat_id,
|
|
from_user=member,
|
|
limit=0
|
|
)
|
|
members.append((
|
|
messages.total,
|
|
f"{messages.total} - {get_who_string(member)}"
|
|
))
|
|
members = (
|
|
m[1] for m in sorted(members, key=lambda m: m[0], reverse=True)
|
|
)
|
|
|
|
await event.edit("\n".join(members), parse_mode='html')
|
|
|
|
|
|
@borg.on(events.NewMessage(pattern=r"\.active_members", outgoing=True))
|
|
async def _(event):
|
|
members = []
|
|
async for member in borg.iter_participants(event.chat_id):
|
|
messages = await borg.get_messages(
|
|
event.chat_id,
|
|
from_user=member,
|
|
limit=1
|
|
)
|
|
date = (messages[0].date if messages
|
|
else datetime.datetime.fromtimestamp(0, tz=datetime.timezone.utc))
|
|
members.append((
|
|
date,
|
|
f"{date:%Y-%m-%d} - {get_who_string(member)}"
|
|
))
|
|
members = (
|
|
m[1] for m in sorted(members, key=lambda m: m[0], reverse=True)
|
|
)
|
|
|
|
await event.edit("\n".join(members), parse_mode='html')
|