o <6j @sddlZddlmZmZddlmZmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZddlmZdd lmZmZmZmZmZmZd ed ed eefd dZ dded ededeed ef ddZdededed dfddZdS)N)datetimetimezone)ListOptional)Bot)TelegramForbiddenErrorTelegramRetryAfterTelegramBadRequest)logger)selectupdate) AsyncSession)BroadcastMessageBroadcastStatusBroadcastTarget MediaTypeUserLanguagesessiontargetreturncs ddlm}ttjtjdktjdk}|tj kr/t t j |dd}|tj |k}nE|tjkrHt t j |dd}|tj |k}n,|tjkrW|tjtjk}n|tjkrf|tjtjk}n|tjkrt|tjtjk}||IdH}d d |DS) z-Target ga qarab user Telegram ID larini olishr) timedeltaFT)daysNcSsg|]}|dqS)r).0rowrrE/var/www/www-root/data/www/ai-bot.pdev.uz/bot/services/broadcaster.py ,sz'get_target_user_ids..)rrr r telegram_idwhere is_banned is_activer ACTIVE_7Dnowrutclast_active_at ACTIVE_30DLANG_UZlanguagerUZLANG_RURULANG_ENENexecuteall)rrrqsinceresultrrrget_target_user_idss"      r5bot broadcastuser_idsc sZt|}d}d}tj|_ttj|_||_ | IdHt d|j d|dt|D]\}} zt|| |IdH|d7}Wntyc|tttj| kjddIdH|d7}Ynbty} z6t d | jd t| jIdHzt|| |IdH|d7}Wn ty|d7}YnwWYd} ~ n%d} ~ wty} zt d | d | |d7}WYd} ~ nd} ~ ww|r|dd dkrz ||||IdHWn tyYnwtdIdH|dddkr||_||_| IdHq/tj |_||_||_ttj|_!| IdHt d|j d|d||S)u Broadcast xabarini yuborish. Telegram rate limit: 30 msg/sec — biz 25 ta/sek yuboramiz. Args: progress_callback: async funksiya (sent, total) — progress bildirish uchun rNz Broadcast #z boshlandi. Jami: z userF)r#z Rate limit! zs kutilyapti...zUser z ga yuborishda xatolik: 2g{Gz?dz yakunlandi. Yuborildi: z, Xato: )"lenrRUNNINGstatusrr%rr& started_at total_userscommitr infoid enumerate _send_to_userrr0r rr!r valuesrwarning retry_afterasynciosleep Exceptionerror total_sent total_failedDONE finished_at) r6rr7r8progress_callbacktotalsentfailediuser_iderrr run_broadcast/sz         rXrVcs|jpd}|j}|j}|tjkr |r |j|||dIdHdS|tjkr4|r4|j|||dIdHdS|tjkrH|rH|j |||dIdHdS|tj kr\|r\|j |||dIdHdS|j ||dIdHdS)uBitta usergа xabar yuborish)chat_idphotocaptionN)rZvideor\)rZdocumentr\)rZ animationr\)rZtext) r` media_file_id media_typerPHOTO send_photoVIDEO send_videoDOCUMENT send_document ANIMATIONsend_animation send_message)r6rVr7r`file_idrbrrrrEs rE)N)rIrrtypingrraiogramraiogram.exceptionsrrr logurur sqlalchemyr r sqlalchemy.ext.asyncior bot.database.modelsrrrrrrintr5rXrErrrrsJ     # R