add asset files to install

pjamesjoyce · Mar 28, 2022 · aed0dc4 · aed0dc4
1 parent 9527457
commit aed0dc4
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/futura_image/Image variable names.xlsx b/futura_image/Image variable names.xlsx
diff --git a/futura_image/mendoza_beltran_utilities.py b/futura_image/mendoza_beltran_utilities.py
@@ -19,7 +19,7 @@
 import os
 
 dir_path = os.path.dirname(os.path.realpath(__file__))
-image_filename = os.path.join(dir_path, 'Image variable names.xlsx')
+image_filename = os.path.join(dir_path, 'assets/Image variable names.xlsx')
 
 image_variable_names = pd.read_excel(image_filename)
 REGIONS = image_variable_names['Regions'].dropna().values

diff --git a/setup.py b/setup.py
@@ -21,14 +21,25 @@
             pkg = pkg.replace(os.path.altsep, '.')
         packages.append(pkg)
 
+def package_files(directory):
+    paths = []
+    for (path, directories, filenames) in os.walk(directory):
+        for filename in filenames:
+            paths.append(os.path.join('..', path, filename))
+    return paths
+
+
+my_package_files = []
+my_package_files.extend(package_files(os.path.join('futura_image', 'assets')))
+
 setup(
     name=PACKAGE_NAME,
     version=VERSION,
     packages=packages,
     author="P. James Joyce",
     author_email="pjamesjoyce@gmail.com",
     license="BSD 3-Clause License",
-    #package_data={'futura': my_package_files},
+    package_data={'futura_image': my_package_files},
     entry_points={
         'futura_plugins': [
             'image = futura_image'