安裝pymysql

pip install pymysql

2|0使用pymysql

2|1使用資料查詢語句

查詢一條資料fetchone()

from pymysql import *

conn = connect(
  host='127.0.0.1',port=3306,user='root',password='123456',database='itcast',charset='utf8')

# 建立遊標
c = conn.cursor()
# 執行sql語句
c.execute("select * from student")
# 查詢一行資料
result = c.fetchone()
print(result)
# 關閉遊標
c.close()
# 關閉資料庫連線
conn.close()
"""
(1,'張三',18,b'\x01')
"""

查詢多條資料fetchall()

from pymysql import *

conn = connect(
  host='127.0.0.1',charset='utf8')

# 建立遊標
c = conn.cursor()
# 執行sql語句
c.execute("select * from student")
# 查詢多行資料
result = c.fetchall()
for item in result:
  print(item)
# 關閉遊標
c.close()
# 關閉資料庫連線
conn.close()
"""
(1,b'\x01')
(2,'李四',19,b'\x00')
(3,'王五',20,b'\x01')
"""

更改遊標的預設設定，返回值為字典

from pymysql import *

conn = connect(
  host='127.0.0.1',charset='utf8')

# 建立遊標，操作設定為字典型別
c = conn.cursor(cursors.DictCursor)
# 執行sql語句
c.execute("select * from student")
# 查詢多行資料
result = c.fetchall()
for item in result:
  print(item)
# 關閉遊標
c.close()
# 關閉資料庫連線
conn.close()
"""
{'id': 1,'name': '張三','age': 18,'sex': b'\x01'}
{'id': 2,'name': '李四','age': 19,'sex': b'\x00'}
{'id': 3,'name': '王五','age': 20,'sex': b'\x01'}
"""

返回一條資料時也是一樣的。返回字典或者時元組看個人需要。

2|2使用資料操作語句

執行增加、刪除、更新語句的操作其實是一樣的。只寫一個作為示範。

from pymysql import *

conn = connect(
  host='127.0.0.1',charset='utf8')

# 建立遊標
c = conn.cursor()
# 執行sql語句
c.execute("insert into student(name,age,sex) values (%s,%s,%s)",("小二",28,1))
# 提交事務
conn.commit()
# 關閉遊標
c.close()
# 關閉資料庫連線
conn.close()

和查詢語句不同的是必須使用commit()提交事務，否則操作就是無效的。

3|0編寫資料庫連線類

普通版

MysqlHelper.py

from pymysql import connect,cursors

class MysqlHelper:
  def __init__(self,host="127.0.0.1",user="root",password="123456",database="itcast",charset='utf8',port=3306):
    self.host = host
    self.port = port
    self.user = user
    self.password = password
    self.database = database
    self.charset = charset
    self._conn = None
    self._cursor = None

  def _open(self):
    # print("連線已開啟")
    self._conn = connect(host=self.host,port=self.port,user=self.user,password=self.password,database=self.database,charset=self.charset)
    self._cursor = self._conn.cursor(cursors.DictCursor)

  def _close(self):
    # print("連線已關閉")
    self._cursor.close()
    self._conn.close()

  def one(self,sql,params=None):
    result: tuple = None
    try:
      self._open()
      self._cursor.execute(sql,params)
      result = self._cursor.fetchone()
    except Exception as e:
      print(e)
    finally:
      self._close()
    return result

  def all(self,params)
      result = self._cursor.fetchall()
    except Exception as e:
      print(e)
    finally:
      self._close()
    return result

  def exe(self,params=None):
    try:
      self._open()
      self._cursor.execute(sql,params)
      self._conn.commit()
    except Exception as e:
      print(e)
    finally:
      self._close()

該類封裝了fetchone、fetchall、execute，省去了資料庫連線的開啟和關閉和遊標的開啟和關閉。
下面的程式碼是呼叫該類的小示例：

from MysqlHelper import *

mysqlhelper = MysqlHelper()
ret = mysqlhelper.all("select * from student")
for item in ret:
  print(item)
"""
{'id': 1,'sex': b'\x01'}
{'id': 5,'name': '小二','age': 28,'sex': b'\x01'}
{'id': 6,'name': '娃哈哈','sex': b'\x01'}
{'id': 7,'sex': b'\x01'}
"""
上下文管理器版
mysql_with.py

from pymysql import connect,cursors

class DB:
  def __init__(self,host='localhost',db='itcast',passwd='123456',charset='utf8'):
    # 建立連線
    self.conn = connect(
      host=host,port=port,db=db,user=user,passwd=passwd,charset=charset)
    # 建立遊標，操作設定為字典型別
    self.cur = self.conn.cursor(cursor=cursors.DictCursor)

  def __enter__(self):
    # 返回遊標
    return self.cur

  def __exit__(self,exc_type,exc_val,exc_tb):
    # 提交資料庫並執行
    self.conn.commit()
    # 關閉遊標
    self.cur.close()
    # 關閉資料庫連線
    self.conn.close()

如何使用：

from mysql_with import DB

with DB() as db:
  db.execute("select * from student")
  ret = db.fetchone()
  print(ret)

"""
{'id': 1,'sex': b'\x01'}
"""

總結

以上所述是小編給大家介紹的使用python連線mysql資料庫之pymysql模組的使用，希望對大家有所幫助，如果大家有任何疑問歡迎給我留言，小編會及時回覆大家的！