Python:为SQL select和update语句创建一个类

时间:2017-08-31 11:01:44

标签: python pymysql

还有其他方法可以改进此代码吗?我正在尝试使用select和update方法创建一个类,然后我将其称为将来使用。

connect.py

import pymysql

class SQL:
    def __init__(self):
        self.cnx = pymysql.connect(host...)
        self.c = self.cnx.cursor()

    def select(self, sql):
        self.c.execute(sql)
        self.sel = self.c.fetchone()
        self.c.close()
        self.cnx.close()
        return self.sel

    def update(self, sql):
        self.c.execute(sql)
        self.upd = self.cnx.commit()
        self.c.close()
        self.cnx.close()
        return self.upd
from connect import SQL

stmt = "SELECT * FROM name;"
result = SQL().select(stmt)

1 个答案:

答案 0 :(得分:0)

在这个类中添加一个函数,函数close()。在main()的末尾使用close()。另外,在你的每个函数中使用try和except。您还可以添加要在每个函数中调用的重新启动函数。