【Python】基於pymysql的資料庫操作類

一簡介
Python和MySQL互動的模組有 MySQLdb 和 PyMySQL(pymysql),MySQLdb是基於C 語言編寫的，而且Python3 不在支援MySQLdb 。PyMySQL是一個純Python寫的MySQL客戶端，它的目標是替代MySQLdb，可以在CPython、PyPy、IronPython和Jython環境下執行,PyMySQL在MIT許可下發布。
在開發基於Python語言的專案中，為了以後系統能相容Python3,我們使用了PyMySQL替換了MySQLdb。下面我們來熟悉一下pymysql的使用。

二安裝方式
pymsql的原始碼 https://github.com/PyMySQL/PyMySQL ，目前還在持續更新。

安裝要求:
Python -- one of the following:
CPython >= 2.6 or >= 3.3
PyPy >= 4.0
IronPython 2.7
MySQL Server -- one of the following:
MySQL >= 4.1 (tested with only 5.5~)
MariaDB >= 5.1
安裝
pip install PyMySQL

相關文件 , 強烈建議大家仔細閱讀一遍。其用法和MySQLdb相差無幾，核心用法一致。這樣使用pymysql替換mysqldb的成本極小。

三基於pymysql的資料庫互動

#!/usr/bin/env python
# encoding: utf-8
"""
author: yangyi@youzan
time: 2015/6/8 上午11:34
func: 基於pymysql的資料庫互動類，支援事務提交和回滾，返回結果記錄行數，和insert的最新id
"""
import pymysql
from warnings import filterwarnings
filterwarnings('ignore', category=pymysql.Warning)
CONNECT_TIMEOUT = 100
IP = 'localhost'
PORT = 3306
USER = 'root'
PASSSWORD = ''
class QueryException(Exception):
"""
"""
class ConnectionException(Exception):
"""
"""
class MySQL_Utils():
def __init__(
self, ip=IP, port=PORT, user=USER, password=PASSSWORD,
connect_timeout=CONNECT_TIMEOUT, remote=False, socket='', dbname='test'):
self.__conn = None
self.__cursor = None
self.lastrowid = None
self.connect_timeout = connect_timeout
self.ip = ip
self.port = port
self.user = user
self.password = password
self.mysocket = socket
self.remote = remote
self.db = dbname
self.rows_affected = 0
def __init_conn(self):
try:
conn = pymysql.connect(
host=self.ip,
port=int(self.port),
user=self.user,
db=self.db,
connect_timeout=self.connect_timeout,
charset='utf8', unix_socket=self.mysocket)
except pymysql.Error as e:
raise ConnectionException(e)
self.__conn = conn
def __init_cursor(self):
if self.__conn:
self.__cursor = self.__conn.cursor(pymysql.cursors.DictCursor)
def close(self):
if self.__conn:
self.__conn.close()
self.__conn = None
#專門處理select 語句
def exec_sql(self, sql, args=None):
try:
if self.__conn is None:
self.__init_conn()
self.__init_cursor()
self.__conn.autocommit = True
self.__cursor.execute(sql, args)
self.rows_affected = self.__cursor.rowcount
results = self.__cursor.fetchall()
return results
except pymysql.Error as e:
raise pymysql.Error(e)
finally:
if self.__conn:
self.close()
# 專門處理dml語句 delete，updete，insert
def exec_txsql(self, sql, args=None):
try:
if self.__conn is None:
self.__init_conn()
self.__init_cursor()
if self.__cursor is None:
self.__init_cursor()
self.rows_affected=self.__cursor.execute(sql, args)
self.lastrowid = self.__cursor.lastrowid
return self.rows_affected
except pymysql.Error as e:
raise pymysql.Error(e)
finally:
if self.__cursor:
self.__cursor.close()
self.__cursor = None
# 提交
def commit(self):
try:
if self.__conn:
self.__conn.commit()
except pymysql.Error as e:
raise pymysql.Error(e)
finally:
if self.__conn:
self.close()
#回滾操作
def rollback(self):
try:
if self.__conn:
self.__conn.rollback()
except pymysql.Error as e:
raise pymysql.Error(e)
finally:
if self.__conn:
self.close()
# 適用於需要獲取插入記錄的主鍵自增id
def get_lastrowid(self):
return self.lastrowid

#獲取dml操作影響的行數
def get_affectrows(self):
return self.rows_affected
#MySQL_Utils初始化的例項銷燬之後，自動提交
def __del__(self):
self.commit()

四小結
前幾天剛剛將我們的系統中的MySQLdb 替換為PyMySQL, 還未遇到問題。歡迎大家使用測試上述指令碼，有問題歡迎和我討論。如果本文對您有幫助，可以贊助一瓶飲料。

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/22664653/viewspace-2140569/，如需轉載，請註明出處，否則將追究法律責任。

【Python】基於pymysql的資料庫操作類

相關文章