浏览代码

Выпуск в предварительное тестирование

root 2 年之前
父节点
当前提交
d6b28bea09
共有 4 个文件被更改,包括 14 次插入10 次删除
  1. 8 6
      bot/database.py
  2. 1 1
      handlers/start.py
  3. 5 1
      main.py
  4. 0 2
      requirements.txt

+ 8 - 6
bot/database.py

@@ -4,12 +4,14 @@ from bot import config as conf
 
 class Database:
     def __init__(self):
-        self.connection = mysql.connector.connect(host=conf.DB_HOST,
-                                                  user=conf.DB_USER,
-                                                  password=conf.DB_PASS,
-                                                  database=conf.DB_NAME)
-        self.cursor = self.connection.cursor(dictionary=True)
-
+        try:
+            self.connection = mysql.connector.connect(host=conf.DB_HOST,
+                                                      user=conf.DB_USER,
+                                                      password=conf.DB_PASS,
+                                                      database=conf.DB_NAME)
+            self.cursor = self.connection.cursor(dictionary=True)
+        except mysql.connector.Error as err:
+            print("Something went wrong: {}".format(err))
     def cursor(self):
         return self.cursor
 

+ 1 - 1
handlers/start.py

@@ -50,7 +50,7 @@ async def get_report(message: types.Message, state: FSMContext):
     data = await state.get_data()
     await state.finish()
     await new_bug(data)
-
+    await message.delete()
 
 def main_register(dp: Dispatcher):
     dp.register_message_handler(start_cmd, commands=['start', 'help'])

+ 5 - 1
main.py

@@ -4,7 +4,11 @@ import handlers.admin.admin_menu
 import logging
 
 if __name__ == '__main__':
-    logging.basicConfig(level=logging.INFO)
+    logging.basicConfig(handlers=[logging.FileHandler(filename="./log_records.txt",
+                                                      encoding='utf-8', mode='a+')],
+                        format="%(asctime)s %(name)s:%(levelname)s:%(message)s",
+                        datefmt="%F %A %T",
+                        level=logging.DEBUG)
     handlers.user.registration.register_handlers(dp)
     handlers.start.main_register(dp)
     handlers.user.make_events.events_register(dp)

+ 0 - 2
requirements.txt

@@ -1,6 +1,4 @@
 intervaltree
-mysql
-dotenv
 aiogram==2.22.2
 aiohttp==3.8.3
 aiosignal==1.2.0