FilesExpand file tree

 master

/

sysinfo.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

167 lines (133 loc) · 5.08 KB

 master

/

sysinfo.py

Copy path

Top

File metadata and controls

• Code
• Blame

167 lines (133 loc) · 5.08 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

# encoding: utf-8

"""

Utilities for getting information about IPython and the system it's running in.

"""

#-----------------------------------------------------------------------------

# Copyright (C) 2008-2011 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 pprint

import sys

import subprocess

from IPython.core import release

from IPython.utils import py3compat, _sysinfo, encoding

#-----------------------------------------------------------------------------

# Code

#-----------------------------------------------------------------------------

def pkg_commit_hash(pkg_path):

"""Get short form of commit hash given directory `pkg_path`

We get the commit hash from (in order of preference):

* IPython.utils._sysinfo.commit

* git output, if we are in a git repository

If these fail, we return a not-found placeholder tuple

Parameters

----------

pkg_path : str

directory containing package

only used for getting commit from active repo

Returns

-------

hash_from : str

Where we got the hash from - description

hash_str : str

short form of hash

"""

# Try and get commit from written commit text file

if _sysinfo.commit:

return "installation", _sysinfo.commit

# maybe we are in a repository

proc = subprocess.Popen('git rev-parse --short HEAD',

stdout=subprocess.PIPE,

stderr=subprocess.PIPE,

cwd=pkg_path, shell=True)

repo_commit, _ = proc.communicate()

if repo_commit:

return 'repository', repo_commit.strip().decode('ascii')

return '(none found)', u'<not found>'

def pkg_info(pkg_path):

"""Return dict describing the context of this package

Parameters

----------

pkg_path : str

path containing __init__.py for package

Returns

-------

context : dict

with named parameters of interest

"""

src, hsh = pkg_commit_hash(pkg_path)

return dict(

ipython_version=release.version,

ipython_path=pkg_path,

commit_source=src,

commit_hash=hsh,

sys_version=sys.version,

sys_executable=sys.executable,

sys_platform=sys.platform,

platform=platform.platform(),

os_name=os.name,

default_encoding=encoding.DEFAULT_ENCODING,

)

def get_sys_info():

"""Return useful information about IPython and the system, as a dict."""

p = os.path

path = p.realpath(p.dirname(p.abspath(p.join(__file__, '..'))))

return pkg_info(path)

@py3compat.doctest_refactor_print

def sys_info():

"""Return useful information about IPython and the system, as a string.

Examples

--------

::

In [2]: print sys_info()

{'commit_hash': '144fdae', # random

'commit_source': 'repository',

'ipython_path': '/home/fperez/usr/lib/python2.6/site-packages/IPython',

'ipython_version': '0.11.dev',

'os_name': 'posix',

'platform': 'Linux-2.6.35-22-generic-i686-with-Ubuntu-10.10-maverick',

'sys_executable': '/usr/bin/python',

'sys_platform': 'linux2',

'sys_version': '2.6.6 (r266:84292, Sep 15 2010, 15:52:39) \\n[GCC 4.4.5]'}

"""

return pprint.pformat(get_sys_info())

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

}

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