51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
# TODO this probably belongs to cachew? or cachew.experimental
|
|
from contextlib import contextmanager
|
|
from pathlib import Path
|
|
|
|
|
|
def disable_cachew():
|
|
'''
|
|
NOTE: you need to use it before importing any function using @cachew.cachew
|
|
'''
|
|
# TODO not sure... maybe it should instead use some hook.. it's a ibt ugly do
|
|
import cachew
|
|
|
|
@cachew.doublewrap
|
|
def cachew_off(func=None, *args, **kwargs):
|
|
return func
|
|
old = cachew.cachew
|
|
cachew.cachew = cachew_off
|
|
return old
|
|
|
|
|
|
@contextmanager
|
|
def disabled_cachew():
|
|
try:
|
|
import cachew
|
|
except ModuleNotFoundError:
|
|
# no need to disable anything
|
|
yield
|
|
return
|
|
old = disable_cachew()
|
|
try:
|
|
yield
|
|
finally:
|
|
cachew.cachew = old
|
|
|
|
|
|
def cache_dir() -> Path:
|
|
'''
|
|
Base directory for cachew.
|
|
To override, add to your config file:
|
|
class config:
|
|
cache_dir = '/your/custom/cache/path'
|
|
'''
|
|
import my.config as C
|
|
common_config = getattr(C, 'common', object())
|
|
# TODO if attr is set _and_ it's none, disable cache?
|
|
cdir = getattr(common_config, 'cache_dir', None)
|
|
if cdir is None:
|
|
# TODO fallback to default cachew dir instead?
|
|
return Path('/var/tmp/cachew')
|
|
else:
|
|
return Path(cdir)
|