##// END OF EJS Templates
pykernel handles shutdown_request
pykernel handles shutdown_request

File last commit:

r2498:3eae1372
r3088:b7d82c58
Show More
sysinfo.py
100 lines | 3.3 KiB | text/x-python | PythonLexer
Brian Granger
Work to address the review comments on Fernando's branch....
r2498 # encoding: utf-8
"""
Utilities for getting information about a system.
"""
#-----------------------------------------------------------------------------
# Copyright (C) 2008-2009 The IPython Development Team
#
# Distributed under the terms of the BSD License. The full license is in
# the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
import os
import platform
import sys
import subprocess
from IPython.core import release
#-----------------------------------------------------------------------------
# Code
#-----------------------------------------------------------------------------
def sys_info():
"""Return useful information about IPython and the system, as a string.
Examples
--------
In [1]: print(sys_info())
IPython version: 0.11.bzr.r1340 # random
BZR revision : 1340
Platform info : os.name -> posix, sys.platform -> linux2
: Linux-2.6.31-17-generic-i686-with-Ubuntu-9.10-karmic
Python info : 2.6.4 (r264:75706, Dec 7 2009, 18:45:15)
[GCC 4.4.1]
"""
out = []
out.append('IPython version: %s' % release.version)
out.append('BZR revision : %s' % release.revision)
out.append('Platform info : os.name -> %s, sys.platform -> %s' %
(os.name,sys.platform) )
out.append(' : %s' % platform.platform())
out.append('Python info : %s' % sys.version)
out.append('') # ensure closing newline
return '\n'.join(out)
def _num_cpus_unix():
"""Return the number of active CPUs on a Unix system."""
return os.sysconf("SC_NPROCESSORS_ONLN")
def _num_cpus_darwin():
"""Return the number of active CPUs on a Darwin system."""
p = subprocess.Popen(['sysctl','-n','hw.ncpu'],stdout=subprocess.PIPE)
return p.stdout.read()
def _num_cpus_windows():
"""Return the number of active CPUs on a Windows system."""
return os.environ.get("NUMBER_OF_PROCESSORS")
def num_cpus():
"""Return the effective number of CPUs in the system as an integer.
This cross-platform function makes an attempt at finding the total number of
available CPUs in the system, as returned by various underlying system and
python calls.
If it can't find a sensible answer, it returns 1 (though an error *may* make
it return a large positive number that's actually incorrect).
"""
# Many thanks to the Parallel Python project (http://www.parallelpython.com)
# for the names of the keys we needed to look up for this function. This
# code was inspired by their equivalent function.
ncpufuncs = {'Linux':_num_cpus_unix,
'Darwin':_num_cpus_darwin,
'Windows':_num_cpus_windows,
# On Vista, python < 2.5.2 has a bug and returns 'Microsoft'
# See http://bugs.python.org/issue1082 for details.
'Microsoft':_num_cpus_windows,
}
ncpufunc = ncpufuncs.get(platform.system(),
# default to unix version (Solaris, AIX, etc)
_num_cpus_unix)
try:
ncpus = max(1,int(ncpufunc()))
except:
ncpus = 1
return ncpus