setup.py

#!/usr/bin/env python
#
# setup.py
#
# Copyright (C) 2009 Damien Churchill <damoxc@gmail.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.    See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.    If not, write to:
#   The Free Software Foundation, Inc.,
#   51 Franklin Street, Fifth Floor
#   Boston, MA    02110-1301, USA.
#

import os
import sys
import getpass

sys.path.append(os.path.join(os.path.dirname(__file__), '.pyrex'))

try:
    from setuptools import setup, Extension
except ImportError:
    import ez_setup
    ez_setup.use_setuptools()
    from setuptools import setup, Extension

from distutils import log
from distutils.cmd import Command
from distutils.command.clean import clean as _clean
from Cython.Distutils import build_ext as _build_ext

_extra_compile_args = [
    '-DMSDBLIB'
]

ROOT = os.path.dirname(__file__)
WINDOWS = False

if sys.platform == 'win32':
    WINDOWS = True
    WIN32 = os.path.join(ROOT, 'win32')
    FREETDS = os.path.join(WIN32, 'freetds')
    include_dirs = [os.path.join(FREETDS, 'include')]
    library_dirs = [os.path.join(FREETDS, 'lib')]
    libraries = [
        'libiconv',
        'iconv',
        'sybdb',
        'ws2_32',
        'wsock32',
        'kernel32',
    ]
    
    _extra_compile_args.append('-Wl,-allow-multiple-definition')
    _extra_compile_args.append('-Wl,-subsystem,windows-mthreads')
    _extra_compile_args.append('-mwindows')
    _extra_compile_args.append('-Wl,--strip-all')

else:
    include_dirs = [
        '/usr/local/include', '/usr/local/include/freetds',  # first local install
        '/usr/include', '/usr/include/freetds',   # some generic Linux paths
        '/usr/include/freetds_mssql',             # some versions of Mandriva 
        '/usr/local/freetds/include',             # FreeBSD
        '/usr/pkg/freetds/include'	              # NetBSD
    ]
    library_dirs = [
        '/usr/local/lib', '/usr/local/lib/freetds',
        '/usr/lib64',
        '/usr/lib', '/usr/lib/freetds',
        '/usr/lib/freetds_mssql', 
        '/usr/local/freetds/lib',
        '/usr/pkg/freetds/lib'
    ]
    libraries = [ "sybdb" ]   # on Mandriva you may have to change it to sybdb_mssql

if sys.platform == 'darwin':
    fink = '/sw/'
    include_dirs.insert(0, fink + 'include')
    library_dirs.insert(0, fink + 'lib')

    # some mac ports paths
    include_dirs += [
        '/opt/local/include',
        '/opt/local/include/freetds',
        '/opt/local/freetds/include'
    ]
    library_dirs += [
        '/opt/local/lib',
        '/opt/local/lib/freetds',
        '/opt/local/freetds/lib'
    ]

class build_ext(_build_ext):
    """
    Subclass the Cython build_ext command so it extracts freetds.zip if it
    hasn't already been done.
    """

    def run(self):
        # Not running on windows means we don't want to do this
        if not WINDOWS:
            return _build_ext.run(self)


        if os.path.isdir(FREETDS):
            return _build_ext.run(self)

        log.info('extracting FreeTDS')
        from zipfile import ZipFile
        zip_file = ZipFile(os.path.join(WIN32, 'freetds.zip'))
        for name in zip_file.namelist():
            dest = os.path.normpath(os.path.join(WIN32, name))
            if name.endswith('/'):
                os.makedirs(dest)
            else:
                f = open(dest, 'wb')
                f.write(zip_file.read(name))
                f.close()
        zip_file.close()
        return _build_ext.run(self)

