#!/usr/bin/env python

# Copyright (c) 2009 Giampaolo Rodola'. All rights reserved.

# Use of this source code is governed by a BSD-style license that can be

# found in the LICENSE file.

"""Cross-platform lib for process and system monitoring in Python."""

import contextlib

import io

import os

import platform

import sys

import tempfile

import warnings

with warnings.catch_warnings():

warnings.simplefilter("ignore")

try:

import setuptools

from setuptools import setup, Extension

except ImportError:

setuptools = None

from distutils.core import setup, Extension

HERE = os.path.abspath(os.path.dirname(__file__))

# ...so we can import _common.py

sys.path.insert(0, os.path.join(HERE, "psutil"))

from _common import AIX # NOQA

from _common import BSD # NOQA

from _common import FREEBSD # NOQA

from _common import LINUX # NOQA

from _common import MACOS # NOQA

from _common import NETBSD # NOQA

from _common import OPENBSD # NOQA

from _common import POSIX # NOQA

from _common import SUNOS # NOQA

from _common import WINDOWS # NOQA

macros = []

if POSIX:

macros.append(("PSUTIL_POSIX", 1))

if BSD:

macros.append(("PSUTIL_BSD", 1))

sources = ['psutil/_psutil_common.c']

if POSIX:

sources.append('psutil/_psutil_posix.c')

tests_require = []

if sys.version_info[:2] <= (2, 6):

tests_require.append('unittest2')

if sys.version_info[:2] <= (2, 7):

tests_require.append('mock')

if sys.version_info[:2] <= (3, 2):

tests_require.append('ipaddress')

extras_require = {}

if sys.version_info[:2] <= (3, 3):

extras_require.update(dict(enum='enum34'))

def get_version():

raise ValueError("couldn't find version string")

VERSION = get_version()

macros.append(('PSUTIL_VERSION', int(VERSION.replace('.', ''))))

def get_description():

README = os.path.join(HERE, 'README.rst')

with open(README, 'r') as f:

return f.read()

@contextlib.contextmanager

def silenced_output(stream_name):

setattr(sys, stream_name, orig)

if WINDOWS:

def get_winver():

maj, min = sys.getwindowsversion()[0:2]

return '0x0%s' % ((maj * 100) + min)

if sys.getwindowsversion()[0] < 6:

msg = "warning: Windows versions < Vista are no longer supported or "

msg = "maintained; latest official supported version is psutil 3.4.2; "

msg += "psutil may still be installed from sources if you have "

msg += "Visual Studio and may also (kind of) work though"

warnings.warn(msg, UserWarning)

macros.append(("PSUTIL_WINDOWS", 1))

macros.extend([

# be nice to mingw, see:

# http://www.mingw.org/wiki/Use_more_recent_defined_functions

('_WIN32_WINNT', get_winver()),

('_AVAIL_WINVER_', get_winver()),

('_CRT_SECURE_NO_WARNINGS', None),

# see: https://github.com/giampaolo/psutil/issues/348

('PSAPI_VERSION', 1),

])

ext = Extension(

'psutil._psutil_windows',

sources=sources + [

'psutil/_psutil_windows.c',

'psutil/arch/windows/process_info.c',

'psutil/arch/windows/process_handles.c',

'psutil/arch/windows/security.c',

'psutil/arch/windows/inet_ntop.c',

'psutil/arch/windows/services.c',

],

define_macros=macros,

libraries=[

"psapi", "kernel32", "advapi32", "shell32", "netapi32",

"iphlpapi", "wtsapi32", "ws2_32", "PowrProf",

],

# extra_compile_args=["/Z7"],

# extra_link_args=["/DEBUG"]

)

elif MACOS:

macros.append(("PSUTIL_OSX", 1))

ext = Extension(

'psutil._psutil_osx',

sources=sources + [

'psutil/_psutil_osx.c',

'psutil/arch/osx/process_info.c',

],

define_macros=macros,

extra_link_args=[

'-framework', 'CoreFoundation', '-framework', 'IOKit'

])

elif FREEBSD:

macros.append(("PSUTIL_FREEBSD", 1))

ext = Extension(

'psutil._psutil_bsd',

sources=sources + [

'psutil/_psutil_bsd.c',

'psutil/arch/freebsd/specific.c',

'psutil/arch/freebsd/sys_socks.c',

'psutil/arch/freebsd/proc_socks.c',

],

define_macros=macros,

libraries=["devstat"])

elif OPENBSD:

macros.append(("PSUTIL_OPENBSD", 1))

ext = Extension(

'psutil._psutil_bsd',

sources=sources + [

'psutil/_psutil_bsd.c',

'psutil/arch/openbsd/specific.c',

],

define_macros=macros,

libraries=["kvm"])

elif NETBSD:

macros.append(("PSUTIL_NETBSD", 1))

ext = Extension(

'psutil._psutil_bsd',

sources=sources + [

'psutil/_psutil_bsd.c',

'psutil/arch/netbsd/specific.c',

'psutil/arch/netbsd/socks.c',

],

define_macros=macros,

libraries=["kvm"])

elif LINUX:

def get_ethtool_macro():

pass

macros.append(("PSUTIL_LINUX", 1))

ETHTOOL_MACRO = get_ethtool_macro()

if ETHTOOL_MACRO is not None:

macros.append(ETHTOOL_MACRO)

ext = Extension(

'psutil._psutil_linux',

sources=sources + ['psutil/_psutil_linux.c'],

define_macros=macros)

elif SUNOS:

macros.append(("PSUTIL_SUNOS", 1))

ext = Extension(

'psutil._psutil_sunos',

sources=sources + [

'psutil/_psutil_sunos.c',

'psutil/arch/solaris/v10/ifaddrs.c',

'psutil/arch/solaris/environ.c'

],

define_macros=macros,

libraries=['kstat', 'nsl', 'socket'])

# AIX

elif AIX:

macros.append(("PSUTIL_AIX", 1))

ext = Extension(

'psutil._psutil_aix',

sources=sources + [

'psutil/_psutil_aix.c',

'psutil/arch/aix/net_connections.c',

'psutil/arch/aix/common.c',

'psutil/arch/aix/ifaddrs.c'],

libraries=['perfstat'],

define_macros=macros)

else:

sys.exit('platform %s is not supported' % sys.platform)

if POSIX:

posix_extension = Extension(

'psutil._psutil_posix',

define_macros=macros,

sources=sources)

if SUNOS:

posix_extension.libraries.append('socket')

if platform.release() == '5.10':

posix_extension.sources.append('psutil/arch/solaris/v10/ifaddrs.c')

posix_extension.define_macros.append(('PSUTIL_SUNOS10', 1))

elif AIX:

posix_extension.sources.append('psutil/arch/aix/ifaddrs.c')

extensions = [ext, posix_extension]

else:

extensions = [ext]

def main():

setup(**kwargs)

if __name__ == '__main__':

main()