jyoung/tools/lib/cache.py

import os
import urllib.parse

DEFAULT_CACHE_DIR = os.getenv("CACHE_ROOT", os.path.expanduser("~/.commacache"))

def cache_path_for_file_path(fn, cache_dir=DEFAULT_CACHE_DIR):
  dir_ = os.path.join(cache_dir, "local")
  os.makedirs(dir_, exist_ok=True)
  fn_parsed = urllib.parse.urlparse(fn)
  if fn_parsed.scheme == '':
    cache_fn = os.path.abspath(fn).replace("/", "_")
  else:
    cache_fn = f'{fn_parsed.hostname}_{fn_parsed.path.replace("/", "_")}'
  return os.path.join(dir_, cache_fn)