Shoraky commited on
Commit
43f10b3
·
verified ·
1 Parent(s): 490eb58

Fix writable runtime storage path

Browse files
Files changed (1) hide show
  1. api.py +3 -1
api.py CHANGED
@@ -24,13 +24,15 @@ CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
24
  def default_runtime_root():
25
  if os.path.isdir("/data"):
26
  return os.path.join("/data", "sporalize_runtime")
27
- return os.path.join(CURRENT_DIR, ".sporalize_runtime")
28
 
29
 
30
  RUNTIME_ROOT = os.environ.get("SPORALIZE_RUNTIME_DIR", default_runtime_root())
31
  ASSETS_RUNTIME_ROOT = os.environ.get("SPORALIZE_ASSETS_DIR", os.path.join(RUNTIME_ROOT, "assets"))
32
  WEIGHTS_RUNTIME_ROOT = os.environ.get("SPORALIZE_WEIGHTS_DIR", os.path.join(RUNTIME_ROOT, "weights"))
33
  DEFAULT_LOCAL_STORAGE_ROOT = os.path.join(CURRENT_DIR, "Storage")
 
 
34
  STORAGE_ROOT = os.environ.get(
35
  "SPORALIZE_STORAGE_DIR",
36
  os.path.join("/data", "sporalize_storage") if os.path.isdir("/data") else DEFAULT_LOCAL_STORAGE_ROOT,
 
24
  def default_runtime_root():
25
  if os.path.isdir("/data"):
26
  return os.path.join("/data", "sporalize_runtime")
27
+ return os.path.join(os.path.expanduser("~"), ".sporalize_runtime")
28
 
29
 
30
  RUNTIME_ROOT = os.environ.get("SPORALIZE_RUNTIME_DIR", default_runtime_root())
31
  ASSETS_RUNTIME_ROOT = os.environ.get("SPORALIZE_ASSETS_DIR", os.path.join(RUNTIME_ROOT, "assets"))
32
  WEIGHTS_RUNTIME_ROOT = os.environ.get("SPORALIZE_WEIGHTS_DIR", os.path.join(RUNTIME_ROOT, "weights"))
33
  DEFAULT_LOCAL_STORAGE_ROOT = os.path.join(CURRENT_DIR, "Storage")
34
+ if not os.path.isdir("/data") and not os.access(CURRENT_DIR, os.W_OK):
35
+ DEFAULT_LOCAL_STORAGE_ROOT = os.path.join(RUNTIME_ROOT, "storage")
36
  STORAGE_ROOT = os.environ.get(
37
  "SPORALIZE_STORAGE_DIR",
38
  os.path.join("/data", "sporalize_storage") if os.path.isdir("/data") else DEFAULT_LOCAL_STORAGE_ROOT,