import os

if not hasattr(os.path, 'relpath'): # python < 2.6

# http://www.saltycrane.com/blog/2010/03/ospathrelpath-source-code-python-25/

from posixpath import curdir, sep, pardir, join, abspath, commonprefix

def relpath(path, start=curdir):

"""Return a relative version of a path"""

if not path:

raise ValueError("no path specified")

start_list = abspath(start).split(sep)

path_list = abspath(path).split(sep)

# Work out how much of the filepath is shared by start and path.

i = len(commonprefix([start_list, path_list]))

rel_list = [pardir] * (len(start_list)-i) + path_list[i:]

if not rel_list:

return curdir

return join(*rel_list)

os.path.relpath = relpath

__all__ = ['find_files']

def fullsplit(path, result=None):

"""

Split a pathname into components (the opposite of os.path.join) in a

platform-neutral way.

"""

if result is None:

result = []

head, tail = os.path.split(path)

if head == '':

return [tail] + result

if head == path:

return result

return fullsplit(head, [tail] + result)

def find_packages(package_dir):

"""

Returns a tuple consisting of a ``packages`` list and a ``package_data``

dictionary suitable for passing on to ``distutils.core.setup``

Requires the folder name containing the package files; ``find_files``

assumes that ``setup.py`` is located in the same folder as the folder

containing those files.

Code lifted from Django's ``setup.py``, with improvements by PSyton.

"""

# Compile the list of packages available, because distutils doesn't have

# an easy way to do this.

packages = []

package_data = {}

root_dir = os.path.dirname(__file__)

if root_dir != '':

os.chdir(root_dir)

for dirpath, dirnames, filenames in sorted(os.walk(package_dir)):

# Ignore dirnames that start with '.'

for i, dirname in enumerate(dirnames):

if dirname.startswith('.'): del dirnames[i]

if '__init__.py' in filenames:

packages.append('.'.join(fullsplit(dirpath)))

elif filenames:

cur_pack = packages[0] # Assign all data files to the toplevel package

if cur_pack not in package_data:

package_data[cur_pack] = []

package_dir = os.path.join(*cur_pack.split("."))

dir_relpath = os.path.relpath(dirpath, package_dir)

for f in filenames:

package_data[cur_pack].append(os.path.join(dir_relpath, f))

return packages, package_data