allow loading after global installation

chemprop · Sep 21, 2020 · 6b5cc17 · 6b5cc17
1 parent e9a28de
commit 6b5cc17
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/chemprop/__init__.py b/chemprop/__init__.py
@@ -2,7 +2,7 @@
 import chemprop.features
 import chemprop.models
 import chemprop.train
-import chemprop.web
+# import chemprop.web
 
 import chemprop.args
 import chemprop.constants

diff --git a/chemprop/web/config.py b/chemprop/web/config.py
@@ -11,7 +11,12 @@
 _TEMP_FOLDER_OBJECT = TemporaryDirectory()
 
 DEFAULT_USER_ID = 1
-ROOT_FOLDER = os.path.dirname(os.path.realpath(__file__))
+if os.access(os.path.dirname(os.path.realpath(__file__)), os.W_OK):
+    ROOT_FOLDER = os.path.dirname(os.path.realpath(__file__))
+elif os.access(os.getcwd(), os.W_OK):
+    ROOT_FOLDER = os.path.join(os.getcwd(), "chemprop_web_app")
+else :
+    raise ValueError("Failed to find a writable ROOT_FOLDER for web app data and checkpoints.")
 DATA_FOLDER = os.path.join(ROOT_FOLDER, 'app/web_data')
 CHECKPOINT_FOLDER = os.path.join(ROOT_FOLDER, 'app/web_checkpoints')
 TEMP_FOLDER = os.path.join(ROOT_FOLDER, _TEMP_FOLDER_OBJECT.name)