1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import os
from pathlib import Path
from typing import TYPE_CHECKING
import click
if TYPE_CHECKING:
from jupyter_cache.base import JupyterCacheAbstract
class CacheContext:
"""Context for retrieving the cache."""
def __init__(self, cache_path=None) -> None:
if cache_path is None:
self._cache_path = os.environ.get(
"JUPYTERCACHE", os.path.join(os.getcwd(), ".jupyter_cache")
)
else:
self._cache_path = cache_path
@property
def cache_path(self) -> Path:
return Path(self._cache_path)
def get_cache(self, ask_on_missing=True) -> "JupyterCacheAbstract":
"""Get the cache."""
from jupyter_cache import get_cache
if (not self.cache_path.exists()) and ask_on_missing:
click.secho("Cache path: ", fg="green", nl=False)
click.echo(str(self.cache_path))
if not click.confirm(
"The cache does not yet exist, do you want to create it?"
):
raise click.Abort()
# gets created lazily
return get_cache(self.cache_path)
def set_cache_path(self, cache_path: str) -> None:
self._cache_path = cache_path
pass_cache = click.make_pass_decorator(CacheContext, ensure=True)
|
