MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2025-03-13 22:28:33 +00:00
Code Issues Releases Wiki Activity

#439 #447 Except anydbm instead of gdbm for PY2

Browse Source 
Fix #439 and close #447
This commit is contained in:
Pablo Santiago Blum de Aguiar 2016-02-14 00:26:51 -02:00
parent 470c0ef699
commit 3cc4940842
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
thefuck/utils.py
View File

@ -1,4 +1,3 @@
import dbm
import os
import pickle
import pkg_resources
@ -16,10 +15,12 @@ from warnings import warn
DEVNULL = open(os.devnull, 'w')
shelve_open_errors = (dbm.error, )
if six.PY2:
import gdbm
shelve_open_errors += (gdbm.error, )
import anydbm
shelve_open_error = anydbm.error
else:
import dbm
shelve_open_error = dbm.error
def memoize(fn):
@ -227,7 +228,7 @@ def cache(*depends_on):
value = fn(*args, **kwargs)
db[key] = {'etag': etag, 'value': value}
return value
except shelve_open_errors:
except shelve_open_error:
# Caused when going from Python 2 to Python 3 and vice-versa
warn("Removing possibly out-dated cache")
os.remove(cache_path)