Update ash_animator/basemaps.py

ash_animator/basemaps.py

@@ -1,23 +1,46 @@
 
 import os
 import hashlib
-import contextily as ctx
-from mpl_toolkits.basemap import Basemap
-import cartopy.crs as ccrs
-import cartopy.feature as cfeature
 from PIL import Image
 import matplotlib.pyplot as plt
 
-# Define cache directories
-# Optional: Set tile cache directory (must be done before contextily downloads tiles)
-os.environ["XDG_CACHE_HOME"] = os.path.expanduser("~/.contextily_cache")
+import os
+import sys
+
+def get_cache_dir(app_name):
+    if os.name == 'nt':
+        # Windows: Use LOCALAPPDATA or fallback
+        return os.path.join(os.getenv('LOCALAPPDATA', os.getcwd()), f"{app_name}_cache")
+    else:
+        # Unix: Attempt to use root-level /code/app_name_cache
+        base_dir = "/code"
+        cache_path = os.path.join(base_dir, f"{app_name}_cache")
+        try:
+            os.makedirs(cache_path, exist_ok=True)
+            os.chmod(cache_path, 0o777)  # Make it writable by all (if necessary)
+        except PermissionError:
+            print(f"[PermissionError] Cannot write to {cache_path}. Falling back to /tmp.")
+            cache_path = os.path.join("/tmp", f"{app_name}_cache")
+            os.makedirs(cache_path, exist_ok=True)
+        return cache_path
+
+CTX_TILE_CACHE_DIR = get_cache_dir("contextily")
+BASEMAP_TILE_CACHE_DIR = get_cache_dir("basemap")
+CARTOPY_CACHE_DIR = get_cache_dir("cartopy")
+
+os.environ["CARTOPY_USER_BACKGROUNDS"] = CARTOPY_CACHE_DIR
+os.environ["CARTOPY_CACHE_DIR"] = CARTOPY_CACHE_DIR
 
-CTX_TILE_CACHE_DIR = os.path.expanduser("~/.contextily_cache")
-BASEMAP_TILE_CACHE_DIR = os.path.expanduser("~/.basemap_cache")
 
 os.makedirs(CTX_TILE_CACHE_DIR, exist_ok=True)
 os.makedirs(BASEMAP_TILE_CACHE_DIR, exist_ok=True)
 
+
+import contextily as ctx
+from mpl_toolkits.basemap import Basemap
+import cartopy.crs as ccrs
+import cartopy.feature as cfeature
+
 def draw_etopo_basemap(ax, mode="basemap", zoom=11):
     """
     Draws a high-resolution basemap background on the provided Cartopy GeoAxes.