setup.py

import os
import re
import subprocess
import sys

from setuptools import setup, find_packages, Command
from setuptools.command.test import test as TestCommand

try:
    with open('pyramid_orb/_version.py', 'r') as f:
        content = f.read()
        major = re.search('__major__ = (\d+)', content).group(1)
        minor = re.search('__minor__ = (\d+)', content).group(1)
        rev = re.search('__revision__ = "(\d+)"', content).group(1)
        version = '.'.join((major, minor, rev))
except StandardError:
     version = '0.0.0'

class tag(Command):
    description = 'Command used to release new versions of the website to the internal pypi server.'
    user_options = [
        ('no-tag', None, 'Do not tag the repo before releasing')
    ]

    def initialize_options(self):
        self.no_tag = False

    def finalize_options(self):
        pass

    def run(self):
        # generate the version information from the current git commit
        cmd = ['git', 'describe', '--match', 'v[0-9]*.[0-9]*.0']
        desc = subprocess.check_output(cmd).strip()
        result = re.match('v([0-9]+)\.([0-9]+)\.0-([0-9]+)-(.*)', desc)

        print 'generating version information from:', desc
        with open('./pyramid_orb/_version.py', 'w') as f:
            f.write('__major__ = {0}\n'.format(result.group(1)))
            f.write('__minor__ = {0}\n'.format(result.group(2)))
            f.write('__revision__ = "{0}"\n'.format(result.group(3)))
            f.write('__hash__ = "{0}"'.format(result.group(4)))

        # tag this new release version
        if not self.no_tag:
            version = '.'.join([result.group(1), result.group(2), result.group(3)])

            print 'creating git tag:', 'v' + version

            os.system('git tag -a v{0} -m "releasing {0}"'.format(version))
            os.system('git push --tags')
        else:
            print 'warning: tagging ignored...'


setup(
    name='pyramid_orb',
    version=version,
    author='Eric Hulser',
    author_email='eric.hulser@gmail.com',
    maintainer='Eric Hulser',
    maintainer_email='eric.hulser@gmail.com',
    description='Bindings for the pyramid webframework and the ORB database ORM library.',
    license='MIT',
    keywords='',
    url='https://github.com/ProjexSoftware/pyramid_orb',
    include_package_data=True,
    packages=find_packages(),
    install_requires=[
        'projex',
        'orb-api',
        'pyramid',
        'pyramid_restful'
    ],
    cmdclass={
        'tag': tag
    },
    long_description='Bindings for the pyramid webframework and the ORB database ORM library.',
    classifiers=[],
)