首頁 > 軟體

Python pymysql操作MySQL詳細

2021-09-26 19:02:13

1、使用

1.1 簡單使用

import pymysql

# 建立連線
con = pymysql.connect(
                        host='localhost',
                        port=3306,
                        user='root',
                        password='123456',
                        database='test',
                        charset='utf8'
)
# 建立遊標
cursor = con.cursor()

# 執行新增SQL,返回受影響行數
row1 = cursor.execute("insert into user (username, password) values ('username3','password3')")
print(row1)

# 執行更新SQL,返回受影響行數
row2 = cursor.execute("update user set password = '123456' where id > 2;")
# 執行查詢SQL
res = cursor.execute("SELECT * FROM user;")
result = cursor.fetchall()
for info in result:
    print(info[0], info[1])
# 提交,不然無法儲存新增或者更新的資料
con.commit()
# 關閉遊標
cursor.close()
# 關閉連線
con.close()

注意:資料庫表中存在中文時,建立連線需要指定charset='utf8',否則中文顯示會亂碼。 其中cursor.fetchall()是獲取所有結果集,如果只有一條結果集,可以使用cursor.fetchone()

1.2 封裝工具類

為了使用方便,可以直接封裝成工具類:

import pymysql


class MysqlHelper:
    def __init__(self, config):
        self.host = config["host"]
        self.port = config["port"]
        self.user = config["user"]
        self.password = config["password"]
        self.db = config["db"]
        self.charset = config["charset"]
        self.con = None
        self.cursor = None

    def create_con(self):
        """
        建立連線
        """
        try:
            self.con = pymysql.connect(host=self.host, port=self.port, user=self.user, password=self.password,
                                       database=self.db, charset='utf8')
            self.cursor = self.con.cursor()
            return True
        except Exception as e:
            print(e)
            return False

    #
    def close_con(self):
        """
        關閉連結
        """
        if self.cursor:
            self.cursor.close()
        if self.con:
            self.con.close()

    # sql執行
    def execute_sql(self, sql):
        """
        執行插入/更新/刪除語句
        """
        try:
            self.create_con()
            print(sql)
            self.cursor.execute(sql)
            self.con.commit()
        except Exception as e:
            print(e)
        finally:
            self.close_con()

    def select(self, sql, *args):
        """
        執行查詢語句
        """
        try:
            self.create_con()
            print(sql)
            self.cursor.execute(sql, args)
            res = self.cursor.fetchall()
            return res
        except Exception as e:
            print(e)
            return False
        finally:
            self.close_con()

使用工具類:

config = {
    "host": 'localhost',
    "port": 3306,
    "user": 'root',
    "password": '123456',
    "db": 'test',
    "charset": 'utf8'
}
db = MysqlHelper(config)
db.execute_sql("insert into user (username, password) values ('username4','password4')")
db.select("SELECT * FROM user;")

到此這篇關於Python pymysql操作MySQL詳細的文章就介紹到這了,更多相關Python pymysql操作MySQL內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com