p3lib 1.1.124__tar.gz → 1.1.125__tar.gz

{p3lib-1.1.124 → p3lib-1.1.125}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: p3lib
-Version: 1.1.124
+Version: 1.1.125
 Summary: A group of python modules for networking, plotting data, config storage, automating boot scripts, ssh access and user input output.
 License: MIT
 Author: Paul Austen

{p3lib-1.1.124 → p3lib-1.1.125}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "p3lib"
-version = "1.1.124"
+version = "1.1.125"
 description = "A group of python modules for networking, plotting data, config storage, automating boot scripts, ssh access and user input output."
 authors = ["Paul Austen <pjaos@gmail.com>"]
 license = "MIT License"

{p3lib-1.1.124 → p3lib-1.1.125}/src/p3lib/helper.py

@@ -9,6 +9,7 @@ import platform
 import json
 import traceback
 import socket
+import inspect
 
 def initArgs(parser, lastCmdLineArg=None, checkHostArg=True):
     """This method is responsible for
@@ -451,3 +452,69 @@ def getProgramVersion():
         return -999.99
     return programVersion
 
+def get_assets_folders():
+    """@return A list of all the assets folders found."""
+    searchFolders = []
+    assetsFolders = []
+    calling_file = None
+    # Get the full path to the python file that called this get_assets_folders() function.
+    frame = inspect.stack()[1]
+    module = inspect.getmodule(frame[0])
+    if module and hasattr(module, '__file__'):
+        calling_file = os.path.abspath(module.__file__)
+
+    if calling_file:
+        startup_path = os.path.dirname(calling_file)
+        searchFolders.append( os.path.join(startup_path, 'assets') )
+        pp1 = os.path.join(startup_path, '..')
+        searchFolders.append( os.path.join(pp1, 'assets') )
+        pp2 = os.path.join(pp1, '..')
+        searchFolders.append( os.path.join(pp2, 'assets') )
+        # Try all the site packages folders we know about.
+        for path in sys.path:
+            if 'site-packages' in path:
+                site_packages_path = path
+                searchFolders.append( os.path.join(site_packages_path, 'assets') )
+
+        for folder in searchFolders:
+            absPath = os.path.abspath(folder)
+            if os.path.isdir(absPath):
+                assetsFolders.append(absPath)
+
+    return assetsFolders
+
+def get_assets_folder(raise_error=True):
+    """@brief Get the assets folder.
+       @param raise_error If True then raise an error if the assets folder is not found.
+       @return The abs assets folder path string."""
+    searchFolders = []
+    assetsFolder = None
+    calling_file = None
+    # Get the full path to the python file that called this get_assets_folder() function.
+    frame = inspect.stack()[1]
+    module = inspect.getmodule(frame[0])
+    if module and hasattr(module, '__file__'):
+        calling_file = os.path.abspath(module.__file__)
+
+    if calling_file:
+        startup_path = os.path.dirname(calling_file)
+        searchFolders.append( os.path.join(startup_path, 'assets') )
+        pp1 = os.path.join(startup_path, '..')
+        searchFolders.append( os.path.join(pp1, 'assets') )
+        pp2 = os.path.join(pp1, '..')
+        searchFolders.append( os.path.join(pp2, 'assets') )
+        # Try all the site packages folders we know about.
+        for path in sys.path:
+            if 'site-packages' in path:
+                site_packages_path = path
+                searchFolders.append( os.path.join(site_packages_path, 'assets') )
+
+        for folder in searchFolders:
+            absPath = os.path.abspath(folder)
+            if os.path.isdir(absPath):
+                assetsFolder = absPath
+
+    if raise_error and assetsFolder is None:
+        raise Exception('Failed to find assets folder.')
+
+    return assetsFolder