class clean(_clean):
    """
    Subclass clean so it removes all the Cython generated C files.
    """
    
    def run(self):
        _clean.run(self)
        for ext in self.distribution.ext_modules:
            cy_sources = [s for s in ext.sources if s.endswith('.pyx')]
            for cy_source in cy_sources:
                c_source = cy_source[:-3] + 'c'
                if os.path.exists(c_source):
                    log.info('removing %s', c_source)
                    os.remove(c_source)

        # Check if we need to remove the freetds directory
        if WINDOWS:
            # If the directory exists, remove it
            if os.path.isdir(FREETDS):
                import shutil
                shutil.rmtree(FREETDS)

class release(Command):
    """
    Setuptools command to run all the required commands to perform
    a release. This acts differently depending on the platform it
    is being run on.
    """

    description = "Run all the commands required for a release."

    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        self.username = None
        self.password = None
        self.store = None
        
        if WINDOWS:
            self.release_windows()
        else:
            self.release_unix()

    def release_windows(self):
        # generate windows source distributions
        sdist = self.distribution.get_command_obj('sdist')
        sdist.formats = 'zip'
        sdist.ensure_finalized()
        sdist.run()
        
        # generate a windows egg
        self.run_command('bdist_egg')

        # generate windows installers
        bdist = self.reinitialize_command('bdist')
        bdist.formats = 'zip,wininst'
        bdist.ensure_finalized()
        bdist.run()
        
        (name, version, fullname) = self.get_info()
        
        self.upload(fullname + '.zip', '%s %s source zipped' % (name, version))
        self.upload(fullname + '.win32.zip', '%s %s win32 zip installer' % (name, version))
        self.upload(fullname + '.win32-py2.6.exe', '%s %s windows installer' % (name, version))
        self.upload(fullname + '-py2.6-win32.egg', '%s %s windows egg' % (name, version))

    def release_unix(self):
        # generate linux source distributions
        sdist = self.distribution.get_command_obj('sdist')
        sdist.formats = 'gztar,bztar'
        sdist.ensure_finalized()
        sdist.run()
        
        (name, version, fullname) = self.get_info()
        self.upload(fullname + '.tar.gz', '%s %s source gzipped' % (name, version))
        self.upload(fullname + '.tar.bz2', '%s %s source bzipped' % (name, version))
    
    def get_info(self):
        """
        Return the project name and version
        """
        return (
            self.distribution.get_name(),
            self.distribution.get_version(),
            self.distribution.get_fullname()
        )

    def upload(self, filename, comment):
        from gc_upload import upload
        
        if self.username is None:
            username = raw_input('Username: ')
            password = getpass.getpass('Password: ')
            
            if self.store is None:
                store = raw_input('Store credentials for later use? [Y/n]')
                self.store = store in ('', 'y', 'Y')
            
            if self.store:
                self.username = username
                self.password = password
                
        else:
            username = self.username
            password = self.password

        filename = os.path.join('dist', filename)
        log.info('uploading %s to googlecode', filename)
        (status, reason, url) = upload(filename, 'pymssql', username, password, comment)
        if not url:
            log.error('upload to googlecode failed: %s', reason)

setup(
    name  = 'pymssql',
    version = '1.9.909',
    description = 'A simple database interface to MS-SQL for Python.',
    long_description = 'A simple database interface to MS-SQL for Python.',
    author = 'Damien Churchill',
    author_email = 'damoxc@gmail.com',
    license = 'LGPL',
    url = 'http://pymssql.sourceforge.net',
    cmdclass = {
        'build_ext': build_ext,
        'clean': clean,
        'release': release
    },
    data_files = [
        ('', ['_mssql.pyx', 'pymssql.pyx'])
    ],
	zip_safe = False,
    setup_requires=["Cython>=0.13.1"],
    ext_modules = [Extension('_mssql', ['_mssql.pyx'],
                             extra_compile_args = _extra_compile_args,
                             include_dirs = include_dirs,
                             library_dirs = library_dirs,
                             libraries = libraries),
                   Extension('pymssql', ['pymssql.pyx'],
                             extra_compile_args = _extra_compile_args,
                             include_dirs = include_dirs,
                             library_dirs = library_dirs,
                             libraries = libraries)]
)