github/dejavu
1
0
Fork 0
mirror of https://github.com/correl/dejavu.git synced 2024-11-23 19:19:53 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge remote-tracking branch 'wessie/master'

Browse source
This commit is contained in:
Vin 2013-12-17 01:14:13 +00:00
commit 776995e454
2 changed files with 11 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
dejavu/cursor.py
View file

@ -2,8 +2,8 @@ from __future__ import unicode_literals
from __future__ import absolute_import
import Queue
import pymysql
import pymysql.cursors
import MySQLdb as mysql
import MySQLdb.cursors
def cursor_factory(**factory_options):
@ -26,18 +26,18 @@ class Cursor(object):
"""
_cache = Queue.Queue(maxsize=5)
def __init__(self, cursor_type=pymysql.cursors.DictCursor, **options):
def __init__(self, cursor_type=mysql.cursors.DictCursor, **options):
super(Cursor, self).__init__()
try:
conn = self._cache.get_nowait()
except Queue.Empty:
conn = pymysql.connect(**options)
conn = mysql.connect(**options)
self.conn = conn
self.cursor_type = cursor_type
def __enter__(self):
def __enter__(slf):
self.cursor = self.conn.cursor(self.cursor_type)
return self.cursor

3
dejavu/database.py
View file

@ -1,6 +1,8 @@
from __future__ import absolute_import
from binascii import unhexlify
from MySQLdb.cursors import Cursor
class Database(object):
def __init__(self):
super(Database, self).__init__()
@ -281,7 +283,6 @@ class SQLDatabase(Database):
values.
"""
from pymysql.cursors import Cursor
# Create a dictionary of hash => offset pairs for later lookups
mapper = {}
for hash, (_, offset) in hashes: