Update ash_animator/basemaps.py

ash_animator/basemaps.py

@@ -1,3 +1,4 @@
+import tempfile
 
 # import contextily as ctx
 # from mpl_toolkits.basemap import Basemap
@@ -42,17 +43,11 @@ import matplotlib.pyplot as plt
 # Optional: Set tile cache directory (must be done before contextily downloads tiles)
 os.environ["XDG_CACHE_HOME"] = os.path.expanduser("~/.contextily_cache")
 
-#CTX_TILE_CACHE_DIR = os.path.expanduser("~/.contextily_cache")
-#BASEMAP_TILE_CACHE_DIR = os.path.expanduser("~/.basemap_cache")
-
-import tempfile
-
-CTX_TILE_CACHE_DIR = os.path.join(tempfile.gettempdir(), "contextily_cache")
-os.makedirs(CTX_TILE_CACHE_DIR, exist_ok=True)
-
+CTX_TILE_CACHE_DIR = os.path.expanduser("~/.contextily_cache")
+os.path.join(tempfile.gettempdir(), 'contextily_cache') = os.path.expanduser("~/.basemap_cache")
 
 os.makedirs(CTX_TILE_CACHE_DIR, exist_ok=True)
-os.makedirs(BASEMAP_TILE_CACHE_DIR, exist_ok=True)
+os.makedirs(os.path.join(tempfile.gettempdir(), 'contextily_cache'), exist_ok=True)
 
 def draw_etopo_basemap(ax, mode="basemap", zoom=11):
     """
@@ -99,7 +94,7 @@ def draw_etopo_basemap(ax, mode="basemap", zoom=11):
             # Create a hash key for this extent
             extent_str = f"{extent[0]:.4f}_{extent[1]:.4f}_{extent[2]:.4f}_{extent[3]:.4f}"
             cache_key = hashlib.md5(extent_str.encode()).hexdigest()
-            cache_file = os.path.join(BASEMAP_TILE_CACHE_DIR, f"{cache_key}_highres.png")
+            cache_file = os.path.join(os.path.join(tempfile.gettempdir(), 'contextily_cache'), f"{cache_key}_highres.png")
 
             if os.path.exists(cache_file):
                 img = Image.open(cache_file)