52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from datetime import UTC, datetime
|
|
|
|
from pyrogram.types import Chat
|
|
|
|
from userbot.modules.avatars import note_avatar
|
|
from userbot.modules.capture.context import CaptureContext
|
|
from userbot.modules.groups.repository import insert_chat_history
|
|
from userbot.modules.profiles.parse import snapshot_from_chat
|
|
from userbot.modules.profiles.repository import write_profile
|
|
from userbot.modules.raw_json import raw_json
|
|
|
|
|
|
async def save_private(ctx: CaptureContext, chat: Chat) -> bool:
|
|
chat_id = chat.id or 0
|
|
fields, photo_file_id, photo_unique_id = snapshot_from_chat(chat)
|
|
await write_profile(ctx.pool, ctx.account_id, chat_id, fields, raw_json(chat))
|
|
if photo_file_id and photo_unique_id:
|
|
await note_avatar(
|
|
ctx.pool, ctx.account_id, chat_id, "peer", photo_unique_id, photo_file_id
|
|
)
|
|
return bool(fields.first_name or fields.last_name or fields.username)
|
|
|
|
|
|
async def save_group(ctx: CaptureContext, chat: Chat) -> None:
|
|
chat_id = chat.id or 0
|
|
photo = chat.photo
|
|
photo_unique_id = photo.big_photo_unique_id if photo else None
|
|
photo_file_id = photo.big_file_id if photo else None
|
|
await insert_chat_history(
|
|
ctx.pool,
|
|
ctx.account_id,
|
|
chat_id,
|
|
0,
|
|
"meta",
|
|
chat.title,
|
|
photo_unique_id,
|
|
None,
|
|
datetime.now(UTC),
|
|
raw_json(chat),
|
|
)
|
|
if photo_file_id and photo_unique_id:
|
|
await note_avatar(
|
|
ctx.pool, ctx.account_id, chat_id, "chat", photo_unique_id, photo_file_id
|
|
)
|
|
|
|
|
|
async def save_chat(ctx: CaptureContext, chat: Chat) -> None:
|
|
if (chat.id or 0) > 0:
|
|
await save_private(ctx, chat)
|
|
else:
|
|
await save_group(ctx, chat)
|