Changeset 8ab3302 in sasview for fittingview/src/sans/perspectives/fitting/models.py

Timestamp:

Dec 22, 2011 9:55:32 AM (12 years ago)

Author:

Jae Cho <jhjcho@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

273801b

Parents:

7983000f

Message:

fixed a little bit the setting of .sansview folder

File:

• fittingview/src/sans/perspectives/fitting/models.py (modified) (3 diffs)

fittingview/src/sans/perspectives/fitting/models.py

@@ -18 +18 @@
 from sans.models.pluginmodel import Model1DPlugin
    
-PLUGIN_DIR = 'plugins' 
+PLUGIN_DIR = 'plugin_models' 
 
 def log(message):
@@ -75 +75 @@
         The plugin directory is located in the user's home directory.
     """
-    dir = os.path.join(os.path.expanduser("~"),'.sansview',PLUGIN_DIR)
+    dir = os.path.join(os.path.expanduser("~"),'.sansview', PLUGIN_DIR)
     
     # If the plugin directory doesn't exist, create it
@@ -81 +81 @@
         os.makedirs(dir)
         
+    # Find paths needed
+    try:
+        # For source
+        if os.path.isdir(os.path.dirname(__file__)):
+            p_dir =  os.path.join(os.path.dirname(__file__), PLUGIN_DIR)
+        else:
+            raise
+    except:
+        # Check for data path next to exe/zip file.
+        #Look for maximum n_dir up of the current dir to find plugins dir
+        n_dir = 12
+        p_dir = None
+        f_dir = os.path.join(os.path.dirname(__file__))
+        for i in range(n_dir):
+            if i > 1:
+                f_dir, _ = os.path.split(f_dir)
+            plugin_path = os.path.join(f_dir, PLUGIN_DIR)
+            if os.path.isdir(plugin_path):
+                p_dir = plugin_path
+                break
+        if not p_dir:
+            raise
+
     # Place example user models as needed
-    if not os.path.isfile(os.path.join(dir,"testmodel.py")):
-        shutil.copy(os.path.join(os.path.dirname(__file__),"plugin_models","testmodel.py"), dir)
-    if not os.path.isfile(os.path.join(dir,"testmodel_2.py")):
-        shutil.copy(os.path.join(os.path.dirname(__file__),"plugin_models","testmodel_2.py"), dir)
-    if not os.path.isfile(os.path.join(dir,"sum_p1_p2.py")):
-        shutil.copy(os.path.join(os.path.dirname(__file__),"plugin_models","sum_p1_p2.py"), dir)
-        
+    for file in os.listdir(p_dir): 
+        file_path = os.path.join(p_dir, file)
+        if os.path.isfile(file_path):
+            if file.split(".")[-1] == 'py' and\
+                file.split(".")[0] != '__init__':
+                if not os.path.isfile(os.path.join(dir, file)):
+                    shutil.copy(file_path, dir)
     return dir