setup.py

#!/usr/bin/env python

# This file is part of pdar.
#
# Copyright 2011-2013 Jason Penney
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""setup.py for pdar"""

from setuptools import setup, find_packages
import distutils
import os
import sys
import imp
import re

if sys.version_info < (2, 6):
    print "Python 2.6 or greater required"
    sys.exit(1)


py2exe = None

if True or sys.platform == 'win32':
    try:
        import py2exe
    except ImportError:
        print "WARNING: py2exe is not available. '.exe' creation disabled."
        py2exe = None

_pkgname = 'pdar'


def get_meta(pkgname, path='.'):
    '''Try to extract metadata fields (__version__, __author__, etc.)
    from a package without importing it'''

    pkgfile, pathname, description = imp.find_module(pkgname)
    if not pkgfile and os.path.isdir(pathname):
        initfile = os.path.join(pathname, '__init__.py')
        if os.path.exists(initfile):
            pkgfile = open(initfile)

    meta = dict()
    if pkgfile:
        match = False
        tripple = False
        matchlines = []
        for line in pkgfile:
            line = line.strip(os.linesep)
            if (line.startswith('__') and '=' in line) \
                   or (match and line.startswith(' ')) \
                   or tripple:
                if line.strip().endswith("'''"):
                    if tripple:
                        line += '.strip(os.linesep)'
                    match = False
                    tripple = not tripple
                matchlines.append(line)

            else:
                match = False
        code_re = re.compile(r'^(__[^ ]+__)', flags=re.M | re.S)
        code = code_re.sub(r"meta['\1']",
                           os.linesep.join(matchlines))
        try:
            exec(code)
        except Exception, err:
            sys.stderr.write("error parsing metadata from '%s': %s"
                             % (pkgname, str(err)))
    return meta

extra = {}
if sys.version_info >= (3,):
    extra.update({'use_2to3': True})

if py2exe:
    script_path = os.path.join('build', 'py2exe', 'scripts', 'winpdar.py')
    distutils.dir_util.mkpath(os.path.dirname(script_path))
    with open(script_path, 'w') as script_file:
        script_file.write('''\
import sys
import pdar.console

if __name__ == "__main__":
    sys.exit(pdar.console.pdar_cmd())
''')
    extra.setdefault('options', {})
    extra['options'].update({
            'py2exe': {
                'optimize': 2,
                'bundle_files': 1,
                'dist_dir': os.path.join('dist', 'py2exe'),
                }})
    extra.update({'zipfile': None,
                  'console': [{'script': script_path,
                               'dest_base': 'pdar'}]})


meta = get_meta(_pkgname)
# since we already quit if Python version < 2.6, we can use `str.format`
default_email = 'jpenney{at}jczorkmid{dot}net'.format(at='@', dot='.')

setup(
    name=_pkgname,
    version=meta.get('__version__', 'unknown'),
    author=meta.get('__author__', 'Jason Penney'),
    author_email=meta.get('__email__', default_email),
    url=meta.get('__url__', 'http://jasonpenney.net/'),
    maintainer=meta.get('__maintainer__', 'Jason Penney'),
    maintainer_email=meta.get('__maintainer_email__',
                              meta.get('__email__',
                                       default_email)),
    description=meta.get('__description__', None),
    long_description=meta.get('__long_description__', None),
    download_url=meta.get('__download_url__', None),
    license=meta.get('__license__', None),
    packages=find_packages(exclude=['tests']),
    install_requires='''
    bsdiff4 >= 1.0.1
    argparse
    ''',
    test_suite='unittest2.collector',
    tests_require='''
    unittest2
    ''',
    entry_points="""
    [console_scripts]
    pdar = pdar.console:pdar_cmd
    """,
    include_package_data=True,
    exclude_package_data={'': ['tests/*']},
    **extra
    )