Added function for finding data for setup.py files

1 files changed, 67 insertions, 0 deletions

diff --git a/paste/util/finddata.py b/paste/util/finddata.py
new file mode 100644
index 0000000..72e257d
--- /dev/null
+++ b/paste/util/finddata.py
@@ -0,0 +1,67 @@
+# Note: you may want to copy this into your setup.py file verbatim, as
+# you can't import this from another package, when you don't know if
+# that package is installed yet.
+
+import os
+from fnmatch import fnmatchcase
+from distutils.util import convert_path
+
+def find_package_data(where='.', package='', wildcards=(),
+                      exclude=('.*', 'CVS', '_darcs'),
+                      only_in_packages=True):
+    """
+    Return a dictionary suitable for use in ``package_data``
+    in a distutils ``setup.py`` file.
+
+    The dictionary looks like::
+
+        {'package': [wildcards]}
+
+    Except for any directories under the package (that aren't
+    themselves subpackages) it will expand the wildcards.  So if you
+    have a package ``pygo`` and a subdirectory ``static`` (with no
+    ``__init__.py`` file, hence not a subpackages) and
+    ``wildcards=('*.css', '*.js')``, then you'll get::
+
+        {'pygo': ['*.css', '*.js', 'static/*.css', 'static/*.js']}
+
+    If ``only_in_packages`` is true, then top-level directories that
+    are not packages won't be included (but directories under packages
+    will).
+
+    Directories matching any pattern in ``exclude`` will be ignored;
+    by default directories with leading ``.``, ``CVS``, and ``_darcs``
+    will be ignored.
+    """
+    out = {}
+    stack = [(convert_path(where), '', package, only_in_packages)]
+    while stack:
+        where, prefix, package, only_in_packages = stack.pop(0)
+        if package or not only_in_packages:
+            out.setdefault(package, []).extend([
+                prefix + wildcard
+                for wildcard in wildcards])
+        for name in os.listdir(where):
+            bad_name = False
+            for pattern in exclude:
+                if fnmatchcase(name, pattern):
+                    bad_name = True
+                    break
+            if bad_name:
+                continue
+            fn = os.path.join(where, name)
+            if os.path.isdir(fn):
+                if os.path.isfile(os.path.join(fn, '__init__.py')):
+                    if not package:
+                        new_package = name
+                    else:
+                        new_package = package + '.' + name
+                    stack.append((fn, '', new_package, False))
+                else:
+                    stack.append((fn, prefix + name + '/', package, only_in_packages))
+    return out
+
+if __name__ == '__main__':
+    import sys, pprint
+    pprint.pprint(
+        find_package_data(wildcards=sys.argv[1:]))