o Ej-R@sUddlZddlZddlZddlZddlZddlZddlmZddlZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZeeZdZdZdZdaejed<iZiZeZ iZ!gZ"e#ed<da$ej%ed <d ej%fd d Z&e'Z(e'ed <iZ)e*ed<de+d ej%fddZ,dZ-dZ.dZ/dZ0dZ1e'Z2e'ed<ga3e#ed<iZ4e*ed<de+d ej%fddZ5e'Z6e'ed<de+d e7fddZ8de+fddZ9ddZ:d d!Z;dd"d#Zd(d)Z?d e@fd*d+ZAd,d-ZBd.d/ZCdd0d1ZDd2d3ZEd4d5ZFd6d7ZGdd e7fd9d:ZHd;d<ZId=d>ZJd?d@ZKdAdBZLdCdDZMdEdFZNdGdHZOdIdJZPdKe*fdLdMZQddNdOZRddPe+dQe+dRe@fdSdTZSddPe+d e#fdUdVZTd e'fdWdXZUd eVfdYdZZWdPe+fd[d\ZXdde+fd]d^ZYd_d`ZZdae[d e[fdbdcZ\dde[fdedfZ]dgdhZ^didjZ_de+d e+fdkdlZ`dme+d e*fdndoZaidpdqdrdsdtdudvdwdxdydzd8d{d|d}d~ddddddddddsdd8dd|dd~dddddwddZbd e+fddZcd e7fddZdde+d eVe7e+ffddZedS)N)datetime) UserDeactivatedUserDeactivatedBan UserInvalid PeerIdInvalidInputUserDeactivatedSessionRevokedSessionExpiredAuthKeyUnregisteredAuthKeyDuplicatedUserBannedInChannel FloodWaitz0postgresql://mahde:MahdeFab1YssR@localhost/mahde i,_pg_pool_join_notify_buffer_join_notify_lockreturncCstdurtatSN)rasyncioLockrr shared.pyget_join_notify_lock%sr dead_sessions account_locksphonecC.|tvr tt|<td|dt|S)Nz[LOCK] account_lock  ▶)rrrlogdebugrrrrget_account_lock/ r!F_background_tasks _admins_cache_get_client_lockscCr)Nz[LOCK] client_open_lock r)r%rrrrr rrrget_client_open_lock?r"r&_banned_channelstargetcCs t|tvSr) norm_ch_keyr')r(rrris_channel_bannedFs r*c st|}t|td|t4IdH}|d|t IdHWdIdHdS1IdHs:wYdS)Nu%[CHANNEL BAN] قناة محظورة: zAINSERT INTO banned_channels VALUES ($1,$2) ON CONFLICT DO NOTHING) r)r'addrwarningracquireexecuternow isoformat)r(keyconrrr ban_channelIs   .r3c stjtddddIdHat4IdH<}|dIdHgd}|D]\}}|d||IdHq$|d d d IdH|d d d IdHWdIdHn 1IdHsWwYtdIdH}|D] }t|dqft dt tdt ddS)u@إنشاء Pool والجداول إذا لم تكن موجودة)min_sizemax_sizecommand_timeoutNa CREATE TABLE IF NOT EXISTS settings ( key TEXT PRIMARY KEY, value TEXT ); CREATE TABLE IF NOT EXISTS admins ( user_id BIGINT PRIMARY KEY ); CREATE TABLE IF NOT EXISTS accounts ( phone TEXT PRIMARY KEY, banned INTEGER DEFAULT 0 ); CREATE TABLE IF NOT EXISTS banned_accounts ( phone TEXT PRIMARY KEY, banned_at TEXT, reason TEXT ); CREATE TABLE IF NOT EXISTS bots ( username TEXT PRIMARY KEY, url TEXT ); CREATE TABLE IF NOT EXISTS channels ( id BIGINT, title TEXT, username TEXT, url TEXT, joined_at TEXT, PRIMARY KEY (id, url) ); CREATE TABLE IF NOT EXISTS force_sub_channels ( bot_username TEXT NOT NULL, url TEXT NOT NULL, channel_id BIGINT, PRIMARY KEY (bot_username, url) ); ))gathering_speed5) concurrent20) is_active0) work_minutes15) rest_minutesr;) active_botz?INSERT INTO settings VALUES ($1,$2) ON CONFLICT(key) DO NOTHINGKINSERT INTO settings VALUES ($1,$2) ON CONFLICT(key) DO UPDATE SET value=$2r>r?rCrDz'SELECT channel_key FROM banned_channels channel_keyu[DB] تم تحميل u قناة محظورةu[DB] PostgreSQL pool جاهز) asyncpg create_poolPG_DSNrr-r.fetchr'r+rinfolen)r2defaultskvrowsrrrrinit_dbTs: %    (>rRcCs\ddl}z|t}|}|d|}|dd|DWSty-gYSw)uAللاستخدام عند الإقلاع قبل بدء event looprNSELECT user_id FROM adminscSg|]}|dqS)rr.0rQrrr z#get_admins_sync..)psycopg2connectrIcursorr.fetchallclose Exception)rYr2currPrrrget_admins_syncs   r`c sdt4IdH}|d|IdH}|r|dn|WdIdHS1IdHs+wYdS)Nz'SELECT value FROM settings WHERE key=$1valuerr-fetchrow)r1defaultr2rowrrr get_settings 0rfc s\t4IdH}|d|t|IdHWdIdHdS1IdHs'wYdS)NrE)rr-r.str)r1rar2rrr set_settings .rhc4ztttddIdHdWStyYdSw)Nr:r4rmaxintrfr^rrrr get_speed  rmcri)Nr<r5rjrrrrget_concurrentrnrpcri)Nconcurrent_all_bots<rorjrrrrget_concurrent_all_botsrnrscri)Nr@rorjrrrrget_work_minutesrnrucri)NrBr4rorjrrrrget_rest_minutesrnrvc st4IdH5}|r%|dIdH}dd|DWdIdHS|dIdH}dd|DWdIdHS1IdHsDwYdS)Nz)SELECT phone FROM accounts WHERE banned=0cSsg|]}d|diqSr rrUrrrrWz get_accounts..z1SELECT phone, banned FROM accounts WHERE banned=0cSs"g|] }|dt|ddqS)rbanned)rrx)boolrUrrrrWs"rr-rJ) active_onlyr2rPrrr get_accountss  0r|c sbt4IdH}|dIdH}|r|dndWdIdHS1IdHs*wYdS)Nz,SELECT COUNT(*) FROM accounts WHERE banned=0countrrb)r2rerrrget_active_counts 0r~c Vt4IdH}|d|IdHWdIdHdS1IdHs$wYdS)Nz9INSERT INTO accounts VALUES ($1,0) ON CONFLICT DO NOTHINGrr-r.rr2rrr add_accounts .rc r)N#DELETE FROM accounts WHERE phone=$1rrrrr del_account  .r محظورc std|d|t4IdH,}|d|IdH}|d|IdH|d|t|IdHWdIdHn 1IdHsFwY|rXtd|ddSt |IdHt j d d |}t j |d d d }|d|dfD]=}t j d |}zt j |rt |t j |||d7}Wqzty}ztd|d|WYd}~qzd}~wwtd|d|dd S)uz يعيد True إذا كان الحظر جديداً يعيد False إذا كان محظوراً مسبقاً z[BAN] u ▶ حظر: Nz,SELECT 1 FROM banned_accounts WHERE phone=$1rzXINSERT INTO banned_accounts VALUES ($1,$2,$3) ON CONFLICT(phone) DO UPDATE SET reason=$3u' ■ كان محظوراً مسبقاًFsessions quarantineT)exist_okrz.sessionz.session-journalrou% ✗ نقل لل quarantين فشل: u+ ✓ جلسة نُقلت لل quarantين (u ملف))rr,rr-rcr.rr/r0cleanup_clientospathjoinmakedirsexistsrenamer^errorrK) rreasonr2alreadyqdirmovednamererrr ban_accountsD   (  "rc `t4IdH}|dIdH}dd|DWdIdHS1IdHs)wYdS)Nz4SELECT phone, banned_at, reason FROM banned_accountscSs(g|]}|d|d|dpddqS)r banned_atrr)rrrrrUrrrrW4s zget_banned..rzr2rPrrr get_banned/s 0rc r)NzSELECT username, url FROM botscSsg|] }|d|ddqS)usernameurlrrrrUrrrrWAszget_bots..rzrrrrget_bots>  0rc sXt4IdH}|d||IdHWdIdHdS1IdHs%wYdS)Nz6INSERT INTO bots VALUES ($1,$2) ON CONFLICT DO NOTHINGr)rrr2rrradd_botDs .rc r)Nz"DELETE FROM bots WHERE username=$1r)rr2rrrdel_botLrrc r)NrScSrT)user_idrrUrrrrWVrXzget_admins..rzrrrr get_adminsSrrc `t4IdH}|d|IdHWdIdHn 1IdHs#wYtIdHadS)Nz5INSERT INTO admins VALUES ($1) ON CONFLICT DO NOTHINGrr-r.rr$uidr2rrr add_adminYs (rc r)Nz#DELETE FROM admins WHERE user_id=$1rrrrr del_admincs (rc r)Nz8SELECT id, title, username, url, joined_at FROM channelscSs0g|]}|d|d|d|d|ddqS)idtitlerr joined_at)rrrrrrrUrrrrWqsz get_channels..rzrrrr get_channelsls 0rchc s|dpd}|dpd}t4IdH>}|d||IdH}|sJ|d||d|d||d tIdHWdIdHdSWdIdHdS1IdHs[wYdS) NrrrrDzFSELECT 1 FROM channels WHERE (id=$1 AND id!=0) OR (url=$2 AND url!='')zCINSERT INTO channels VALUES ($1,$2,$3,$4,$5) ON CONFLICT DO NOTHINGrrr)getrr-rcr.rr/r0)rch_idrr2rrrr add_channel}s(  .rc st4IdH9}|r|d|IdHn!|r,|d|IdHWdIdHdSWdIdHdSWdIdHdS1IdHsHwYdS)Nz DELETE FROM channels WHERE id=$1z!DELETE FROM channels WHERE url=$1r)rrr2rrr del_channels  .r bot_usernamer channel_idc sht4IdH}|d|d||IdHWdIdHdS1IdHs-wYdS)NzINSERT INTO force_sub_channels (bot_username, url, channel_id) VALUES ($1,$2,$3) ON CONFLICT (bot_username, url) DO UPDATE SET channel_id = COALESCE(EXCLUDED.channel_id, force_sub_channels.channel_id)@rr-r.lowerstrip)rrrr2rrradd_force_sub_channels .rc st4IdH+}|r|d|dIdH}n|dIdH}dd|DWdIdHS1IdHs:wYdS)NzRSELECT bot_username, url, channel_id FROM force_sub_channels WHERE bot_username=$1rz