mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-16 22:34:19 +02:00
218 lines
9.1 KiB
Python
218 lines
9.1 KiB
Python
|
||
# ╔╗╔┌─┐┬─┐┌─┐┬ ┬
|
||
# ║║║├┤ ├┬┘│ └┬┘
|
||
# ╝╚╝└─┘┴└─└─┘ ┴
|
||
|
||
# Code is licensed under CC-BY-NC-ND 4.0 unless otherwise specified.
|
||
# https://creativecommons.org/licenses/by-nc-nd/4.0/
|
||
# You CANNOT edit this file without direct permission from the author.
|
||
# You can redistribute this file without any changes.
|
||
|
||
# meta developer: @nercymods
|
||
# scope: hikka_min 1.6.2
|
||
# requires: matplotlib
|
||
|
||
from hikkatl.types import Message, PeerUser, PeerChat, PeerChannel
|
||
from .. import loader, utils
|
||
|
||
from collections import defaultdict
|
||
import matplotlib.pyplot as plt
|
||
import io
|
||
import asyncio
|
||
import warnings
|
||
import numpy as np
|
||
from matplotlib.colors import LinearSegmentedColormap
|
||
from telethon.tl.functions.messages import SearchRequest, GetHistoryRequest
|
||
from telethon.tl.types import InputMessagesFilterEmpty
|
||
|
||
plt.style.use('dark_background')
|
||
|
||
@loader.tds
|
||
class Top(loader.Module):
|
||
"""Module for viewing the top list in chat"""
|
||
strings = {
|
||
"name": "Top",
|
||
"top": "Top users by message count",
|
||
"topchat": "<emoji document_id=5323538339062628165>💬</emoji><b>Top users in</b>",
|
||
"msgcount": "Message count",
|
||
"loading": "<emoji document_id=5780543148782522693>🕒</emoji><b>Message counting has started, please wait, it may take a long time if there are a lot of messages in the chat</b>",
|
||
"private_chat": "<emoji document_id=5323538339062628165>💬</emoji><b>Message count in private chat with</b>"
|
||
}
|
||
|
||
strings_ru = {
|
||
"top": "Топ пользователей по количеству сообщений",
|
||
"topchat": "<emoji document_id=5323538339062628165>💬</emoji><b>Топ пользователей в</b>",
|
||
"msgcount": "Количество сообщений",
|
||
"loading": "<emoji document_id=5780543148782522693>🕒</emoji><b>Подсчет сообщений начался, пожалуйста подождите, это может занять много времени если в чате много сообщений</b>",
|
||
"private_chat": "<emoji document_id=5323538339062628165>💬</emoji><b>Количество сообщений в личном чате с</b>"
|
||
}
|
||
|
||
@loader.command(ru_doc="Посмотреть топ в чате")
|
||
async def top(self, m: Message):
|
||
"""View top in the chat"""
|
||
await utils.answer(m, self.strings['loading'])
|
||
|
||
client = self.client
|
||
|
||
if isinstance(m.peer_id, PeerUser):
|
||
chat_type = 'private'
|
||
chat_id = m.peer_id.user_id
|
||
elif isinstance(m.peer_id, PeerChat) or isinstance(m.peer_id, PeerChannel):
|
||
chat_type = 'chat'
|
||
chat_id = m.chat.id
|
||
else:
|
||
await utils.answer(m, "Unsupported chat type.")
|
||
return
|
||
|
||
if chat_type == 'chat':
|
||
users = await client.get_participants(chat_id)
|
||
users_dict = {user.id: (user.username or user.first_name) for user in users}
|
||
message_count = defaultdict(int)
|
||
|
||
for user_id in users_dict:
|
||
result = await client(SearchRequest(
|
||
peer=chat_id,
|
||
q='',
|
||
filter=InputMessagesFilterEmpty(),
|
||
from_id=user_id,
|
||
limit=0,
|
||
min_date=None,
|
||
max_date=None,
|
||
offset_id=0,
|
||
add_offset=0,
|
||
max_id=0,
|
||
min_id=0,
|
||
hash=0
|
||
))
|
||
message_count[user_id] = result.count
|
||
|
||
sorted_message_count = sorted(message_count.items(), key=lambda item: item[1], reverse=True)
|
||
top_users = sorted_message_count[:20]
|
||
usernames = [users_dict[user_id] or "Unknown" for user_id, _ in top_users]
|
||
counts = [count for _, count in top_users]
|
||
|
||
fig, ax = plt.subplots(figsize=(10, 5))
|
||
|
||
colors = self._generate_gradient('#8A2BE2', '#4B0082', len(usernames))
|
||
bars = ax.barh(usernames, counts, color=colors, edgecolor='black', linewidth=0.5)
|
||
|
||
for bar in bars:
|
||
bar.set_alpha(0.8)
|
||
bar.set_hatch('///')
|
||
|
||
ax.set_xlabel(self.strings['msgcount'], fontsize=12, color='white')
|
||
ax.set_title(self.strings['top'], fontsize=14, color='white', pad=20)
|
||
ax.invert_yaxis()
|
||
|
||
ax.spines['top'].set_visible(False)
|
||
ax.spines['right'].set_visible(False)
|
||
ax.spines['left'].set_color('#8A2BE2')
|
||
ax.spines['bottom'].set_color('#8A2BE2')
|
||
|
||
ax.grid(True, linestyle='--', alpha=0.6, color='gray')
|
||
|
||
for i, (bar, username) in enumerate(zip(bars, usernames)):
|
||
if i < 3:
|
||
bar.set_color('#FFD700')
|
||
ax.text(bar.get_width() + 5, bar.get_y() + bar.get_height() / 2,
|
||
f'#{i+1}', va='center', ha='left', color='#FFD700', fontsize=12)
|
||
|
||
buf = io.BytesIO()
|
||
with warnings.catch_warnings():
|
||
warnings.filterwarnings("ignore")
|
||
plt.savefig(buf, format='png', bbox_inches='tight', dpi=100)
|
||
buf.seek(0)
|
||
|
||
caption = f"{self.strings['topchat']} <b>{m.chat.title}:</b>\n"
|
||
caption += "\n".join([f"{i+1}. {user} - {count}" for i, (user, count) in enumerate(zip(usernames, counts))])
|
||
|
||
await utils.answer_file(m, buf, caption, force_document=False)
|
||
|
||
else:
|
||
me = await client.get_me()
|
||
target = await client.get_entity(chat_id)
|
||
|
||
my_count, their_count = await asyncio.gather(
|
||
self._get_message_count_fast(client, chat_id, me.id),
|
||
self._get_message_count_fast(client, chat_id, target.id)
|
||
)
|
||
|
||
message_counts = [(me.first_name, my_count), (target.first_name, their_count)]
|
||
sorted_message_counts = sorted(message_counts, key=lambda item: item[1], reverse=True)
|
||
|
||
usernames = [user for user, _ in sorted_message_counts]
|
||
counts = [count for _, count in sorted_message_counts]
|
||
|
||
fig, ax = plt.subplots(figsize=(10, 5))
|
||
|
||
colors = self._generate_gradient('#8A2BE2', '#4B0082', len(usernames))
|
||
bars = ax.barh(usernames, counts, color=colors, edgecolor='black', linewidth=0.5)
|
||
|
||
for bar in bars:
|
||
bar.set_alpha(0.8)
|
||
bar.set_hatch('///')
|
||
|
||
ax.set_xlabel(self.strings['msgcount'], fontsize=12, color='white')
|
||
ax.set_title(self.strings['top'], fontsize=14, color='white', pad=20)
|
||
ax.invert_yaxis()
|
||
|
||
ax.spines['top'].set_visible(False)
|
||
ax.spines['right'].set_visible(False)
|
||
ax.spines['left'].set_color('#8A2BE2')
|
||
ax.spines['bottom'].set_color('#8A2BE2')
|
||
|
||
ax.grid(True, linestyle='--', alpha=0.6, color='gray')
|
||
|
||
for i, (bar, username) in enumerate(zip(bars, usernames)):
|
||
if i < 3:
|
||
bar.set_color('#FFD700')
|
||
ax.text(bar.get_width() + 5, bar.get_y() + bar.get_height() / 2,
|
||
f'#{i+1}', va='center', ha='left', color='#FFD700', fontsize=12)
|
||
|
||
buf = io.BytesIO()
|
||
with warnings.catch_warnings():
|
||
warnings.filterwarnings("ignore")
|
||
plt.savefig(buf, format='png', bbox_inches='tight', dpi=100)
|
||
buf.seek(0)
|
||
|
||
caption = f"{self.strings['private_chat']} <b>{target.first_name}:</b>\n"
|
||
caption += "\n".join([f'"{user}" - {count}' for user, count in zip(usernames, counts)])
|
||
|
||
await utils.answer_file(m, buf, caption, force_document=False)
|
||
|
||
async def _get_message_count_fast(self, client, chat_id, user_id):
|
||
"""Получает количество сообщений от конкретного пользователя с использованием GetHistoryRequest"""
|
||
total_count = 0
|
||
offset_id = 0
|
||
limit = 100
|
||
|
||
while True:
|
||
history = await client(GetHistoryRequest(
|
||
peer=chat_id,
|
||
offset_id=offset_id,
|
||
offset_date=None,
|
||
add_offset=0,
|
||
limit=limit,
|
||
max_id=0,
|
||
min_id=0,
|
||
hash=0
|
||
))
|
||
|
||
if not history.messages:
|
||
break
|
||
|
||
for message in history.messages:
|
||
if message.sender_id == user_id:
|
||
total_count += 1
|
||
|
||
offset_id = history.messages[-1].id
|
||
|
||
if len(history.messages) < limit:
|
||
break
|
||
|
||
return total_count
|
||
|
||
def _generate_gradient(self, start_color, end_color, n):
|
||
"""Генерация градиента между двумя цветами"""
|
||
cmap = LinearSegmentedColormap.from_list('custom_gradient', [start_color, end_color], N=n)
|
||
return [cmap(i) for i in np.linspace(0, 1, n)] |