python - 解决循环程序中的“ERROR 2006 (HY000): MySQL server has gone away”（mysql-connector-python）

Question

我为最近遇到的类似问题提供了解决方案：

我有带有 MySQL 用户数据库的 Telegram (messenger) 机器人。在代码开始时存在与它的连接，导致连接在大约 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,)

如果您有一些优化建议，请写下它！