o ^jHc@sUddlZddlZddlmZejejejedZe Z e e d<dZ dddd d d d dZ d edefddZd efddZdd edededefddZd ededefddZd edefddZdefddZdZdd ed edBdeefd!d"Zdd ed edBdeefd#d$Zd ed%ededBfd&d'Zd ed edBdeefd(d)Zd ed edBdefd*d+Zd ed edBd,edefd-d.Z  / dd ed edBd0ed1ed2ed3ed4ed,edefd5d6Zd ed%ed0efd7d8Z dd ed%ed2ed3efd9d:Z!d ed%ed4efd;d<Z"d ed%ed,efd=d>Z#d ed%ed?edBd@efdAdBZ$d ed%efdCdDZ%d ed%efdEdFZ&d ed%edGefdHdIZ'd ed%efdJdKZ(dd edLedeefdMdNZ)d edefdOdPZ*dd edQedBdeefdRdSZ+  dd ed edBdTedefdUdVZ,d edWedXedBdYedef dZd[Z-d edeefd\d]Z.d edWede/fd^d_Z0d edWefd`daZ1d edefdbdcZ2d edeefdddeZ3d edWede/fdfdgZ4d edWedXedBdYefdhdiZ5d edWefdjdkZ6d edeefdldmZ7 ndd edoedpedqedBdredBdsef dtduZ8d edvefdwdxZ9d edeefdydzZ:d edoedpedqedBdredBdsed{efd|d}Z;d edvefd~dZeeffddZ?d edvefddZ@dS)N)datetimedatabase _INIT_DONE)textphotovideodocumentaudiourlsubmenuu📝u🖼u🎬u📎u🎵u🔗u📂 bot_usernamereturncCs&tjtddtjt|dS)NT)exist_okz.db)osmakedirsDB_DIRpathjoinlower)r r*/var/www/kodo/factory/cobots/buttons/db.py get_db_pathsrc s\|}|tvr dStt|4IdH}|dIdH|dIdH|dIdH|dIdH|dIdH|dIdH|dIdHz|dIdH|d IdH|IdHWn tyqYnwgd }|D]\}}|d ||fIdHqx|IdHt|WdIdHdS1IdHswYdS) Nz CREATE TABLE IF NOT EXISTS settings ( key TEXT PRIMARY KEY, value TEXT ) a\ CREATE TABLE IF NOT EXISTS users ( id INTEGER PRIMARY KEY, username TEXT, first_name TEXT, is_banned INTEGER DEFAULT 0, is_blocked INTEGER DEFAULT 0, last_active TEXT, joined_at TEXT ) z CREATE TABLE IF NOT EXISTS admins ( id INTEGER PRIMARY KEY, username TEXT, first_name TEXT, added_at TEXT ) a CREATE TABLE IF NOT EXISTS mandatory_channels ( id INTEGER PRIMARY KEY AUTOINCREMENT, channel_id TEXT UNIQUE NOT NULL, channel_username TEXT, channel_title TEXT NOT NULL, invite_link TEXT, display_type TEXT DEFAULT 'buttons', is_active INTEGER DEFAULT 1, added_at TEXT ) av CREATE TABLE IF NOT EXISTS funded_channels ( id INTEGER PRIMARY KEY AUTOINCREMENT, channel_id TEXT UNIQUE NOT NULL, channel_username TEXT, channel_title TEXT NOT NULL, invite_link TEXT, display_type TEXT DEFAULT 'buttons', target_count INTEGER NOT NULL DEFAULT 100, current_count INTEGER DEFAULT 0, is_active INTEGER DEFAULT 1, added_at TEXT, completed_at TEXT ) a] CREATE TABLE IF NOT EXISTS funded_subscriptions ( id INTEGER PRIMARY KEY AUTOINCREMENT, funded_channel_id INTEGER NOT NULL, user_id INTEGER NOT NULL, subscribed_at TEXT, UNIQUE(funded_channel_id, user_id) ) a CREATE TABLE IF NOT EXISTS buttons ( id INTEGER PRIMARY KEY AUTOINCREMENT, parent_id INTEGER DEFAULT NULL, label TEXT NOT NULL, response_type TEXT NOT NULL, response_data TEXT DEFAULT '', response_caption TEXT DEFAULT '', columns INTEGER DEFAULT 5, row_number INTEGER DEFAULT 0, sort_order INTEGER DEFAULT 0, press_count INTEGER DEFAULT 0, is_active INTEGER DEFAULT 1, created_at TEXT ) z;ALTER TABLE buttons ADD COLUMN row_number INTEGER DEFAULT 0z*UPDATE buttons SET row_number = sort_order))is_open1)closed_messageu(🔴 البوت مقفول حالياً.)welcome_message) welcome_imager)notify_new_usersr)notify_blockedr) fund_notifyr) root_columns5z9INSERT OR IGNORE INTO settings (key, value) VALUES (?, ?)) rr aiosqliteconnectrexecutecommit Exceptionadd)r keydbdefaultsdkeyvaluerrrinit_db s8      .r.rr)defaultc stt|4IdHF}|d|f4IdH$}|IdH}|r&|dn|WdIdHWdIdHS1IdHsBwYWdIdHdS1IdHsXwYdS)Nz&SELECT value FROM settings WHERE key=?rr#r$rr%fetchone)r r)r/r*currowrrr get_settings.r4r-c sntt|4IdH}|d||fIdH|IdHWdIdHdS1IdHs0wYdS)Nz:INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)r#r$rr%r&)r r)r-r*rrr set_settings .r6c tt|4IdHA}|d4IdH!}dd|IdHDWdIdHWdIdHS1IdHs=wYWdIdHdS1IdHsSwYdS)NzSELECT key, value FROM settingscSsi|] }|d|dqS)rr.0rrrr z$get_all_settings..r#r$rr%fetchallr r*r2rrrget_all_settingss.rAc CsR|d|d|d|d|d|d|d|d|d |d t|d |d d S)Nrr8 ) id parent_idlabel response_type response_dataresponse_captioncolumns row_number sort_order press_count is_active created_at)bool)r3rrr _row_to_btns rYz~id,parent_id,label,response_type,response_data,response_caption,columns,row_number,sort_order,press_count,is_active,created_atrMc tt|4IdHG}|dtd|f4IdH!}dd|IdHDWdIdHWdIdHS1IdHsCwYWdIdHdS1IdHsYwYdS)NSELECT zR FROM buttons WHERE is_active=1 AND parent_id IS ? ORDER BY row_number, sort_ordercSg|]}t|qSrrYr9rrr z&get_active_buttons..r#r$rr% _BTN_FIELDSr?r rMr*r2rrrget_active_buttons  .rcc rZ)Nr[zB FROM buttons WHERE parent_id IS ? ORDER BY row_number, sort_ordercSr\rr]r9rrrr^r_z,get_all_buttons_at_level..r`rbrrrget_all_buttons_at_levelrdre button_idc stt|4IdHJ}|dtd|f4IdH$}|IdH}|r*t|ndWdIdHWdIdHS1IdHsFwYWdIdHdS1IdHs\wYdS)Nr[z FROM buttons WHERE id=?)r#r$rr%rar1rY)r rfr*r2r3rrr get_buttons .rgcsRt||IdH}i}|D]}|d}||dd||<q ddt|DS)NrSrr8cSsg|] \}}||dqS))rScountr)r:rncntrrrr^r=z!get_rows_info..)regetsorteditems)r rMbuttonsrowsbtnrirrr get_rows_infosrqc stt|4IdHF}|d|f4IdH$}|IdH}|r&|dndWdIdHWdIdHS1IdHsBwYWdIdHdS1IdHsXwYdS)NzJSELECT COALESCE(MAX(row_number), -1) + 1 FROM buttons WHERE parent_id IS ?rr0)r rMr*r2r3rrrget_next_row_numbers .rrrSc stt|4IdHG}|d||f4IdH$}|IdH}|r'|dndWdIdHWdIdHS1IdHsCwYWdIdHdS1IdHsYwYdS)Nz[SELECT COALESCE(MAX(sort_order), -1) + 1 FROM buttons WHERE parent_id IS ? AND row_number=?rr0)r rMrSr*r2r3rrrget_next_sort_in_rows .rsrErNrOrPrQrRc st}t|||IdH} tt|4IdH'} | d|||||||| |f IdH} | IdH| j WdIdHS1IdHsHwYdS)NzINSERT INTO buttons (parent_id,label,response_type,response_data,response_caption,columns,row_number,sort_order,press_count,is_active,created_at) VALUES (?,?,?,?,?,?,?,?,0,1,?)) rutcnow isoformatrsr#r$rr%r& lastrowid) r rMrNrOrPrQrRrSnowrTr*r2rrr add_buttons   0rxc ntt|4IdH}|d||fIdH|IdHWdIdHdS1IdHs0wYdS)Nz%UPDATE buttons SET label=? WHERE id=?r5)r rfrNr*rrrupdate_button_label .rzc sptt|4IdH}|d|||fIdH|IdHWdIdHdS1IdHs1wYdS)NzAUPDATE buttons SET response_data=?, response_caption=? WHERE id=?r5)r rfrPrQr*rrrupdate_button_contents .r|c ry)Nz'UPDATE buttons SET columns=? WHERE id=?r5)r rfrRr*rrrupdate_button_columnsr{r}c st||IdH}|s dSt||d|IdH}tt|4IdH}|d|||fIdH|IdHWdIdHdS1IdHsHwYdS)NrM8UPDATE buttons SET row_number=?, sort_order=? WHERE id=?)rgrsr#r$rr%r&)r rfrSrprTr*rrrupdate_button_rows .r new_parent_idnew_rowc st|||IdH}tt|4IdH }|d||||fIdH|IdHWdIdHdS1IdHs;wYdS)NzEUPDATE buttons SET parent_id=?, row_number=?, sort_order=? WHERE id=?)rsr#r$rr%r&)r rfrrrTr*rrrmove_button_to$s  .rc ltt|4IdH}|d|fIdH|IdHWdIdHdS1IdHs/wYdS)Nz7UPDATE buttons SET is_active = 1 - is_active WHERE id=?r5r rfr*rrr toggle_button/ .rc stt|4IdHh}|g}|g}|rS|d}|d|f4IdH!}dd|IdHD}||||WdIdHn 1IdHsLwY|s|D] }|d|fIdHqU|IdHWdIdHdS1IdHszwYdS)Nrz(SELECT id FROM buttons WHERE parent_id=?cSsg|]}|dqS)rrr9rrrr^@r_z)delete_button_cascade..zDELETE FROM buttons WHERE id=?)r#r$rpopr%r?extendr&) r rfr* to_deletequeuecur_idr2childrenbidrrrdelete_button_cascade7s&    (.r directionc s8t|IdH}|s dSt||dIdH}tfddt|Dd}|dur+dS|dkr:|dkr:||d}n|dkrM|t|dkrM||d}ndStt|4IdH5}|d|d |d fIdH|d|d |d |d fIdH| IdHWdIdHdS1IdHswYdS) NrMc3s$|] \}}|dkr|VqdS)rLNr)r:ibrfrr Ms"zmove_button..uprr8downr~rSrTrL) rgrenext enumeratelenr#r$rr%r&)r rfrrpall_btnsidxneighborr*rrr move_buttonHs0  .rc r)Nz;UPDATE buttons SET press_count = press_count + 1 WHERE id=?r5rrrrincrement_pressbrrlimitc stt|4IdHC}|d|f4IdH!}dd|IdHDWdIdHWdIdHS1IdHs?wYWdIdHdS1IdHsUwYdS)NzXSELECT id,label,response_type,press_count FROM buttons ORDER BY press_count DESC LIMIT ?cSs*g|]}|d|d|d|ddqS)rr8rBrC)rLrNrOrUrr9rrrr^qsz#get_top_buttons..r>)r rr*r2rrrget_top_buttonsjs  .rc stt|4IdH>}|d4IdH}|IdHdWdIdHWdIdHS1IdHs:wYWdIdHdS1IdHsPwYdS)NzSELECT COUNT(*) FROM buttonsrr0r@rrr count_buttonsws.r exclude_idc stt|4IdH6}|d4IdH}dd|IdHD}WdIdHn 1IdHs3wYWdIdHn 1IdHsHwY|rz|h|g}|rq|dfdd|D}||||sWfdd|D}|S)NzDSELECT id,parent_id,label FROM buttons WHERE response_type='submenu'cS$g|]}|d|d|ddqS)rr8rB)rLrMrNrr9rrrr^z$get_all_submenus..rcs g|] }|dkr|dqS)rMrLrr9)rrrr^s csg|] }|dvr|qS)rLrr9)excludedrrr^s)r#r$rr%r?rupdater)r rr*r2rorrr)rrrget_all_submenus}s,  *(   rprefixc st||IdH}g}t|D]t\}}|t|dk}|rdnd}t|dd} |dr/dnd} |d d kr?d |d d nd } d|ddd} |||d| d| d|d| | |ddkr||rndnd} t||d| IdH}|r||qd|S)Nr8u└─u├─rOu•rVu🟢u🔴rUrz (u ✦)ru ┤rSu├ rNr z u│ rL )rerr TYPE_ICONSrkappendbuild_tree_textr)r rMrrnlinesrrpis_last connectoriconstatuspressesrow_tag child_prefix child_treerrrrs$  0   ruser_idusername first_namec sFt}tt|4IdH}|d|f4IdH}|IdH}WdIdHn 1IdHs6wY|se|d|||||fIdH|IdH|||ddddWdIdHS|d||||fIdH|IdH|d|d|d|d |d d dWdIdHS1IdHswYdS) NzHSELECT id,username,first_name,is_banned,is_blocked FROM users WHERE id=?zRINSERT INTO users (id,username,first_name,last_active,joined_at) VALUES(?,?,?,?,?)rT)rLrr is_banned is_blockedis_newzAUPDATE users SET username=?,first_name=?,last_active=? WHERE id=?r8rBrCrDF rrtrur#r$rr%r1r&)r rrrrwr*r2r3rrrget_or_create_users:  (    0rc r7)NzKSELECT id,username,first_name FROM users WHERE is_banned=0 AND is_blocked=0cSrrr8rB)rLrrrr9rrrr^rz(get_all_active_users..r>r@rrrget_all_active_userss  .rc stt|4IdH[}|d|f4IdH'}|IdHs5 WdIdHWdIdHdSWdIdHn 1IdHsEwY|d|fIdH|IdH WdIdHdS1IdHsmwYdS)NzSELECT id FROM users WHERE id=?Fz'UPDATE users SET is_banned=1 WHERE id=?T)r#r$rr%r1r&r rr*r2rrrban_users(0rc r)Nz'UPDATE users SET is_banned=0 WHERE id=?r5r rr*rrr unban_user .rc sJtt|4IdH}|d4IdH}|IdHd}WdIdHn 1IdHs0wY|d4IdH}|IdHd}WdIdHn 1IdHsXwY|d4IdH}|IdHd}WdIdHn 1IdHswYWdIdHn 1IdHswY||||||dS)NzSELECT COUNT(*) FROM usersrz,SELECT COUNT(*) FROM users WHERE is_banned=1z-SELECT COUNT(*) FROM users WHERE is_blocked=1)totalbannedblockedactiver0)r r*r2rrrrrrget_statisticss((*( rc r7)Nz)SELECT id,username,first_name FROM adminscSrrrr9rrrr^rzget_admins..r>r@rrr get_adminss .rc stt|4IdH@}|d|f4IdH}t|IdHWdIdHWdIdHS1IdHs.r>r@rrrget_mandatory_channelss  .rrnrrrrrc st}tt|4IdH"}|d||||||fIdH|IdHWdIdHdS1IdHs:wYdS)NzINSERT OR REPLACE INTO mandatory_channels (channel_id,channel_username,channel_title,invite_link,display_type,is_active,added_at) VALUES(?,?,?,?,?,1,?)r)r rrrrrrwr*rrradd_mandatory_channel s  .r channel_db_idc r)Nz4UPDATE mandatory_channels SET is_active=0 WHERE id=?r5r rr*rrrremove_mandatory_channelrrc r7)NzSELECT id,channel_id,channel_username,channel_title,invite_link,display_type,target_count,current_count FROM funded_channels WHERE is_active=1c SsBg|]}|d|d|d|d|d|d|d|ddqS) rr8rBrCrDrErFrG)rLrrrrr target_count current_countrr9rrrr^%s   z'get_funded_channels..r>r@rrrget_funded_channelss  .rrc st}tt|4IdH#}|d|||||||fIdH|IdHWdIdHdS1IdHs;wYdS)NzINSERT OR REPLACE INTO funded_channels (channel_id,channel_username,channel_title,invite_link,display_type,target_count,current_count,is_active,added_at) VALUES(?,?,?,?,?,?,0,1,?)r) r rrrrrrrwr*rrradd_funded_channel+s  .rc r)Nz1UPDATE funded_channels SET is_active=0 WHERE id=?r5rrrrremove_funded_channel:rrc st}tt|4IdHh}|d||f4IdH'}|IdHr< WdIdHWdIdHdSWdIdHn 1IdHsLwY|d|||fIdH|d|fIdH|IdH WdIdHdS1IdHswYdS)NzKSELECT id FROM funded_subscriptions WHERE funded_channel_id=? AND user_id=?FzXINSERT INTO funded_subscriptions (funded_channel_id,user_id,subscribed_at) VALUES(?,?,?)zCUPDATE funded_channels SET current_count=current_count+1 WHERE id=?Tr)r rrrwr*r2rrrrecord_funded_subBs2  (  0rc stt|4IdHJ}|d|f4IdH(}|IdH}|r*|d|dfndWdIdHWdIdHS1IdHsFwYWdIdHdS1IdHs\wYdS)NzASELECT current_count,target_count FROM funded_channels WHERE id=?rr8)rrr0)r rr*r2r3rrrget_funded_channel_countWs .rc szt}tt|4IdH}|d||fIdH|IdHWdIdHdS1IdHs6wYdS)NzAUPDATE funded_channels SET is_active=0, completed_at=? WHERE id=?r)r rrwr*rrrmark_funded_completeas  .r)r)N)rrrEr)rE)Nr)rn)Arr#rrrdirnameabspath__file__rsetr__annotations__RESPONSE_TYPESrstrrr.r4r6dictrArYraintlistrcrergrqrrrsrxrzr|r}rrrrrrrrrrrrrXrrrrrrrrrrrrrrtuplerrrrrrs   l