Python中Mysqldb的使用以及事务的支持-pybatis首次露面-pybatis的准备阶段（十）

http://zetcode.com/db/mysqlpython/

这个地址中有一些关于mysqldb模块的使用介绍。

其中关键一点的是事务支持。

“

For databases that support transactions, the Python interface silently starts a transaction when the cursor is created. The commit() method commits the updates made using that cursor, and the rollback() method discards them.

”
上面这句话说的比较好。在py的mysqldb中，一旦创建了一个cursor，那么就默认创建了一个事务。直到提交事务为止，数据库的数据才会进行更改。但是对于不支持事务的数据库来说，事务不起作用。

------------------------------------------------------------

上面是mysqldb的事务支持。那么，我自己的pybatis要怎么实现呢？

目前是这样的：

对于单一的查询，开启一个数据库连接，执行sql，然后提交事务，关闭cursor和conn即可。

但是对于用户需要全局事务来说，是这样的：

把所有的查询都放在同一个connection中。在最后的时候再使用conn.commit()进行提交。

贴一下临时的transactionmanager的代码：

# -*- coding:utf-8 -*-
'''
Created on 2013-3-10

@author: naughty
'''
from _mysql_exceptions import IntegrityError
from com.util.Data2Object import Data2Object
from com.util.SqlPreprocessing import processSql
import MySQLdb as mdb

class TransactionManager(object):
    '''
    handles all transaction and db query
    '''
    def __init__(self):
        '''
        初始化的时候传入一个连接
        '''
        self.conn = mdb.connect(host='localhost', user='', passwd='', db='test')
        #=======================================================================
        # 初始化一个cursor
        #=======================================================================
        self.cursor = self.conn.cursor()
        
    def startTransaction(self):
        '''
        For databases that support transactions, 
        the Python interface silently starts a transaction when the cursor is created.
        so we do nothing here.
        '''
        pass
    def commitTransaction(self):
        '''
        提交事务
        '''
        self.cursor.close()
        self.conn.commit()
    def endTransaction(self):
        '''
        结束事务
        '''
        pass
    def rollbackTransaction(self):
        '''
        回滚事务
        '''
        self.cursor.close()
        self.conn.rollback()
        
    def queryInsert(self, sqlid, inputObject):
        '''
        查询插入
        '''
        #=======================================================================
        # resultclasstype参数在没有返回值的时候用不到
        #=======================================================================
        sql , resultclasstype = processSql(sqlid, inputObject)
        try:
            var=self.cursor.execute(sql)
            return var
        except IntegrityError:
            return -1
        
    def queryUpdate(self, sqlid, inputObject):
        '''
        查询更新
        '''
        self.queryInsert(sqlid, inputObject)
    def queryDelete(self, sqlid, inputObject):
        '''
        查询删除
        '''
        self.queryInsert(sqlid, inputObject)

    def queryForObject(self, sqlid, inputObject):
        '''
        查询并返回一个对象
        '''
        sql, resultclasstype = processSql(sqlid, inputObject)
        self.cursor.execute(sql)
        objList = Data2Object.data2object(self.cursor, resultclasstype)
        if len(objList) == 0:
            return None
        elif len(objList) == 1:
            return objList[0]
        else:
            raise Exception('query for one object, but get many.');
        
    def queryForList(self, sqlid, inputObject):
        '''
        查询并返回一个列表
        '''
        sql , resultclasstype = processSql(sqlid, inputObject)
        self.cursor.execute(sql)
        objList = Data2Object.data2object(self.cursor, resultclasstype)
        return objList
    

上面的代码只有在commitTransaction的时候才会提交事务。【rollbackTransaction的时候，最后还需要关闭连接。上面的代码忘记关闭了】

看下面的测试用例：

# -*- coding:utf-8 -*-
'''
Created on 2013-3-5

@author: naughty
'''
from com.domain import beans
from com.domain.TransactionManager import TransactionManager
from com.util.DBExecuteCenter import queryForObject, queryInsert, queryUpdate, \
    queryDelete

