setup.py

##
# Copyright (c) 2006-2008 Apple Inc. All rights reserved.
#
# 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.
##

from setuptools import setup, Extension
import subprocess
import sys

long_description = """
This Python package is a high-level wrapper for Kerberos (GSSAPI) operations.
The goal is to avoid having to build a module that wraps the entire Kerberos.framework,
and instead offer a limited set of functions that do what is needed for client/server
Kerberos authentication based on <http://www.ietf.org/rfc/rfc4559.txt>.

"""

def check_krb5_config(*options, **kwargs):
    try:
        cmd = kwargs.get('command_name', 'krb5-config')
        process = subprocess.Popen((cmd,) + options, stdout=subprocess.PIPE, universal_newlines=True)
        output, unused_err = process.communicate()
        retcode = process.poll()
        if retcode:
            raise subprocess.CalledProcessError(retcode, cmd, output=output)
        return output.split()
    except OSError as e:
        if e.errno == 2 and cmd != "krb5-config.mit":
            try:
                return check_krb5_config(*options, command_name="krb5-config.mit")
            except OSError as e2:
                if e2.errno == 2:
                    raise Exception("You are missing krb5-config(.mit)")

def check_krb5_version():
    krb5_vers = check_krb5_config("--version")
    if krb5_vers and len(krb5_vers) == 4:
        if int(krb5_vers[3].split('.')[1].split('-')[0]) >= 10:
            return r'-DGSSAPI_EXT'

extra_link_args = check_krb5_config("--libs", "gssapi")
extra_compile_args = check_krb5_config("--cflags", "gssapi")

krb5_ver = check_krb5_version()
if krb5_ver:
    extra_compile_args.append(krb5_ver)

setup (
    name = "pykerberos",
    version = "1.2.2.dev0",
    description = "High-level interface to Kerberos",
    long_description=long_description,
    license="ASL 2.0",
    url="https://github.com/02strich/pykerberos",
    classifiers = [
        "License :: OSI Approved :: Apache Software License",
        "Programming Language :: Python :: 2",
        "Programming Language :: Python :: 3",
        "Topic :: Software Development :: Libraries :: Python Modules",
        "Topic :: System :: Systems Administration :: Authentication/Directory"
        ],
    ext_modules = [
        Extension(
            "kerberos",
            extra_link_args = extra_link_args,
            extra_compile_args = extra_compile_args,
            sources = [
                "src/kerberos.c",
                "src/kerberosbasic.c",
                "src/kerberosgss.c",
                "src/kerberospw.c",
                "src/base64.c"
            ],
        ),
    ],
)