setup.py

from distutils.core import setup
from distutils.command.install_data import install_data
from distutils.command.install import INSTALL_SCHEMES
import os
import sys


# Add a test command to setup.py
# Code borrowed from: http://da44en.wordpress.com/2002/11/22/using-distutils/
from distutils.core import Command
from unittest import TextTestRunner, TestLoader
from glob import glob
from os.path import splitext, basename, join as pjoin, walk
import os


class TestCommand(Command):
    user_options = [ ]

    def initialize_options(self):
        self._dir = os.getcwd()

    def finalize_options(self):
        pass

    def run(self):
        '''
        Finds all the tests modules in tests/, and runs them.
        '''
        testfiles = [ ]
        for t in glob(pjoin(self._dir, 'tests', '*.py')):
            if not t.endswith('__init__.py'):
                testfiles.append('.'.join(
                    ['tests', splitext(basename(t))[0]])
                )

        tests = TestLoader().loadTestsFromNames(testfiles)
        t = TextTestRunner(verbosity = 2)
        t.run(tests)


class CleanCommand(Command):
    user_options = [ ]

    def initialize_options(self):
        self._clean_me = [ ]
        for root, dirs, files in os.walk('.'):
            for f in files:
                if f.endswith('.pyc'):
                    self._clean_me.append(pjoin(root, f))

    def finalize_options(self):
        pass

    def run(self):
        for clean_me in self._clean_me:
            try:
                os.unlink(clean_me)
            except:
                pass




def return_version():
    return __import__('dirtt').get_version()


class osx_install_data(install_data):
    # On MacOS, the platform-specific lib dir is /System/Library/Framework/Python/.../
    # which is wrong. Python 2.5 supplied with MacOS 10.5 has an Apple-specific fix
    # for this in distutils.command.install_data#306. It fixes install_lib but not
    # install_data, which is why we roll our own install_data class.

    def finalize_options(self):
        # By the time finalize_options is called, install.install_lib is set to the
        # fixed directory, so we set the installdir to install_lib. The
        # install_data class uses ('install_data', 'install_dir') instead.
        self.set_undefined_options('install', ('install_lib', 'install_dir'))
        install_data.finalize_options(self)

if sys.platform == "darwin":
    cmdclasses = {'install_data': osx_install_data}
else:
    cmdclasses = {'install_data': install_data}

cmdclasses['test'] = TestCommand
cmdclasses['clean'] = CleanCommand


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)


# Tell distutils to put the data_files in platform-specific installation
# locations. See here for an explanation:
# http://groups.google.com/group/comp.lang.python/browse_thread/thread/35ec7b2fed36eaec/2105ee4d9e8042cb
for scheme in INSTALL_SCHEMES.values():
    scheme['data'] = scheme['purelib']


# Compile the list of packages available, because distutils doesn't have
# an easy way to do this.
packages, data_files = [], []
root_dir = os.path.dirname(__file__)
if root_dir != '':
    os.chdir(root_dir)

dirtt_dir = 'dirtt'

for dirpath, dirnames, filenames in os.walk(dirtt_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:
        data_files.append([dirpath, [os.path.join(dirpath, f) for f in filenames]])

# Small hack for working with bdist_wininst.
# See http://mail.python.org/pipermail/distutils-sig/2004-August/004134.html
if len(sys.argv) > 1 and sys.argv[1] == 'bdist_wininst':
    for file_info in data_files:
        file_info[0] = '\\PURELIB\\%s' % file_info[0]


setup(
    name='python-dirtt',
    packages=packages,
    cmdclass = cmdclasses,
    scripts=['dirtt/scripts/mkproject.py','dirtt/scripts/mktemplate.py','dirtt/scripts/mktree.py'],
    data_files = data_files,
#    data_files=[
#        ('/var/dirtt/dirtt.dtd', ['dirtt/data/dirtt.dtd']),
#        ('/var/dirtt/templates/project.xml', ['dirtt/data/templates/project.xml']),
#        ('/var/dirtt/templates/project_sequence.xml', ['dirtt/data/templates/project_sequence.xml']),
#        ('/var/dirtt/templates/project_shot.xml', ['dirtt/data/templates/project_shot.xml']),
#        ('/var/dirtt/templates/project_work.xml', ['dirtt/data/templates/project_work.xml']),
#        ('/var/dirtt/templates/project_master.xml', ['dirtt/data/templates/project_master.xml']),
#        ('/var/dirtt/templates/project_production.xml', ['dirtt/data/templates/project_production.xml'])
#        ('/var/dirtt/templates/workspace.mel', ['dirtt/data/templates/workspace.mel'])
#    ],
    version=return_version(),
    description="Directory Tree Templater",
    long_description="""
        python-dirtt - Directory Tree Templater
        (c) 2015 Robert Moggach
        Licensed under the MIT license: http://www.opensource.org/licenses/mit-license.php

        Dirtt is a standalone tool and library used to generate
        directory and file structures from xml templates that describe
        repeatedly used filesystem layouts such as project structures
        or elements therein.

        It provides a subclassed implementation of xml.sax.handler ContentHandler
        with internal methods that read,parse,render,and execute builds of
        user defined XML directory tree templates.

        """,
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'Intended Audience :: System Administrators',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python',
        'Natural Language :: English',
        'Operating System :: POSIX :: Linux',
        'Operating System :: MacOS :: MacOS X',
        'Topic :: System :: Systems Administration',
        'Topic :: Text Processing :: Markup :: XML'
    ],
    keywords='filesystem template utilities',
    url='http://robmoggach.github.io/python-dirtt/',
    download_url = 'https://github.com/robmoggach/python-dirtt/tarball/v0.2.0',
    author='Robert Moggach',
    author_email='rob@moggach.com',
    maintainer='Robert Moggach',
    maintainer_email='rob@moggach.com',
    license='MIT'
)