o fjU@sUddlZddlZddlmZejejejedZe Z e e d<ddddd d ddd d ddd dddddddddddddddddddddddddddddddg Z e ee d<dZd ed!efd"d#Zd efd$d%Zd}d ed'ed(ed!efd)d*Zd ed'ed+efd,d-Zd ed!efd.d/Zd ed!e efd0d1Zd ed!e efd2d3Zd ed4efd5d6Zd ed7ed8edBd9ed!ef d:d;Zd ed7ed!edBfdd?Zd ed7ed!efd@dAZd ed7efdBdCZd ed!efdDdEZ d ed!e efdFdGZ!d ed7ed!efdHdIZ"d ed7ed8edBd9efdJdKZ#d ed7efdLdMZ$d ed!e efdNdOZ% Pd~d edQedRedSedBdTedBdUef dVdWZ&d edXefdYdZZ'd ed!e efd[d\Z(d edQedRedSedBdTedBdUed]efd^d_Z)d edXefd`daZ*d edXed7ed!efdbdcZ+d edXed!e,eeffdddeZ-d edXefdfdgZ.dhdiZ/d edjedkedled!ef dmdnZ0d ed!e efdodpZ1d edqed!edBfdrdsZ2   t dd edqedjedkedleduedvedwedxefdydzZ3d edqefd{d|Z4dS)Ndatetimedatabase _INIT_DONEzopenai/gpt-oss-120bz GPT-OSS 120Bgroqmodel_id model_nameplatformzopenai/gpt-oss-20bz GPT-OSS 20Bzqwen/qwen3.8-27bz Qwen 3.8 27Bz allam-2-7bz ALLaM 2 7Bz2nvidia/nemotron-3-nano-omni-30b-a3b-reasoning:freez Nemotron Omni openrouterzpoolside/laguna-xs.2:freez Laguna XS.2zmoonshotai/kimi-k2.6:freez Kimi K2.6zgoogle/gemma-4-26b-a4b-it:freez Gemma 4 26Bz!liquid/lfm-2.5-1.2b-thinking:freez LFM Thinkingz!liquid/lfm-2.5-1.2b-instruct:freez LFM Instructz#nvidia/nemotron-3-nano-30b-a3b:freez Nemotron 30BDEFAULT_MODELS)zllama-3.3-70b-versatilezllama-3.1-8b-instantzmixtral-8x7b-32768 bot_usernamereturncCs&tjtddtjt|dS)NT)exist_okz.db)osmakedirsDB_DIRpathjoinlower)r r%/var/www/kodo/factory/cobots/ai/db.py get_db_pathsrc sr|}|tvr dStt|4IdH}|dIdH|dIdH|dIdH|dIdH|dIdH|dIdH|dIdH|dIdHgd }|D]\}}|d ||fIdHq]tD]}|d |d |d |dfIdHqotD] }|d|fIdHq|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) ) z CREATE TABLE IF NOT EXISTS models ( model_id TEXT PRIMARY KEY, model_name TEXT NOT NULL, platform TEXT NOT NULL, is_enabled INTEGER DEFAULT 1 ) 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(🔴 البوت مقفول حالياً.)welcome_message)notify_new_usersr)notify_blockedr) fund_notifyrz9INSERT OR IGNORE INTO settings (key, value) VALUES (?, ?)z]INSERT OR IGNORE INTO models (model_id, model_name, platform, is_enabled) VALUES (?, ?, ?, 1)rr r z7DELETE FROM models WHERE model_id=? AND platform='groq') rr aiosqliteconnectrexecuter _OLD_GROQ_MODELScommitadd)r keydbdefaultsdkeyvaluem_oldrrrinit_db#s>        .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 sz$get_all_settings..r!r"rr#fetchallr r(r2rrrget_all_settingss.r@c r7)Nz=SELECT model_id, model_name, platform, is_enabled FROM modelscSs.g|]}|d|d|dt|ddqS)rr8)rr r is_enabled)boolr9rrr s zget_models..r=r?rrr get_modelss  .rFc r7)NzDSELECT model_id, model_name, platform FROM models WHERE is_enabled=1cS$g|]}|d|d|ddqS)rr8rArrr9rrrrEsz&get_enabled_models..r=r?rrrget_enabled_modelss  .rHrc ltt|4IdH}|d|fIdH|IdHWdIdHdS1IdHs/wYdS)Nz>UPDATE models SET is_enabled = 1 - is_enabled WHERE model_id=?r5)r rr(rrr toggle_models .rJuser_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) NHSELECT 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)idrLrM is_banned is_blockedis_newzAUPDATE users SET username=?,first_name=?,last_active=? WHERE id=?r8rArBF rutcnow isoformatr!r"rr#r1r%)r rKrLrMnowr(r2r3rrrget_or_create_users8  (     0rXc stt|4IdHg}|d|f4IdHE}|IdH}|s7 WdIdHWdIdHdS|d|d|d|d|ddWdIdHWdIdHS1IdHscwYWdIdHdS1IdHsywYdS)NrNrr8rArBrS)rOrLrMrPrQr0)r rKr(r2r3rrrget_users&  .rYc r7)NzKSELECT id,username,first_name FROM users WHERE is_banned=0 AND is_blocked=0cSrGrr8rA)rOrLrMrr9rrrrEz(get_all_active_users..r=r?rrrget_all_active_userss  .r\c 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 rKr(r2rrrban_users(0r^c rI)Nz'UPDATE users SET is_banned=0 WHERE id=?r5r rKr(rrr unban_user .r`c sNtt|4IdH}|d4IdH}|IdHd}WdIdHn 1IdHs0wY|d4IdH}|IdHd}WdIdHn 1IdHsXwY|d4IdH}|IdHd}WdIdHn 1IdHswY|||}WdIdHn 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(r2rbrcrdrerrrget_statistics s((((rfc r7)Nz)SELECT id,username,first_name FROM adminscSrGrZrr9rrrrEr[zget_admins..r=r?rrr get_adminss .rgc stt|4IdH@}|d|f4IdH}t|IdHWdIdHWdIdHS1IdHs.r=r?rrrget_mandatory_channels4s  .rrbuttonsrmrornrprqc 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,?)ri)r rmrornrprqrWr(rrradd_mandatory_channelAs  .rt channel_db_idc rI)Nz4UPDATE mandatory_channels SET is_active=0 WHERE id=?r5r rur(rrrremove_mandatory_channelO .rwc 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) rr8rArBrSrl)rOrmrnrorprq target_count current_countrr9rrrrE\s z'get_funded_channels..r=r?rrrget_funded_channelsVs  .r}r{c 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,?)ri) r rmrornrprqr{rWr(rrradd_funded_channelds  .r~c rI)Nz1UPDATE funded_channels SET is_active=0 WHERE id=?r5rvrrrremove_funded_channelsrxrc 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=?TrT)r rurKrWr(r2rrrrecord_funded_sub{s2  (  0rc stt|4IdHJ}|d|f4IdH(}|IdH}|r*|d|dfndWdIdHWdIdHS1IdHsFwYWdIdHdS1IdHs\wYdS)NzASELECT current_count,target_count FROM funded_channels WHERE id=?rr8)rrr0)r rur(r2r3rrrget_funded_channel_counts .rc szt}tt|4IdH}|d||fIdH|IdHWdIdHdS1IdHs6wYdS)NzAUPDATE funded_channels SET is_active=0, completed_at=? WHERE id=?ri)r rurWr(rrrmark_funded_completes  .rc 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)r(colddlrrr_ensure_cbtn_columnss  rbtn_typelabelcontentc sddlm}tt|4IdH-}t|IdH|d||||fIdH}|IdH|j WdIdHS1IdHsEwYdS)NrrzINSERT INTO custom_buttons (btn_type, label, content, position, created_at, is_visible) VALUES (?, ?, ?, (SELECT COALESCE(MAX(position),0)+1 FROM custom_buttons), ?, 1)) rr!r"rrr#rUrVr% lastrowid)r rrr_dtr(r2rrradd_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) rr8rArBrSrlryrz rOrrrpositionrrrrrr9rrrrEs z&get_custom_buttons..)r!r"rrr#r>)r r(r2rowsrrrget_custom_buttonss .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 = ?rr8rArBrSrlryrzrr)r!r"rrr#r1)r rr(r2r;rrrget_custom_buttons(  .r__keep__rrrrc 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 = ?)r!r"rrappendr#rr%) r rrrrrrrrr(setsvalsrrrupdate_custom_buttons2   .rc rI)Nz'DELETE FROM custom_buttons WHERE id = ?r5)r rr(rrrdelete_custom_buttonrar)r)rs)NNNrrrN)5rr!rrrdirnameabspath__file__rsetr__annotations__r listdictr$strrr.r4r6r@rFrHrJintrXrYr\rDr^r`rfrgrhrjrkrrrtrwr}r~rrtuplerrrrrrrrrrrrs             s