python torndb模块(转)
一、torndb概述
torndb是一个轻量级的基于MySQLdb封装的一个模块,其是tornado框架的一部分。其项目主页为:https://github.com/bdarnell/torndb 。从tornado3.0版本以后,其已经作为一个独立模块发行了。可以通过easy_install 或pip的方式直接安装。
- 修改MySQLdb,torndb是依赖于MySQLdb实现的对MySQL数据库操作,但是python3中不支持MySQLdb,而是使用pymysql,所以需要将源码中使用MySQLdb的地方修改为pymysql。
1)修改导入模块
1 import pymysql.connections 2 import pymysql.converters 3 import pymysql.cursors 4 # import MySQLdb.constants 5 # import MySQLdb.converters 6 # import MySQLdb.cursors
1 def reconnect(self): 2 """Closes the existing database connection and re-opens it.""" 3 self.close() 4 self._db = pymysql.connect(**self._db_args)# MySQLdb.connect(**self._db_args) 5 self._db.autocommit(True) 6 2)修改连接mysql的方式
3)修改连接参数,以及遍历字段类型时所使用的列表增加元素(python3使用append进行元素的添加,而不是使用加号)
1 # if MySQLdb is not None: 2 if pymysql is not None: 3 # Fix the access conversions to properly recognize unicode/binary 4 FIELD_TYPE = pymysql.connections.FIELD_TYPE # MySQLdb.constants.FIELD_TYPE 5 FLAG = pymysql.constants.FLAG# MySQLdb.constants.FLAG 6 CONVERSIONS = copy.copy (pymysql.converters.conversions)# (MySQLdb.converters.conversions) 7 8 field_types = [FIELD_TYPE.BLOB, FIELD_TYPE.STRING, FIELD_TYPE.VAR_STRING] 9 if 'VARCHAR' in vars(FIELD_TYPE): 10 field_types.append(FIELD_TYPE.VARCHAR) 11 12 for field_type in field_types: 13 # CONVERSIONS[field_type] = [(FLAG.BINARY, str)] + CONVERSIONS[field_type] 14 CONVERSIONS[field_type] = [(FLAG.BINARY, str)].append(CONVERSIONS[field_type]) 15 16 # Alias some common MySQL exceptions 17 IntegrityError = pymysql.IntegrityError# MySQLdb.IntegrityError 18 OperationalError = pymysql.OperationalError# MySQLdb.OperationalError 19
修改连接超时时间,在torndb初始化方法中设置,需要传递给pymysql
1 def __init__(self, host, database, user=None, password=None, 2 max_idle_time=7 * 3600, connect_timeout=10,# 设置连接超时时间,时间是秒 3 time_zone="+0:00", charset = "utf8", sql_mode="TRADITIONAL"):
- 修改查询方法中的迭代方法(将izip改为zip_longest)
1 def query(self, query, *parameters, **kwparameters): 2 """Returns a row list for the given query and parameters.""" 3 cursor = self._cursor() 4 try: 5 self._execute(cursor, query, parameters, kwparameters) 6 column_names = [d[0] for d in cursor.description] 7 return [Row(itertools.zip_longest(column_names, row)) for row in cursor] 8 finally: 9 cursor.close()
二、连接与查询
1、创建连接
1 import torndb 2 db = torndb.Connection("127.0.0.1:3306", "test", user="root", password="admin")
默认字符集UTF8,没必要在加上 charset = "utf8" 。另外需要注意的是其默认时区为time_zone='+0:00',这个可以手动在连接的时候指定为东8区。
2、查询
在MySQLdb中所有的操作都是通过execute执行的,而把TA封装之后的torndb,提供了3种,execute,get,query。execute的示例:
1 cre='create table blog(id int,content text)' 2 db.execute(cre) 3 string='wawuee' 4 exe='insert into blog(id,content)values(%d,"%s")'%(1,string) 5 db.execute(exe)
execute包括创建表,插入表,删除表等等,另外其也单独封装了insert、insertmany、update、updatemany函数,同时除了一般的execute函数,其还有execute_lastrowid、execute_rowcount、executemany、executemany_lastrowid、executemany_rowcount函数。
query与get
两个都是用于返回数据结果,不同的是query可以返回多条结果;get查询的结果为空时,返回None,返回多于一条结果时会报出一个异常,一般只用于只有一条结果返回的情况。
1 //query查询 2 >>>sql = 'SELECT * FROM test WHERE name = %s AND id < %s' 3 >>>db.query(sql, 'bbb', 11) 4 [{'date': None, 'id': 1L, 'name': u'bbb'}, {'date': None, 'id': 2L, 'name': u'bbb'}] 5 //get查询 6 >>>sql = 'SELECT * FROM test WHERE id = %s' 7 >>>db.get(sql, 2) 8 {'date': None, 'id': 2L, 'name': u'bbb'}
query查询时,结果为多行时为list列表,单行只为字典。
insert与insertmany
>>>sql = "INSERT INTO test (id,name,date) VALUES (%s,%s,%s)" >>>db.insert(sql, 100, "aaa", '0000-01-01') 100L
insert的参数不支持列表或元组,如果想插入列表或元组的话可以用insertmany
1 #插入单行记录 2 sql = "INSERT INTO test (id,name,date) VALUES (%s,%s,%s)" 3 db.insertmany(sql,[[200,'bbb',None]]) 4 200L 5 #插入多行记录 6 db.insertmany(sql,[[300,'bbb',None],[400,'bbb',None]]) 7 400L 8 db.insertmany(sql,[(301,'bbb',None),(401,'bbb',None)]) 9 401L