Python 3.x 連接數(shù)據(jù)庫示例（pymysql 方式）

更新時間：2017年01月19日 11:08:02 作者：woider

這篇文章主要介紹了Python 3.x 連接數(shù)據(jù)庫示例（pymysql 方式）,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

由于 MySQLdb 模塊還不支持 Python3.x，所以 Python3.x 如果想連接MySQL需要安裝 pymysql 模塊。

pymysql 模塊可以通過 pip 安裝。但如果你使用的是 pycharm IDE，則可以使用 project python 安裝第三方模塊。

[File] >> [settings] >> [Project: python] >> [Project Interpreter] >> [Install按鈕]

由于Python統(tǒng)一了數(shù)據(jù)庫連接的接口，所以 pymysql 和 MySQLdb 在使用方式上是類似的：

pymysql.Connect()參數(shù)說明

host(str): MySQL服務(wù)器地址
port(int): MySQL服務(wù)器端口號
user(str): 用戶名
passwd(str): 密碼
db(str): 數(shù)據(jù)庫名稱
charset(str): 連接編碼

connection對象支持的方法

cursor() 使用該連接創(chuàng)建并返回游標(biāo)
commit() 提交當(dāng)前事務(wù)
rollback() 回滾當(dāng)前事務(wù)
close() 關(guān)閉連接

cursor對象支持的方法

execute(op) 執(zhí)行一個數(shù)據(jù)庫的查詢命令
fetchone() 取得結(jié)果集的下一行
fetchmany(size) 獲取結(jié)果集的下幾行
fetchall() 獲取結(jié)果集中的所有行
rowcount() 返回數(shù)據(jù)條數(shù)或影響行數(shù)
close() 關(guān)閉游標(biāo)對象

==================MySQL===================

首先在連接數(shù)據(jù)庫之前，先創(chuàng)建一個交易表，方便測試 pymysql 的功能：

DROP TABLE IF EXISTS `trade`;

CREATE TABLE `trade` (
 `id` int(4) unsigned NOT NULL AUTO_INCREMENT,
 `name` varchar(6) NOT NULL COMMENT '用戶真實姓名',
 `account` varchar(11) NOT NULL COMMENT '銀行儲蓄賬號',
 `saving` decimal(8,2) unsigned NOT NULL DEFAULT '0.00' COMMENT '賬戶儲蓄金額',
 `expend` decimal(8,2) unsigned NOT NULL DEFAULT '0.00' COMMENT '賬戶支出總計',
 `income` decimal(8,2) unsigned NOT NULL DEFAULT '0.00' COMMENT '賬戶收入總計',
 PRIMARY KEY (`id`),
 UNIQUE KEY `name_UNIQUE` (`name`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;
INSERT INTO `trade` VALUES (1,'喬布斯','18012345678',0.00,0.00,0.00);

==================Python===================

使用Python腳本實現(xiàn)增刪改查和事務(wù)處理，源碼如下：

import pymysql.cursors

# 連接數(shù)據(jù)庫
connect = pymysql.Connect(
  host='localhost',
  port=3310,
  user='woider',
  passwd='3243',
  db='python',
  charset='utf8'
)

# 獲取游標(biāo)
cursor = connect.cursor()

# 插入數(shù)據(jù)
sql = "INSERT INTO trade (name, account, saving) VALUES ( '%s', '%s', %.2f )"
data = ('雷軍', '13512345678', 10000)
cursor.execute(sql % data)
connect.commit()
print('成功插入', cursor.rowcount, '條數(shù)據(jù)')

# 修改數(shù)據(jù)
sql = "UPDATE trade SET saving = %.2f WHERE account = '%s' "
data = (8888, '13512345678')
cursor.execute(sql % data)
connect.commit()
print('成功修改', cursor.rowcount, '條數(shù)據(jù)')

# 查詢數(shù)據(jù)
sql = "SELECT name,saving FROM trade WHERE account = '%s' "
data = ('13512345678',)
cursor.execute(sql % data)
for row in cursor.fetchall():
  print("Name:%s\tSaving:%.2f" % row)
print('共查找出', cursor.rowcount, '條數(shù)據(jù)')

# 刪除數(shù)據(jù)
sql = "DELETE FROM trade WHERE account = '%s' LIMIT %d"
data = ('13512345678', 1)
cursor.execute(sql % data)
connect.commit()
print('成功刪除', cursor.rowcount, '條數(shù)據(jù)')

# 事務(wù)處理
sql_1 = "UPDATE trade SET saving = saving + 1000 WHERE account = '18012345678' "
sql_2 = "UPDATE trade SET expend = expend + 1000 WHERE account = '18012345678' "
sql_3 = "UPDATE trade SET income = income + 2000 WHERE account = '18012345678' "

try:
  cursor.execute(sql_1) # 儲蓄增加1000
  cursor.execute(sql_2) # 支出增加1000
  cursor.execute(sql_3) # 收入增加2000
except Exception as e:
  connect.rollback() # 事務(wù)回滾
  print('事務(wù)處理失敗', e)
else:
  connect.commit() # 事務(wù)提交
  print('事務(wù)處理成功', cursor.rowcount)

# 關(guān)閉連接
cursor.close()
connect.close()

==================測試結(jié)果===================