-

Ram Rachum · Ram Rachum · commit 5a66817c8113 · 2010-05-01T01:50:09.000+02:00
diff --git a/garlicsim_wx/setup.py b/garlicsim_wx/setup.py
@@ -7,6 +7,7 @@
 
 import os.path
 import sys
+import glob
 import setuptools
 import distutils # Just for deleting the "build" directory.
 try:
@@ -23,13 +24,28 @@
 except Exception:
     pass
 
+def package_to_path(package):
+    return package.replace('.', '/')
+
 def get_packages():
     return ['garlicsim_wx.' + p for p
             in setuptools.find_packages('./garlicsim_wx')] + \
            ['garlicsim_wx']
 
 packages = get_packages()
 
+def get_all_data_files():
+    total_data_files = []
+    for package in packages:
+        path = package_to_path(package)
+        all_files_and_folders = glob.glob(path + '/*')
+        data_files = [f for f in all_files_and_folders if
+                      (not '.py' in f[4:]) and (not os.path.isdir(f))]
+        total_data_files += data_files
+    return total_data_files
+
+g = get_all_data_files()
+
 my_long_description = \
 '''\
 garlicsim_wx, a wxPython GUI for garlicsim.
@@ -49,24 +65,8 @@ def get_packages():
 ]
 
 
-
-setuptools.setup(
-    name='garlicsim_wx',
-    version='0.4',
-    requires=['garlicsim (== 0.4)'],
-    install_requires=['garlicsim == 0.4'],
-    description='Gui for garlicsim, a Pythonic framework for working with simulations',
-    author='Ram Rachum',
-    author_email='cool-rr@cool-rr.com',
-    url='http://garlicsim.org',
-    packages=packages,
-    license='Proprietary',
-    long_description = my_long_description,
-    classifiers = my_classifiers,
-    include_package_data = True,
-    
-    # For py2exe:
-    windows=[
+py2exe_kwargs = {
+    'windows': [
         {
             'script': 'py2exe_cruft/GarlicSim.py',
             'icon_resources': [
@@ -77,10 +77,31 @@ def get_packages():
             ]
         }
         ],
-    options={
+    'zipfile': 'lib/library.zip',
+    'data_files': g,
+    'options': {
         'py2exe': {
             'dist_dir': 'py2exe_dist',
             'packages': packages
         }
     }
+}
+
+more_kwargs = py2exe_kwargs if `py2exe` in sys.argv else {}
+
+setuptools.setup(
+    name='garlicsim_wx',
+    version='0.4',
+    requires=['garlicsim (== 0.4)'],
+    install_requires=['garlicsim == 0.4'],
+    description='Gui for garlicsim, a Pythonic framework for working with simulations',
+    author='Ram Rachum',
+    author_email='cool-rr@cool-rr.com',
+    url='http://garlicsim.org',
+    packages=packages,
+    license='Proprietary',
+    long_description = my_long_description,
+    classifiers = my_classifiers,
+    include_package_data = True,
+    **more_kwargs
 )
