From 502c07a2c93e1e3facbf4b0b0502f6a9d59ff4b7 Mon Sep 17 00:00:00 2001 From: KatykhinAA Date: Sat, 16 Nov 2024 18:45:05 +0300 Subject: [PATCH] some fix --- database/db.py | 141 ++++++++++++++---- .../handlers/private/private_handlers.py | 46 +++--- .../__pycache__/__init__.cpython-312.pyc | Bin 264 -> 264 bytes .../base_dependency_factory.cpython-312.pyc | Bin 2272 -> 2272 bytes .../__pycache__/helper_func.cpython-312.pyc | Bin 17749 -> 21000 bytes .../utils/__pycache__/state.cpython-312.pyc | Bin 703 -> 703 bytes helper_bot/utils/helper_func.py | 53 +++++++ tests/test_db.py | 8 +- .../{voice_handler => handlers}/__init__.py | 0 .../voice_handler.py | 22 +-- voice_bot/main.py | 2 +- 11 files changed, 206 insertions(+), 66 deletions(-) rename voice_bot/{voice_handler => handlers}/__init__.py (100%) rename voice_bot/{voice_handler => handlers}/voice_handler.py (93%) diff --git a/database/db.py b/database/db.py index ce6621d..999b796 100644 --- a/database/db.py +++ b/database/db.py @@ -100,27 +100,8 @@ class BotDB: finally: self.close() - # TODO: Deprecated. Остался только в voice боте, удалить и оттуда - def get_error_message_from_db(self, id: int): - """ - @deprecated - Функция для запроса к базе данных и получения сообщений ошибки. В аргументы передаются: - id - идентификатор ошибки - """ - # Подключаемся к базе - try: - self.connect() - self.cursor.execute(f"SELECT * FROM error_messages WHERE id=?", (id,)) - response_from_database = str(self.cursor.fetchone()[1]) - return response_from_database - except sqlite3.Error as error: - self.logger.error(f"Ошибка при получении сообщения об ошибка voice_bot: {error}") - finally: - self.close() - def add_new_user_in_db(self, user_id: int, first_name: str, full_name: str, username: str, is_bot: bool, - language_code: str, date_added: str, - date_changed: str): + language_code: str, emoji: str, date_added: str, date_changed: str): """ Добавляет нового пользователя в базу данных. @@ -131,6 +112,7 @@ class BotDB: username (str): Username пользователя в Telegram. is_bot (bool): Флаг, указывающий, является ли пользователь ботом. language_code (str): Код языка пользователя. + emoji (str): Эмодзи закрепленная за пользователем date_added (str): Дата добавления пользователя в базу. date_changed (str): Дата последнего изменения данных пользователя. @@ -142,11 +124,12 @@ class BotDB: try: self.connect() self.cursor.execute("INSERT INTO 'our_users' ('user_id', 'first_name', 'full_name', 'username', 'is_bot', " - "'language_code', 'date_added', 'date_changed') VALUES (?, ?, ?, ?, ?, ?, ?, ?)", + "'language_code', 'emoji', 'date_added', 'date_changed') VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", (user_id, first_name, full_name, - username, is_bot, language_code, date_added, date_changed)) + username, is_bot, language_code, emoji, date_added, date_changed)) self.conn.commit() - self.logger.info(f"Новый пользователь добавлен в базу: user_id={user_id}, first_name={first_name}") + self.logger.info( + f"Новый пользователь добавлен в базу: user_id={user_id}, first_name={first_name}, emoji={emoji}") return None except sqlite3.Error as error: self.logger.error(f"Ошибка при добавлении пользователя в базу: {error}. " @@ -675,15 +658,15 @@ class BotDB: def update_date_for_user(self, date: str, user_id: int): """ #TODO: Не возвращается ошибка sqlite3. Error. Тест не перехватывает. Возвращается no such table: our_users - Обновляет дату последнего изменения данных пользователя в базе. + Обновляет дату последнего изменения данных пользователя в базе Args: - date (str): Новая дата изменения. - user_id (int): Идентификатор пользователя в Telegram. + date (str): Новая дата изменения + user_id (int): Идентификатор пользователя в Telegram Returns: - None: Если обновление прошло успешно. - sqlite3. Error: Если произошла ошибка при выполнении запроса. + None: Если обновление прошло успешно + sqlite3. Error: Если произошла ошибка при выполнении запроса """ self.logger.info(f"Запуск функции update_date_for_user: user_id={user_id}, date={date}") try: @@ -699,6 +682,106 @@ class BotDB: finally: self.close() + def check_emoji(self, emoji: str): + """ + Проверяет, есть ли уже такой emoji в таблице. + + Args: + emoji: emoji для проверки. + + Returns: + True, если эмодзи уже есть, иначе False. + + Raises: + None: В случае ошибки возвращается None + """ + self.logger.info(f"Запуск функции check_emoji: emoji={emoji}") + try: + self.connect() + self.cursor.execute("SELECT 1 FROM our_users WHERE emoji = ?", (emoji,)) + result = self.cursor.fetchone() + return bool(result) + except sqlite3.Error as error: + self.logger.error(f"Ошибка проверки эмодзи в базе: {error}") + return None + finally: + self.close() + + def update_emoji_for_user(self, user_id: int, emoji: str): + """ + Обновляет эмодзи для пользователя в базе если его ранее не было установлено + + Args: + user_id (int): Идентификатор пользователя в Telegram + emoji (str): Эмодзи пользователя + + Returns: + None: Если обновление прошло успешно + sqlite3. Error: Если произошла ошибка при выполнении запроса + """ + self.logger.info(f"Запуск функции update_date_for_user: user_id={user_id}, emoji={emoji}") + try: + self.connect() + self.cursor.execute("UPDATE our_users SET emoji = ? WHERE user_id = ?", + (emoji, user_id,)) + self.conn.commit() + self.logger.info(f"Эмоджи обновлен для пользователя: user_id={user_id}") + return None + except sqlite3.Error as error: + self.logger.error(f"Ошибка обновления эмодзи для пользователя: {error}") + return error + finally: + self.close() + + def check_emoji_for_user(self, user_id: int): + """ + Проверяет, есть ли уже у пользователя назначенный emoji. + + Args: + user_id: user_id пользователя. + + Returns: + True, если эмодзи такого нет, иначе False. + + Raises: + None: В случае ошибки возвращается None + """ + self.logger.info(f"Запуск функции check_emoji_for_user: user_id={user_id}") + try: + self.connect() + self.cursor.execute("SELECT emoji FROM our_users WHERE user_id = ?", (user_id,)) + result = self.cursor.fetchone()[0] + return result + except sqlite3.Error as error: + self.logger.error(f"Ошибка проверки эмодзи в базе: {error}") + return None + finally: + self.close() + + def refresh_listen_audio(self, user_id: int): + """ + Очищает всю информацию о прослушанных аудио пользователем + + Args: + user_id: user_id пользователя. + + Returns: + None - если все очищено успешно + + Raises: + error: В случае ошибки возвращается error + """ + self.logger.info(f"Запуск функции check_emoji_for_user: user_id={user_id}") + try: + self.connect() + self.cursor.execute("DELETE FROM listen_audio_users WHERE user_id = ?", (user_id,)) + return None + except sqlite3.Error as error: + self.logger.error(f"Ошибка проверки эмодзи в базе: {error}") + return error + finally: + self.close() + def is_admin(self, user_id: int): """ Проверяет, является ли пользователь администратором. @@ -719,7 +802,7 @@ class BotDB: result = self.cursor.fetchone() return bool(result) except sqlite3.Error as error: - self.logger.error(f"Ошибка добавления сообщения в базу данных: {error}") + self.logger.error(f"Ошибка проверки прав пользователя админа: {error}") return None finally: self.close() diff --git a/helper_bot/handlers/private/private_handlers.py b/helper_bot/handlers/private/private_handlers.py index d0ab731..22b2381 100644 --- a/helper_bot/handlers/private/private_handlers.py +++ b/helper_bot/handlers/private/private_handlers.py @@ -17,8 +17,9 @@ from helper_bot.middlewares.blacklist_middleware import BlacklistMiddleware from helper_bot.utils import messages from helper_bot.utils.base_dependency_factory import BaseDependencyFactory from helper_bot.utils.helper_func import get_first_name, get_text_message, send_text_message, send_photo_message, \ - send_media_group_message_to_private_chat, prepare_media_group_from_middlewares, check_username_and_full_name, \ - send_video_message, send_video_note_message, send_audio_message, send_voice_message, add_in_db_media + send_media_group_message_to_private_chat, prepare_media_group_from_middlewares, send_video_message, \ + send_video_note_message, send_audio_message, send_voice_message, add_in_db_media, \ + update_user_info, check_user_emoji from logs.custom_logger import logger private_router = Router() @@ -39,6 +40,24 @@ TEST = bdf.settings['Settings']['test'] BotDB = bdf.get_db() +@private_router.message( + ChatTypeFilter(chat_type=["private"]), + Command("restart") +) +async def handle_restart_message(message: types.Message, state: FSMContext): + try: + markup = get_reply_keyboard(BotDB, message.from_user.id) + await message.forward(chat_id=GROUP_FOR_LOGS) + await state.set_state("START") + await update_user_info('love', message) + check_user_emoji(message.from_user.id) + await message.answer('Я перезапущен!', reply_markup=markup, parse_mode='HTML') + except Exception as e: + logger.error(f"Произошла ошибка handle_restart_message. Ошибка:{str(e)}") + await message.bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка handle_restart_message: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") + + @private_router.message( ChatTypeFilter(chat_type=["private"]), Command("start") @@ -50,27 +69,8 @@ BotDB = bdf.get_db() async def handle_start_message(message: types.Message, state: FSMContext): try: await message.forward(chat_id=GROUP_FOR_LOGS) - full_name = message.from_user.full_name - username = message.from_user.username - first_name = get_first_name(message) - is_bot = message.from_user.is_bot - language_code = message.from_user.language_code - user_id = message.from_user.id - current_date = datetime.now() - date = current_date.strftime("%Y-%m-%d %H:%M:%S") - if not BotDB.user_exists(user_id): - BotDB.add_new_user_in_db(user_id, first_name, full_name, username, is_bot, language_code, date, - date) - else: - is_need_update = check_username_and_full_name(user_id, username, full_name) - if is_need_update: - BotDB.update_username_and_full_name(user_id, username, full_name) - await message.answer( - f"Давно не виделись! Вижу что ты изменился;) Теперь буду звать тебя: {full_name} и ник @{username}") - await message.bot.send_message(chat_id=GROUP_FOR_LOGS, - text=f'Для пользователя: {user_id} обновлены данные в БД.\nНовое имя: {full_name}\nНовый ник:{username}') - sleep(1) - BotDB.update_date_for_user(date, user_id) + await update_user_info('love', message) + check_user_emoji(message.from_user.id) await state.set_state("START") logger.info( f"Формирование приветственного сообщения для пользователя. Сообщение: {message.text} " diff --git a/helper_bot/keyboards/__pycache__/__init__.cpython-312.pyc b/helper_bot/keyboards/__pycache__/__init__.cpython-312.pyc index f1c797eadcdec538b8fae329838ecd1b76e03c34..3e943c6ec5ceb1923100a929e666e9b0ca7629b7 100644 GIT binary patch delta 20 acmeBR>R{qI&CAQh00i$OOw%Xw`~d(jh6PIi delta 20 acmeBR>R{qI&CAQh00isL&qrHcR4f#+QS{7)H$^1 zkK(>>RQThx}a zN9`#`)RA&Vos@1$x>D|_o8o23@{}j)VHvKR5#6&G5wvK9STVj`j$0~x<|_CWEC{1M zjuHK0*?5~=ZmnQDW@0A>}abR)?etro1JWqdd4hizi*DWhP2#u%*=9Ad5D zgr5s&CczD~`gO_naw_QoYEalERKOP}c%R{;A;I?y6KxRu02_sXxOH7;PbI`-!`Q%H z6=;Sje+y8ziM2?l(Su*$R|CIE*uAdZit*Gyq~F2;Y;GW53%(;3W}`-D9cX-$mRYC= zt!4`YdVAJAbp-#SUI_GBHhXJr_{R7eL1V{ejgGB<(N7q(S~qJIAXndzhbGWy6Sj+6 zVLl>|!*-#W;w^yRBkZ7fE8rbM8_dwoRmpasbRv$kE!;zApaW=K!cLIt7CM1SR|&(k zZB9K*rcvku(VkUP-GJ{BdO&UWBpd0KqlOSKB}H*UK4-{OnU9G{ag4-Mnsba~aucyr z84^oo#w7VShB=c9Y!koyl5v!;1k^7Z_`R%u;a>hfxymh+hoBmO1lx}Z27@FZA29p* zI{2!VA2Fxds)ditH9XrW|JG4=uN}x5FA|a=5uQ-%^++B?1JxzRkMicTuLPItVlXh0u7Na1l<5MLsCp@ruf8!m=;JGk}U|ZYDCI3 zKAt?8OObm}(v}j1L_9`^ud#^@VOe!fc5*_L&;$+2s1MTH%pU;KLkkn$a%1;SK!3&3 z7>4z5Fp`K_0~3-I9}{B_aUMvWKIGMMKTfO!Xt1Mv0^N%Bs=i z=Np^(W5AeOSZe&e!52a49t3o_dl2iCKWN&f=Msw;B+88BeF!l7`jWw!*UATBg$`%3 zgPB}f7{CQga50ls1T@fZ0~RsUQR+c{2U&EN&@5mU32s)+NEZ~k!s#_3fEk|H9^#J! z?d^rX-CoTd{}o5AfUx*P7N&!a`%zT46LBIr8B4{}m!Q4ynV;4XnYhEz^>Xr&G(PG_>2 zO`hmp+~5vWHm~H4)=GX9sQ)Dox4sO^9~|pzh5Dh}W}{;|TVHofm-i;eRPY-dgGc%9e8OF=}8Ms71AR@pmAkRho@{{dLdT^_of@TQuO}TJwFiUsb#$`&S_0)sj zSC?tvE2^+H^eH?8!f?r8B-z;&kpKCfi){B-QiYp!(;U&=`}I`Cb3$VC98tmlSE{J) z-Iy|%svrclAyrs6)>3thrUxl&3zk-uuZKhn)uzo%1^>@1$^YG1$rG@-Q?_>H*`$22 z>sl99FwLBhVsRmrNJHh7fFR*zNH378c#sVb7C#6_613ernMIaB!9lQ2|PHNXdj=dny?|Gjckk9t+IB)@V z_?#pXiI#26NO2>O_3ey&-rB*cLL3FK2#X)sFnTz6KlSnI8D$ zMcLmUfkLvszk&ZDP@j~?`gcRFuJ!k^?0Na$?mZs(0cu)^0d<~4kVh~e5AWY@!m34L za(;h3J0<^Ue~o_EkmRNP)m!L{ICE*p9iCu`Qwg{-{0tO|@}2!NTk%$^nWRjPjEbX} zxV2^j1haX^Xppr~z1&vgWjVR7Eg;Xjy#YfOD>-d>cJ1OlWu*MyG++xbtszU+r}G(Rb`_bjX8mil@hW+~ zx*|}v=3lDYz^g{yP+8uX=kkU;pEu>Lc}w0hi#Hc&ER(*Nt<`ac$(Ml)H;c)IHf77( zNNR(vdbCx$Nrq+e@C-TwrxB^iv*RI**#b6RVt&lcvhq)M_~A7Kj0XBZ3Cmw>t!Ub& zn}?NU%$L2y2t0}lrn~`QS>6h&=2y(K94c>RnJk{NvUcP)r;y=Af`U0J{JVio66fD_6}f$8zA}o-bhgS z8BpF;URR6ibWoXBpH^N2J&=4=nFFj%?KhHL5a;fRC8nDBJs2U6-Z2F#uIuGvC--yJ^@8c(mYr&)8aY3q|mD%!7`xL z<|eRy%3t&a3UC2WOXpzo>bo7;&Y_2oe&<`U!J`ky4j&yFA$Zx>yt)o8yz~Wvo+5Vg zcNiSK?5&Rm7Zdq=1iwH)3(^!|ni1VIx-CSapPpgUjCPk~ge+!zFL0>g_BZS$6tL&U)2ZUvf66 z&W7U9t%~X^`)2mZ{l&w}wm``iRBb^e)OFL=T|D%8dFW<&WAVYyZSIn-Mzz&k{qD!M z@GY~WWDct4;I+WZ&2x$ODnIN|MjlsACY3Z<4u${6z*-7TUz(ULJ#%A=PkdN)vufm$ z^IvUMx9sI5d%J3HpDSCed~eUk_9M$K-_^jsxPr_6+AHaq^a{rWhPf3uq`2>4H_RQ! z&KKqG7}&rG?hg!GegY^6Kl4;wX_;v$d74#E^K1RfJ4cjbCzO##K>ktoL;>|4;Y93C zaHp_4#hu6QN$#Kly7Sy2><$?o!!Bk>VE33|9J`J)h7|)-(Q=1jE0P9q>PP|^;pb+@ zrTDb@meEo&)~Uw2tJ!Pc{lwV1V#A33xMD<^KS^5|Cd@o_sBVy9-w(0_`;700x&Z&c z#}4!xKkzpp-p>M*3*CM;E6?<`S4s_8T#vNft=?>wk=q(-3s4$g;WpN-NjR*B37(;F z%;GwPr!B5KxRlZRiC4daP}<#9O@4`jzSSkSnYU_&{3D1+SRhxQz?vH@72rCvzP8C5 zz{V0QbpldkZsFPptG4jVV~Q>Ov90HpFMQM2bjeDWNuz3Re0gZGS1~t!Vjf)K8T&4R zr|2mB?lvp}_`CBV0Qf77ozl+>2!sq~+4X~m`b9M;?WllF?lAiNS(E;uM~3(mn>EV` zYt6Rx6IbH$>DIbNo%-z0qZk8M*zJJwG@R5*Wo1jzFy?t zRiDE%M7iDuPkDZVB+^;Uq!&ibGEJ(owGZG&mNbb8u#%U6Bn(V0ho-h0#626OB3>NV)v zmXBRKmMdx%d+oB>efhv&IWC!3Y>chq^2l>-U)mYocJc6q!?3A5hqx~ph9BUTZ7#Y3 zBCN&`54eN1_bzn9k?-vx6@ZX1g6jxyj)@TfJVq1nmkVwYvVh-Z@k9pyJfg2<&9VB` zM&GykpDy%4jrSK~m2V$tV{NkizRCjDVa;$-IHj2-F`G@K$0YI!(u|l!%{p-8=z|Z$ z4jnjlKr@~ZCnfSWGEI0a!Ht+^x$KDkP|^%=rKU$5sY5p{2o4~400I6lp&4M_$T7tD zY;Gbca^$#NxUZ`4I7*BnfJY!h&LDs{4ny#{K_3mIh#213$z=pG0zBsEb(EeHbSLT) zh+Bkam7XhqGVe=egj`bWCkilNuQJj+Y^D_h%d(#`wohG*a^2_0pr$O~JlsrwUr|G6Aa?{aVv@UEt*u-`H t-fCm_uGDN{+pcx4Fwo8U-mIOky@TjVuYnz6ua4efu)kxsux=XR{{p=#8O{Iz delta 3404 zcmaJ^Yit}>6`s30voG80*LrQQ9cS%qVvp^(cG4s!ek68Ds2e#^DgnuW*o^m%ovn9v zmOHzluBbLb0ZLQ|?NwDIP^nOV@DzfT!XG45iGLtbDMd&VP(h*<;!i91Q-OG#b7s5Q z&0{U?x#ymH@44T3-th}>u;1Tg@!xcH#T57p-i`0Pee$LFP<1Y-D6cA9;jEw-{>Tgl z^FSeB2aJFnG=dD}!GdaQhDN+v2-#sHY)6cU9W|nM%!rX(D|FbMMkn#1LYEyk;!FuB zR_BlKQus5vt?p;WeV(YC)gtoxa=2>r^01{nlWAJ3vpiz;Z>v0dLp2gSX7%$9_;rFT z#JfN?@W^vpxKV_F;Fwn7Rvs1w(vBZKhXC4`28U6-A_OO!~`hGv{iDTEr<-9 zKgbVSDTw+Z2+birN&FP>lUyf$8u%$b19>o`*SF4sXc~D;F@BgLI|s5E{upeV-DCPu z*`tojlj^SgmHN|AC*T$RvaIE_0bqONAGIacUmFVjJ#ci8){qPVaiP+bbUG~(FnT>Z zRykX?o-Jo>%XQ5iON@YcKpu-0S+e#nmP%HU zix7$Bb-XfWBEEP`JconOUTr~w1!E3rnWKAs~&p{ z*dH*u>%4z0-l#{sSl#k$o{JM8YQi)nZw(({3$ivmn_kAHhTUYbi{Qvy@TUCVa9TSH z0=Je)y{T34K5-gI(`}x1&B^C1-^|#xJhz-|(J5Pctoc$rxQ^rrBx^{>IZq;o9ugZs z^q}}S^5ncUaIJxrJl(531>zTL&!zVTLQBBrg7WlyqV}_iMOHlrbAj;?2vm;>Y;@NgfV3V`R(a!3qPOJ2eYeS~PieVO6OtC?x- zA}su{_G;!I!GS}#`zR7J^*D0#^0lc$Hu-TA=HR3-ia&+~6U|QxjDgpC7P9cXQ$Fid ziu?>FslbF4(?|&DdEoR2S?Pm(8keXv$O~zl&d3FQ5K?DDAJ;wwg4;Dm9|ZgE_I}r6pnFFlXODb9l@v|U*TP9|2 zvw8W}++9CnWD6FJuKG7L1j0aRz>i~$)RnF}Wv2xSm0K&GmLx_fefQ2bnnG5A_wzkD zIrn|21^~m!1#~x6Q8O3{;DcZ&+grfU#x`JN_xwSc`Q}qIG_-vEV;SH!A)I@1=gyQg^iRp{9bidT{AfBe+Fam<1573(Vh? z53ehKmfv3JlK*|&WLM>tW3OMpZsbMsZr0>>z6f2^0f7rgl=?H$nY(J`cC%(K2d{`g zJCf-sfM6!0__({gf^5%&dptnyfLdm^v<>n^_ z;}AIXB)!YxoAUit9Xv67YC?Mjq!oGk)FOCd@6-xo*W@owUmpAtDv1hw!NnJlJd0#q zPOeRbGRS^LesXR9r9EIht?Y1f&+D!fw+c>f*W%fGzN1^cC>dTf;r+wLie7|2|GOX# z$IK!=PWkOTy!x-fhUesO*Iv4ee$bhxh;C#8;=3@M*AI&x7{vqgW+X6_x+qoU7u$1zuml*>UG`6}RlzS>WNTgxa|m+~V6_2tQfi z5u&6FZ~C7tUeGNIF@TEj1IGXIz}KH1CV>y37t9ySo?6Jm&WkwL%9Rp)yI`aD`k%H+ zg}rlCYj4Xj1>ZEqZlxr!1qc1k(w!Ijr^{z7fj&kXaa_L>=sXr+P{}LLx W6WVt@So_I^$w2I%u@1IGD*q4E47<_* diff --git a/helper_bot/utils/__pycache__/state.cpython-312.pyc b/helper_bot/utils/__pycache__/state.cpython-312.pyc index 9b45a9df5b9b93a8cdba9ec41109cf56a8fb9ca7..79328a961a69e909d0a839819b4ebd2c75f0403c 100644 GIT binary patch delta 21 bcmdnbx}TNjG%qg~0}#BEFiqdcvzZA1J{len(check_audio)}', reply_markup=markup) await state.set_state('START') diff --git a/voice_bot/main.py b/voice_bot/main.py index 199d2f7..c2990f2 100644 --- a/voice_bot/main.py +++ b/voice_bot/main.py @@ -3,7 +3,7 @@ from aiogram.client.default import DefaultBotProperties from aiogram.fsm.storage.memory import MemoryStorage from aiogram.fsm.strategy import FSMStrategy -from voice_bot.voice_handler.voice_handler import voice_router +from voice_bot.handlers.voice_handler import voice_router async def start_bot(bdf):