o ajH<@sddlZddlZddlmZmZejejejedZ de de fddZ de fdd Z dTde d e d e de fd dZ de d e de fddZde defddZde dede dBde def ddZde dededBfddZde deefddZde deefddZde dedefdd Zde defd!d"Zde defd#d$Zde ded%e fd&d'Zde dede dBde fd(d)Zde defd*d+Zde deefd,d-Zde dedefd.d/Zde fd0d1Z 2dUde d3e d4e d5e dBd6e dBd7e f d8d9Z!de deefd:d;Z"de dZ#d?d@Z$de dAe dBe dCe def dDdEZ%de deefdFdGZ&de dHededBfdIdJZ'   K dVde dHedAe dBe dCe dLe dMe dNe dOefdPdQZ(de dHefdRdSZ)dS)WN)datetime timedeltadatabase bot_usernamereturncCs&tjtddtjt|dS)NT)exist_okz.db)osmakedirsDB_DIRpathjoinlower)rr(/var/www/kodo/factory/cobots/decor/db.py get_db_path src stt|4IdHY}|dIdH|dIdH|dIdH|dIdH|dIdH|dIdHgd}|D]\}}|d||fIdHqC|IdHWdIdHdS1IdHskwYdS) 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 ) z CREATE TABLE IF NOT EXISTS activity ( id INTEGER PRIMARY KEY AUTOINCREMENT, user_id INTEGER NOT NULL, action TEXT NOT NULL, created_at TEXT ) a_ CREATE TABLE IF NOT EXISTS custom_buttons ( id INTEGER PRIMARY KEY AUTOINCREMENT, btn_type TEXT NOT NULL, label TEXT NOT NULL, content TEXT NOT NULL, position INTEGER DEFAULT 0, created_at TEXT ) ))is_open1)closed_messageu(🔴 البوت مقفول حالياً.)notify_new_usersr)notify_blockedr)welcome_messagez9INSERT OR IGNORE INTO settings (key, value) VALUES (?, ?) aiosqliteconnectrexecutecommit)rdbdefaultskeyvaluerrrinit_dbs     .r!rrdefaultc stt|4IdHF}|d|f4IdH$}|IdH}|r&|dn|WdIdHWdIdHS1IdHsBwYWdIdHdS1IdHsXwYdS)Nz(SELECT value FROM settings WHERE key = ?rrrrrfetchone)rrr"rcurrowrrr get_settingZs.r'r c sntt|4IdH}|d||fIdH|IdHWdIdHdS1IdHs0wYdS)Nz:INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)r)rrr rrrr set_settingas .r(c tt|4IdHC}|d4IdH#}|IdH}dd|DWdIdHWdIdHS1IdHs?wYWdIdHdS1IdHsUwYdS)NzSELECT key, value FROM settingscSsi|] }|d|dqS)rr).0r&rrr nsz$get_all_settings..rrrrfetchallrrr%rowsrrrget_all_settingsj .r1user_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) N SELECT * FROM users WHERE id = ?z[INSERT INTO users (id, username, first_name, last_active, joined_at) VALUES (?, ?, ?, ?, ?)rT)idr4r5 is_banned is_blockedis_newzKUPDATE users SET username = ?, first_name = ?, last_active = ? WHERE id = ?r*F) rutcnow isoformatrrrrr$r)rr3r4r5nowrr%r&rrrget_or_create_userqs2 (     0rAc stt|4IdHg}|d|f4IdHE}|IdH}|s7 WdIdHWdIdHdS|d|d|d|d|ddWdIdHWdIdHS1IdHscwYWdIdHdS1IdHsywYdS)Nr6rr*r;r<r=)r7r4r5r8r9r#)rr3rr%r&rrrget_users .rBc r))Nz;SELECT id FROM users WHERE is_blocked = 0 AND is_banned = 0cSsg|]}d|diqS)r7rrr+rrrr sz(get_all_active_users..r-r/rrrget_all_active_usersr2rFc r))Nz>SELECT id, username, first_name FROM users WHERE is_banned = 1cS$g|]}|d|d|ddqSrr*r;)r7r4r5rrCrrrrE$z$get_banned_users..r-r/rrrget_banned_userss  .rJc stt|4IdH[}|d|f4IdH'}|IdHs5 WdIdHWdIdHdSWdIdHn 1IdHsEwY|d|fIdH|IdH WdIdHdS1IdHsmwYdS)Nz!SELECT id FROM users WHERE id = ?Fz+UPDATE users SET is_banned = 1 WHERE id = ?T)rrrrr$rrr3rr%rrrban_users(0rLc ltt|4IdH}|d|fIdH|IdHWdIdHdS1IdHs/wYdS)Nz+UPDATE users SET is_banned = 0 WHERE id = ?rrr3rrrr unban_user .rOc sHt}|tdd}tt|4IdHv}|dIdHIdHd}|d|fIdHIdHd}|dIdHIdHd}|dIdHIdHd}|dIdHIdHd}|d IdHIdHd} |d IdHIdHd} WdIdHn 1IdHswY|||||| | d S) N)hourszSELECT COUNT(*) FROM usersrz1SELECT COUNT(*) FROM users WHERE last_active >= ?z.SELECT COUNT(*) FROM users WHERE is_banned = 1z/SELECT COUNT(*) FROM users WHERE is_blocked = 1z6SELECT COUNT(*) FROM activity WHERE action = 'message'z7SELECT COUNT(*) FROM activity WHERE action = 'decorate'z;SELECT COUNT(*) FROM mandatory_channels WHERE is_active = 1) total_users active_24h banned_users blocked_botmessages decorations mand_count) rr>rr?rrrrr$) rr@last_24hrtotalactivebannedblockedrWrXrYrrrget_statisticss"" ( r_actionc s|t}tt|4IdH}|d|||fIdH|IdHWdIdHdS1IdHs7wYdS)NzCINSERT INTO activity (user_id, action, created_at) VALUES (?, ?, ?)rr>r?rrrrr)rr3r`r@rrrrrecord_activitys  .rbc s~t}tt|4IdH }|d||||fIdH|IdHWdIdHdS1IdHs8wYdS)NzVINSERT OR REPLACE INTO admins (id, username, first_name, added_at) VALUES (?, ?, ?, ?)ra)rr3r4r5r@rrrr add_admins   .rcc rM)NzDELETE FROM admins WHERE id = ?rrNrrr remove_adminrPrdc r))NzSELECT * FROM adminscSrGrHrrCrrrrErIzget_admins..r-r/rrr get_adminsr2rec stt|4IdH@}|d|f4IdH}|IdHduWdIdHWdIdHS1IdHsr?rrrrstrr)rrirjrkrlrmr@rrrradd_mandatory_channels  .roc r))Nz4SELECT * FROM mandatory_channels WHERE is_active = 1c Ss6g|]}|d|d|d|d|d|ddqS)rr*r;r<r=)r7rirkrjrlrmrrCrrrrEs  z*get_mandatory_channels..r-r/rrrget_mandatory_channelss.rq channel_db_idc rM)Nz8UPDATE mandatory_channels SET is_active = 0 WHERE id = ?r)rrrrrrrremove_mandatory_channel s .rsc s:dD]\}}z ||IdHWqtyYqwdS)N))emojiz0ALTER TABLE custom_buttons ADD COLUMN emoji TEXT)custom_emoji_idz:ALTER TABLE custom_buttons ADD COLUMN custom_emoji_id TEXT)colorz0ALTER TABLE custom_buttons ADD COLUMN color TEXT) is_visiblezBALTER TABLE custom_buttons ADD COLUMN is_visible INTEGER DEFAULT 1)r Exception)rcolddlrrr_ensure_cbtn_columnss  r{btn_typelabelcontentc sddlm}tt|4IdH-}t|IdH|d||||fIdH}|IdH|j WdIdHS1IdHsEwYdS)Nr)rzINSERT INTO custom_buttons (btn_type, label, content, position, created_at, is_visible) VALUES (?, ?, ?, (SELECT COALESCE(MAX(position),0)+1 FROM custom_buttons), ?, 1)) rrrrr{rr>r?r lastrowid)rr|r}r~_dtrr%rrradd_custom_buttons  0rc stt|4IdHJ}t|IdH|d4IdH#}|IdH}dd|DWdIdHWdIdHS1IdHsFwYWdIdHdS1IdHs\wYdS)NzSELECT id, btn_type, label, content, position, emoji, custom_emoji_id, color, COALESCE(is_visible,1) FROM custom_buttons ORDER BY positionc SsHg|] }|d|d|d|d|d|d|d|d|dd qS) rr*r;r<r=rp r7r|r}r~positionrtrurvrwrrCrrrrE3s z&get_custom_buttons..)rrrr{rr.r/rrrget_custom_buttons+s .rbtn_idc s&tt|4IdHz}t|IdH|d|f4IdHQ}|IdH}|s> WdIdHWdIdHdS|d|d|d|d|d|d|d|d |d d WdIdHWdIdHS1IdHsvwYWdIdHdS1IdHswYdS) NzSELECT id, btn_type, label, content, position, emoji, custom_emoji_id, color, COALESCE(is_visible,1) FROM custom_buttons WHERE id = ?rr*r;r<r=rprrrr)rrrr{rr$)rrrr%rDrrrget_custom_button:s(  .r__keep__rtrurvrwc sztt|4IdH} t| IdHgg} } |dur'| d| ||dur5| d| ||durC| d| ||dkrQ| d| ||dkr_| d| ||dkrm| d| ||dur{| d| || s WdIdHdS| || d d | d | IdH| IdHWdIdHdS1IdHswYdS) Nz btn_type = ?z label = ?z content = ?rz emoji = ?zcustom_emoji_id = ?z color = ?zis_visible = ?zUPDATE custom_buttons SET z, z WHERE id = ?)rrrr{appendrr r) rrr|r}r~rtrurvrwrsetsvalsrrrupdate_custom_buttonIs2   .rc rM)Nz'DELETE FROM custom_buttons WHERE id = ?r)rrrrrrdelete_custom_buttonerPr)r)rh)NNNrrrN)*rrrrr r dirnameabspath__file__r rnrr!r'r(dictr1intrArBlistrFrJboolrLrOr_rbrcrdrerfrgrorqrsr{rrrrrrrrrsxI