iskm/wordfreq
1
0
Fork 0
mirror of https://github.com/rspeer/wordfreq.git synced 2024-12-23 17:31:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow multithreaded SQLite on Python 3

Browse Source
This commit is contained in:
Rob Speer 2014-10-02 18:10:09 -04:00
parent 6d90cef415
commit bf0071fd8b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
wordfreq/query.py
View File

@ -5,8 +5,10 @@ import sys
if sys.version_info.major == 2:
from functools32 import lru_cache
PY2 = True
else:
from functools import lru_cache
PY2 = False
SQLITE_ERROR_TEXT = """
Couldn't open the wordlist database.
@ -20,7 +22,10 @@ This can be configured by setting the WORDFREQ_DATA environment variable.
""" % {'path': DB_FILENAME}
try:
CONN = sqlite3.connect(DB_FILENAME)
if PY2:
CONN = sqlite3.connect(DB_FILENAME)
else:
CONN = sqlite3.connect(DB_FILENAME, check_same_thread=False)
except sqlite3.OperationalError:
raise IOError(SQLITE_ERROR_TEXT)