Changeset df8a13e in sasview for guiframe/local_perspectives/plotting/__init__.py

Timestamp:

May 12, 2011 3:35:05 PM (13 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:

b2448a5

Parents:

0685f45

Message:

assigned help file dir

File:

• guiframe/local_perspectives/plotting/__init__.py (modified) (1 diff)

guiframe/local_perspectives/plotting/__init__.py

@@ -1 +1 @@
 PLUGIN_ID = "Plotting plug-in 1.0"
 from plotting import *
+
+import os
+from distutils.filelist import findall
+
+N_DIR = 12
+def get_data_path(media):
+    """
+    """
+    # Check for data path in the package
+    path = os.path.join(os.path.dirname(__file__), media)
+    if os.path.isdir(path):
+        return path
+    # Check for data path next to exe/zip file.
+    # If we are inside a py2exe zip file, we need to go up
+    # two levels to get to the directory containing the exe
+    # We will check if the exe and the xsf are in the same
+    # directory.
+    path = os.path.dirname(__file__)
+    #Look for maximum n_dir up of the current dir to find media
+   
+    #for i in range(n_dir):
+    i = 0
+    while(i < N_DIR):
+        path, _ = os.path.split(path)
+        media_path = os.path.join(path, media)
+        if os.path.isdir(media_path):
+             module_media_path = os.path.join(media_path,'plotting_media')
+             if os.path.isdir(module_media_path):
+                 return module_media_path
+             return media_path
+        i += 1
+   
+    raise RuntimeError('Could not find plotting media files')
+
+def data_files():
+    """
+    Return the data files associated with media plotting.
+    
+    The format is a list of (directory, [files...]) pairs which can be
+    used directly in setup(...,data_files=...) for setup.py.
+
+    """
+    data_files = []
+    path = get_data_path(media="media")
+    for f in findall(path):
+        data_files.append(('media/plotting_media', [f]))
+    return data_files