Fallback to previous directory if user already has plugins rc file

schrodinger · Mar 27, 2024 · e6c6a70 · e6c6a70
1 parent 3dc6c22
commit e6c6a70
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/modules/pymol/plugins/__init__.py b/modules/pymol/plugins/__init__.py
@@ -20,10 +20,16 @@ def user_data_dir(appname, *args, **kwargs):
         return f"~/.local/share/{appname}"
 
 # variables
-data_dir = os.path.expanduser(user_data_dir("pymol"))
-if not os.path.exists(data_dir):
+def _get_pymolplugins_rc_path():
+    f1 = os.path.expanduser("~/.pymolpluginsrc.py")
+    if os.path.exists(f1):
+        return f1
+
+    data_dir = os.path.expanduser(user_data_dir("pymol"))
     os.makedirs(data_dir, exist_ok=True)
-PYMOLPLUGINSRC = os.path.join(data_dir, 'pymolpluginsrc.py')
+    return os.path.join(data_dir, 'pymolpluginsrc.py')
+
+PYMOLPLUGINSRC = _get_pymolplugins_rc_path()
 
 preferences = {
     'verbose': False,