python链接MySQL

import pymysql

class Mydb:
    def __init__(self):
        # 链接数据库
        self.db = pymysql.connect(host='localhost',user='root',
                                  password='123456',port=3306,
                                  database='demo',
                                  cursorclass=pymysql.cursors.DictCursor)
        # 创建游标对象
        self.cursor = self.db.cursor()
    # 查询
    def query(self,sql):
        # 执行sql
        self.cursor.execute(sql)
        data = self.cursor.fetchall()
        return data

    # 增 删 改
    def alter(self,sql):
        # 执行
        self.cursor.execute(sql)
        num = self.cursor.rowcount
        self.db.commit()
        return num

    def __del__(self):
        # 关闭数据库
        self.cursor.close()
        self.db.close()

if __name__ == '__main__':
    # 实例化对象
    db = Mydb()
    sql = 'select version()'
    print(db.query(sql))