31 lines
926 B
Python
31 lines
926 B
Python
from datetime import UTC, datetime
|
|
|
|
from pyrogram.types import User
|
|
|
|
from userbot.modules.client import PyroClient
|
|
from userbot.modules.presence import repository
|
|
from userbot.modules.raw_json import raw_json
|
|
from utils.events import notify_bg_event
|
|
|
|
|
|
@PyroClient.on_user_status()
|
|
async def on_user_status(client: PyroClient, user: User) -> None:
|
|
ctx = client.capture
|
|
if ctx is None or user.status is None:
|
|
return
|
|
await repository.insert_status(
|
|
ctx.pool,
|
|
ctx.account_id,
|
|
user.id,
|
|
datetime.now(UTC),
|
|
user.status.name.lower(),
|
|
user.last_online_date,
|
|
user.next_offline_date,
|
|
raw_json(user.raw),
|
|
)
|
|
await ctx.watches.on_status(user.id, is_online=user.status.name.lower() == "online")
|
|
await notify_bg_event(ctx.pool, "presence", ctx.account_id, chat_id=user.id)
|
|
|
|
|
|
handlers = on_user_status.handlers # ty: ignore[unresolved-attribute]
|