if __name__ == '__main__':
    s = beans.Student()
    s.age = '1'
    s.name = 'zoer'
    s.score = 11
    t = TransactionManager()
    try:
        t.startTransaction()
        num = t.queryDelete('del', s)
        print num
        s.score = 22
        num = t.queryDelete('del', s)
        print num
        t.commitTransaction()
    except:
        t.rollbackTransaction()

    print 'end'

用例中，尝试删除score是11和22的学生。但是，由于22是有外键约束的【我人为添加的一个外键约束】，所以删除22 的学生的时候会失败。这样，上面的代码在执行到删除22的时候由于抛出异常，就会去回滚事务。

--------------------------------

上面展示的是“大事务”控制。使用TransactionManager来操作。

下面展示一下单个操作时候的事务控制。

单个事务控制是放在 中进行解决的。

# -*- coding:utf-8 -*-
'''
Created on 2013-3-6

@author: naughty
'''

from Data2Object import Data2Object
from com.util.SqlPreprocessing import processSql
import MySQLdb as mdb
import com.pybatis.Global

def queryForObject(sqlid, inputObject=None):
    '''
    查询并返回对象
    '''
    #===========================================================================
    # 每次都新建连接
    #===========================================================================
    conn = mdb.connect(host='localhost', user='', passwd='', db='test')
    cursor = conn.cursor()
    sql, resultclasstype = processSql(sqlid, inputObject)
    try:
        cursor.execute(sql)
    except:
        cursor.close()
        conn.rollback()
        conn.close()
    else:
        objList = Data2Object.data2object(cursor, resultclasstype)
        cursor.close()
        if len(objList) == 0:
            conn.close()
            return None
        elif len(objList) == 1:
            conn.close()
            return objList[0]
        else:
            conn.close()
            raise Exception('query for one object, but get many.');

def queryForList(sqlid, inputObject=None):
    '''
    查询并返回列表
    '''
    #===========================================================================
    # 每次都新建连接
    #===========================================================================
    conn = mdb.connect(host='localhost', user='', passwd='', db='test')
    cursor = conn.cursor()
    sql , resultclasstype = processSql(sqlid, inputObject)
    try:
        cursor.execute(sql)
    except:
        cursor.close()
        conn.rollback()
        conn.close()
    else:
        objList = Data2Object.data2object(cursor, resultclasstype)
        cursor.close()
        conn.close()
        return objList

def queryInsert(sqlid, inputObject=None):
    '''
    执行插入操作
    '''
    conn = mdb.connect(host='localhost', user='', passwd='', db='test')
    cursor = conn.cursor()
    sql , resultclasstype = processSql(sqlid, inputObject)
    try:
        count = cursor.execute(sql)
    except:
        #===========================================================================
        # 关闭游标
        #===========================================================================
        cursor.close()
        conn.rollback()
        conn.close()
        count = -1
    else:
        cursor.close()
        conn.commit()
        conn.close()
    return count

def queryUpdate(sqlid, inputObject=None):
    '''
    执行更新操作
    '''
    queryInsert(sqlid, inputObject)

def queryDelete(sqlid, inputObject=None):
    '''
    执行删除操作
    '''
    queryInsert(sqlid, inputObject)

单个事务控制的时候，把对数据库的操作放在了上面的这些函数中进行。具体看看queryInsert函数。操作是否成功需要根据返回值进行判断。如果返回值是-1，那么说明操作没有成功。相反则成功。

下面是一个使用的例子：

# -*- coding:utf-8 -*-
'''
Created on 2013-3-5

@author: naughty
'''
from com.domain import beans
from com.domain.TransactionManager import TransactionManager
from com.util.DBExecuteCenter import queryForObject, queryInsert, queryUpdate, \
    queryDelete

if __name__ == '__main__':
    s = beans.Student()
    s.age = '1'
    s.name = 'naughty'
    s.score = 11
    count=queryInsert('insert',s)
    print count
    print 'end'

例子中插入了一个学生信息。由于数据库中已经存在了name是naughty的学生【name是主键】，所以插入失败。print出来的值是-1。

--------------------------------------

上面基本展示了pybatis处理全局事务和单个事务的方法。其本质就是将所有的事务处理交给基本的connection去处理，pybatis只是使用了connection。

------------------------------------------

至此，pybatis的事务管理就告一个段落了【不是最终版本】。

接下来还需要一个数据库连接池。