Python連接MySQL數據庫

Python標准數據庫接口為Python DB-API, Python DB-API為開發人員提供了數據庫應用 編程接口。

Python 數據庫接口支持非常多的數據庫,你可以選擇適合你項目的數據庫:

• GadFly
• mSQL
• MySQL
• PostgreSQL
• Microsoft SQL Server 2000 Informix
• Interbase Oracle Sybase

你可以訪問Python數據庫接口及API查看詳細的支持數據庫列表。

不同的數據庫你需要下載不同的DB API模塊,例如你需要訪問Oracle數據庫和Mysql數據,你需要下載Oracle和MySQL數據庫模塊。

DB-API是一個規范。它定義了一系列必須的對象和數據庫存取方式, 以便為各種各樣的底層數據庫系統和多種多樣的數據庫接口程序提供一致的訪問接口。

Python的DB-API,為大多數的數據庫實現了接口,使用它連接各數據庫后,就可以用相同 的方式操作各數據庫。

Python DB-API使用流程:

1. 引入API模塊。
2. 獲取與數據庫的連接。
3. 執行SQL語句和存儲過程。
4. 關閉數據庫連接。

1. MySQLdb

MySQLdb是用於Python鏈接Mysql數據庫的接口，它實現了Python 數據庫API規范V2.0，基於MySQL C API上建立的。

安裝

直接使用pip進行安裝，在此之前需要安裝一些系統依賴包。

• CentOS

  yum install gcc python-devel mysql-devel zlib-devel openssl-devel

• Ubuntu

  sudo apt-get install libmysqlclient-dev libmysqld-dev python-dev python-setuptools

安裝完依賴，直接使用pip安裝，MySQLdb模塊的名字在pip上叫MySQL-python。

pip install MySQL-python

常用函數

Python DB API 2.0 對事務提供了兩個方法：

• commit() 提交
• rollback() 回滾

cursor用來執行命令的方法:

• callproc(self, procname, args) 用來執行存儲過程,接收的參數為存儲過程名和參數列表,返回值為受影響的行數
• execute(self, query, args) 執行單條sql語句,接收的參數為sql語句本身和使用的參數列表,返回值為受影響的行數
• executemany(self, query, args) 執行單挑sql語句,但是重復執行參數列表里的參數,返回值為受影響的行數
• nextset(self) 移動到下一個結果集

cursor用來接收返回值的方法:

• fetchall(self) 接收全部的返回結果行.
• fetchmany(self, size=None) 接收size條返回結果行.如果size的值大於返回的結果行的數量,則會返回cursor.arraysize條數據.
• fetchone(self) 返回一條結果行.
• rowcount 這是一個只讀屬性,並返回執行execute() 方法后影響的行數。
• scroll(self, value, mode='relative') 移動指針到某一行; 如果mode='relative',則表示從當前所在行移動value條,如果 mode='absolute',則表示從結果集的第一行移動value條.

實例

#!/usr/bin/env python # -*- coding: utf-8 -*- import MySQLdb as mdb # 連接數據庫 conn = mdb.connect('localhost', 'root', 'root') # 也可以使用關鍵字參數 conn = mdb.connect(host='127.0.0.1', port=3306, user='root', passwd='root', db='test', charset='utf8') # 也可以使用字典進行連接參數的管理 config = { 'host': '127.0.0.1', 'port': 3306, 'user': 'root', 'passwd': 'root', 'db': 'test', 'charset': 'utf8' } conn = mdb.connect(**config) # 如果使用事務引擎，可以設置自動提交事務，或者在每次操作完成后手動提交事務conn.commit() conn.autocommit(1) # conn.autocommit(True) # 使用cursor()方法獲取操作游標 cursor = conn.cursor() # 因該模塊底層其實是調用CAPI的，所以，需要先得到當前指向數據庫的指針。 try: # 創建數據庫 DB_NAME = 'test' cursor.execute('DROP DATABASE IF EXISTS %s' %DB_NAME) cursor.execute('CREATE DATABASE IF NOT EXISTS %s' %DB_NAME) conn.select_db(DB_NAME) #創建表 TABLE_NAME = 'user' cursor.execute('CREATE TABLE %s(id int primary key,name varchar(30))' %TABLE_NAME) # 插入單條數據 sql = 'INSERT INTO user values("%d","%s")' %(1,"jack") # 不建議直接拼接sql，占位符方面可能會出問題，execute提供了直接傳值 value = [2,'John'] cursor.execute('INSERT INTO test values(%s,%s)',value) # 批量插入數據 values = [] for i in range(3, 20): values.append((i,'kk'+str(i))) cursor.executemany('INSERT INTO user values(%s,%s)',values) # 查詢數據條目 count = cursor.execute('SELECT * FROM %s' %TABLE_NAME) print 'total records: %d' %count print 'total records:', cursor.rowcount # 獲取表名信息 desc = cursor.description print "%s %3s" % (desc[0][0], desc[1][0]) # 查詢一條記錄 print 'fetch one record:' result = cursor.fetchone() print result print 'id: %s,name: %s' %(result[0],result[1]) # 查詢多條記錄 print 'fetch five record:' results = cursor.fetchmany(5) for r in results: print r # 查詢所有記錄 # 重置游標位置，偏移量:大於0向后移動;小於0向前移動，mode默認是relative # relative:表示從當前所在的行開始移動; absolute:表示從第一行開始移動 cursor.scroll(0,mode='absolute') results = cursor.fetchall() for r in results: print r cursor.scroll(-2) results = cursor.fetchall() for r in results: print r # 更新記錄 cursor.execute('UPDATE %s SET name = "%s" WHERE id = %s' %(TABLE_NAME,'Jack',1)) # 刪除記錄 cursor.execute('DELETE FROM %s WHERE id = %s' %(TABLE_NAME,2)) # 如果沒有設置自動提交事務，則這里需要手動提交一次 conn.commit() except: import traceback traceback.print_exc() # 發生錯誤時會滾 conn.rollback() finally: # 關閉游標連接 cursor.close() # 關閉數據庫連接 conn.close()

