MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2025-02-20 20:09:07 +00:00
Code Issues Releases Wiki Activity

Fix cache problem when going from Python 3 to 2

Browse Source
This commit is contained in:
Pablo Santiago Blum de Aguiar 2016-01-21 23:54:31 -02:00
parent 9c9a7343de
commit cb6f964a30
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
thefuck/utils.py
View File

@ -4,6 +4,7 @@ import pickle
import pkg_resources
import re
import shelve
import six
from .conf import settings
from contextlib import closing
from decorator import decorator
@ -15,6 +16,11 @@ from warnings import warn
DEVNULL = open(os.devnull, 'w')
shelve_open_errors = (dbm.error, )
if six.PY2:
import gdbm
shelve_open_errors += (gdbm.error, )
def memoize(fn):
"""Caches previous calls to the function."""
@ -217,8 +223,8 @@ def cache(*depends_on):
value = fn(*args, **kwargs)
db[key] = {'etag': etag, 'value': value}
return value
except dbm.error:
# Caused when going from Python 2 to Python 3
except shelve_open_errors:
# Caused when going from Python 2 to Python 3 and vice-versa
warn("Removing possibly out-dated cache")
os.remove(cache_path)