Решение "ОШИБКА 2006 (HY000): сервер MySQL ушел" в циклических программах (mysql-connector-python)

Question

Я предоставил решение аналогичных проблем с той, которую недавно встретил:

У меня есть бот Telegram (мессенджер) с пользовательской БД MySQL. Связь с ним есть в начале кода, из-за чего соединение пропало примерно через 10 часов. Поэтому бот возвращал ошибки, потому что не мог получить информацию о пользователе.

я использую mysql-connector-pythonфреймворк. Решение - использовать класс для запросов к БД. Вы можете увидеть это и примеры использования ниже.

Answer 1

Класс:

import logging

import mysql.connector

class DB():
    def __init__(self, **kwargs):
        self.conn = mysql.connector.connect(
            host="host",
            user="user",
            passwd="password",
            database="name"
        )
        try:
            self.cursor = self.conn.cursor()
        except Exception as e:
            print(str(e))

    def execute(self, query, data=None, ret1=False):
        try:
            if not self.conn:
                self.__init__()
            else:
                if data:
                    self.cursor.execute(query, data)
                else:
                    self.cursor.execute(query)

                if 'INSERT' in query or 'UPDATE' in query or 'DELETE' in query or 'DROP' in query:
                    self.conn.commit()

                if ret1:
                    return self.cursor.fetchone()
                else:
                    return self.cursor.fetchall()

        except:
            logging.error('end', exc_info=True)
            return 'error'

Пример таблицы:

Пример таблицы

Структура запроса:

res = DB().execute(query, protected data) # fetchall
res = DB().execute(query, protected data, True) # fetchone

Примеры использования:

> DB().execute("CREATE TABLE users (r_id INT AUTO_INCREMENT PRIMARY KEY, id INT UNIQUE, name VARCHAR(255), role INT NULL DEFAULT 3)")
> DB().execute("INSERT INTO users (id, name, role) VALUES (%s, %s, %s)", (65453, "Mike", 1,))
> res = DB().execute(f"SELECT * FROM users")
res = [(1, 146, "Nick", 3,), (2, 155, "John", 1,), (3, 678, "Michelle", 2,)]
> res = DB().execute(f"SELECT * FROM users WHERE name = %s", ("John",), ret1=True)
res = (2, 155, "John", 1,)

Если у вас есть предложения по оптимизации, напишите их!