查詢時返回字典結構

MySQLdb默認查詢結果都是返回tuple，通過使用不同的游標可以改變輸出格式，這里傳遞一個cursors.DictCursor參數。

import MySQLdb.cursors conn = MySQLdb.connect(host='localhost', user='root', passwd='root', db='test', cursorclass=MySQLdb.cursors.DictCursor) cursor = conn.cursor() cursor.execute('select * from user') r = cursor.fetchall() print r # 當使用位置參數或字典管理參數時，必須導入MySQLdb.cursors模塊 # 也可以用下面的寫法 import MySQLdb as mdb conn = mdb.connect('localhost', 'root', 'root', 'test') cursor = conn.cursor(cursorclass=mdb.cursors.DictCursor) cursor.execute('select * from user') r = cursor.fetchall() print r

MySQLdb取回大結果集的技巧

普通的操作無論是fetchall()還是fetchone()都是先將數據載入到本地再進行計算，大量的數據會導致內存資源消耗光。解決辦法是使用SSCurosr光標來處理。

• 解決mysqldb查詢大量數據導致內存使用過高的問題
• MySQLdb取回大結果集的技巧

然而，在python3下，MySQLdb模塊不再提供支持，此時可以使用另一個模塊PyMySQL，它支持python2和python3。

2. PyMySQL

PyMySQL是一個純Python寫的MySQL客戶端，它的目標是替代MySQLdb，可以在CPython、PyPy、IronPython和Jython環境下運行。PyMySQL在MIT許可下發布。

PyMySQL的性能和MySQLdb幾乎相當，如果對性能要求
不是特別的強，使用PyMySQL將更加方便。

PyMySQL的使用方法和MySQLdb幾乎一樣。

安裝

pip install pymysql

實例

#!/usr/bin/env python # -*- coding: utf-8 -*- import pymysql config = { 'host': '127.0.0.1', 'port': 3306, 'user': 'root', 'passwd': 'root', 'charset':'utf8mb4', 'cursorclass':pymysql.cursors.DictCursor } conn = pymysql.connect(**config) conn.autocommit(1) cursor = conn.cursor() try: # 創建數據庫 DB_NAME = 'test' cursor.execute('DROP DATABASE IF EXISTS %s' %DB_NAME) cursor.execute('CREATE DATABASE IF NOT EXISTS %s' %DB_NAME) conn.select_db(DB_NAME) #創建表 TABLE_NAME = 'user' cursor.execute('CREATE TABLE %s(id int primary key,name varchar(30))' %TABLE_NAME) # 批量插入紀錄 values = [] for i in range(20): values.append((i,'kk'+str(i))) cursor.executemany('INSERT INTO user values(%s,%s)',values) # 查詢數據條目 count = cursor.execute('SELECT * FROM %s' %TABLE_NAME) print 'total records:', cursor.rowcount # 獲取表名信息 desc = cursor.description print "%s %3s" % (desc[0][0], desc[1][0]) cursor.scroll(10,mode='absolute') results = cursor.fetchall() for result in results: print result except: import traceback traceback.print_exc() # 發生錯誤時會滾 conn.rollback() finally: # 關閉游標連接 cursor.close() # 關閉數據庫連接 conn.close()

輸出結果：

total records: 20 id name {u'id': 10, u'name': u'kk10'} {u'id': 11, u'name': u'kk11'} {u'id': 12, u'name': u'kk12'} {u'id': 13, u'name': u'kk13'} {u'id': 14, u'name': u'kk14'} {u'id': 15, u'name': u'kk15'} {u'id': 16, u'name': u'kk16'} {u'id': 17, u'name': u'kk17'} {u'id': 18, u'name': u'kk18'} {u'id': 19, u'name': u'kk19'}

參考：

• python操作mysql數據庫
• MySQL Python tutorial
• python操作MySQL數據庫
• python操作MySQL

文／無限小BUG（簡書作者）
原文鏈接：http://www.jianshu.com/p/76fab6cb06f9
著作權歸作者所有，轉載請聯系作者獲得授權，並標注“簡書作者”。