[Pymoul-svn] SF.net SVN: pymoul: [202] enumprocess
Status: Alpha
Brought to you by:
tiran
From: <ti...@us...> - 2007-02-27 14:29:25
|
Revision: 202 http://pymoul.svn.sourceforge.net/pymoul/?rev=202&view=rev Author: tiran Date: 2007-02-27 06:29:15 -0800 (Tue, 27 Feb 2007) Log Message: ----------- Initial import of enumprocess Added Paths: ----------- enumprocess/ enumprocess/branches/ enumprocess/tags/ enumprocess/trunk/ enumprocess/trunk/CHANGES.txt enumprocess/trunk/INSTALL.txt enumprocess/trunk/LICENSE.txt enumprocess/trunk/MANIFEST.in enumprocess/trunk/Makefile enumprocess/trunk/PKG-INFO enumprocess/trunk/README.txt enumprocess/trunk/ez_setup.py enumprocess/trunk/setup.py enumprocess/trunk/src/ enumprocess/trunk/src/enumprocess/ enumprocess/trunk/src/enumprocess/__init__.py enumprocess/trunk/src/enumprocess/processinfo.py enumprocess/trunk/src/enumprocess.egg-info/ enumprocess/trunk/src/enumprocess.egg-info/PKG-INFO enumprocess/trunk/src/enumprocess.egg-info/SOURCES.txt enumprocess/trunk/src/enumprocess.egg-info/dependency_links.txt enumprocess/trunk/src/enumprocess.egg-info/top_level.txt enumprocess/trunk/src/enumprocess.egg-info/zip-safe enumprocess/trunk/version.txt Added: enumprocess/trunk/CHANGES.txt =================================================================== Added: enumprocess/trunk/INSTALL.txt =================================================================== --- enumprocess/trunk/INSTALL.txt (rev 0) +++ enumprocess/trunk/INSTALL.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,13 @@ +============ +Requirements +============ + + * Python 2.4+ http://www.python.org/ + * easy_install http://peak.telecommunity.com/DevCenter/EasyInstall + * setuptools (via easy install) + +============ +Installation +============ + + $ python setup.py install Added: enumprocess/trunk/LICENSE.txt =================================================================== --- enumprocess/trunk/LICENSE.txt (rev 0) +++ enumprocess/trunk/LICENSE.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,29 @@ +Copyright (c) 2007 Christian Heimes. All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions are +met: + + 1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in + the documentation and/or other materials provided with the + distribution. + + 3. Neither the name of Infrae nor the names of its contributors may + be used to endorse or promote products derived from this software + without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL INFRAE OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, +EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, +PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR +PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF +LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. Added: enumprocess/trunk/MANIFEST.in =================================================================== --- enumprocess/trunk/MANIFEST.in (rev 0) +++ enumprocess/trunk/MANIFEST.in 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,3 @@ +include *.txt setup.py ez_setup.py Makefile +recursive-include src/enumprocess *.py + Added: enumprocess/trunk/Makefile =================================================================== --- enumprocess/trunk/Makefile (rev 0) +++ enumprocess/trunk/Makefile 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,54 @@ +PYTHON?=python2.5 +PYTHON24=python2.4 +PYTHON25=python2.5 +TESTFLAGS=-v +TESTOPTS= +SETUPFLAGS= +KEYID=AD16AB1B + +all: inplace + +# Build in-place +inplace: + $(PYTHON) setup.py $(SETUPFLAGS) build_ext -i + +build: + $(PYTHON) setup.py $(SETUPFLAGS) build + +egg24: + $(PYTHON24) setup.py $(SETUPFLAGS) bdist_egg + +egg25: + $(PYTHON25) setup.py $(SETUPFLAGS) bdist_egg + +dist25: + $(PYTHON25) setup.py $(SETUPFLAGS) bdist_egg \ + sdist --formats=zip,gztar + +dist_upload: realclean + $(PYTHON25) setup.py $(SETUPFLAGS) bdist_egg \ + sdist --formats=zip,gztar \ + upload --sign --identity=$(KEYID) + +# What should the default be? +test: + $(PYTHON) src/enumprocess/processinfo.py + +egg: egg24 egg25 + +distribution: realclean dist25 egg24 gpg + +gpg: + for F in `find dist -type f -and -not -name '*.asc'`; do \ + gpg --detach-sign --armor --default-key $(KEYID) $$F; \ + done + md5sum dist/*.zip dist/*.egg dist/*.gz + +clean: + find . \( -name '*.o' -o -name '*~' -o -name '*.so' -o -name '*.py[cod]' -o -name '*.dll' \) -exec rm -f {} \; + rm -rf build + +realclean: clean + rm -f TAGS + rm -rf dist + $(PYTHON) setup.py clean -a Added: enumprocess/trunk/PKG-INFO =================================================================== --- enumprocess/trunk/PKG-INFO (rev 0) +++ enumprocess/trunk/PKG-INFO 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,24 @@ +Metadata-Version: 1.0 +Name: enumprocess +Version: 0.1 +Summary: enumprocess: list pids and processes +Home-page: http://sourceforge.net/projects/pymoul/ +Author: Christian Heimes +Author-email: chr...@ch... +License: BSD License +Download-URL: http://cheeseshop.python.org/pypi/enumprocess +Description: + TODO: long description + +Keywords: pid,process,enum,proc,ps,psapi +Platform: Independant +Classifier: Development Status :: 4 - Beta +Classifier: Intended Audience :: Developers +Classifier: License :: OSI Approved :: BSD License +Classifier: Natural Language :: English +Classifier: Operating System :: Microsoft :: Windows +Classifier: Operating System :: POSIX +Classifier: Operating System :: POSIX :: Linux +Classifier: Programming Language :: Python +Classifier: Topic :: System +Classifier: Topic :: Software Development :: Libraries Added: enumprocess/trunk/README.txt =================================================================== --- enumprocess/trunk/README.txt (rev 0) +++ enumprocess/trunk/README.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,17 @@ +enumprocess +=========== + +The package contains for implementations: + + - an Unsupported implementation that raises UnsupportedError + - a Linux implementation that read the data from /proc + - a Unix/POSIX implementation that parses the output of ps + - a Windows implementation that uses ctypes to get the infos from psapi.dll + +API +=== + +getPids() - list of ints/longs +getPidNames() - mapping pid -> name +getPidDetails(pid) - detailed informations about a process +getPidDetails('self') - detailed informations about current process \ No newline at end of file Added: enumprocess/trunk/ez_setup.py =================================================================== --- enumprocess/trunk/ez_setup.py (rev 0) +++ enumprocess/trunk/ez_setup.py 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,228 @@ +#!python +"""Bootstrap setuptools installation + +If you want to use setuptools in your package's setup.py, just include this +file in the same directory with it, and add this to the top of your setup.py:: + + from ez_setup import use_setuptools + use_setuptools() + +If you want to require a specific version of setuptools, set a download +mirror, or use an alternate download directory, you can do so by supplying +the appropriate options to ``use_setuptools()``. + +This file can also be run as a script to install or upgrade setuptools. +""" +import sys +DEFAULT_VERSION = "0.6c5" +DEFAULT_URL = "http://cheeseshop.python.org/packages/%s/s/setuptools/" % sys.version[:3] + +md5_data = { + 'setuptools-0.6b1-py2.3.egg': '8822caf901250d848b996b7f25c6e6ca', + 'setuptools-0.6b1-py2.4.egg': 'b79a8a403e4502fbb85ee3f1941735cb', + 'setuptools-0.6b2-py2.3.egg': '5657759d8a6d8fc44070a9d07272d99b', + 'setuptools-0.6b2-py2.4.egg': '4996a8d169d2be661fa32a6e52e4f82a', + 'setuptools-0.6b3-py2.3.egg': 'bb31c0fc7399a63579975cad9f5a0618', + 'setuptools-0.6b3-py2.4.egg': '38a8c6b3d6ecd22247f179f7da669fac', + 'setuptools-0.6b4-py2.3.egg': '62045a24ed4e1ebc77fe039aa4e6f7e5', + 'setuptools-0.6b4-py2.4.egg': '4cb2a185d228dacffb2d17f103b3b1c4', + 'setuptools-0.6c1-py2.3.egg': 'b3f2b5539d65cb7f74ad79127f1a908c', + 'setuptools-0.6c1-py2.4.egg': 'b45adeda0667d2d2ffe14009364f2a4b', + 'setuptools-0.6c2-py2.3.egg': 'f0064bf6aa2b7d0f3ba0b43f20817c27', + 'setuptools-0.6c2-py2.4.egg': '616192eec35f47e8ea16cd6a122b7277', + 'setuptools-0.6c3-py2.3.egg': 'f181fa125dfe85a259c9cd6f1d7b78fa', + 'setuptools-0.6c3-py2.4.egg': 'e0ed74682c998bfb73bf803a50e7b71e', + 'setuptools-0.6c3-py2.5.egg': 'abef16fdd61955514841c7c6bd98965e', + 'setuptools-0.6c4-py2.3.egg': 'b0b9131acab32022bfac7f44c5d7971f', + 'setuptools-0.6c4-py2.4.egg': '2a1f9656d4fbf3c97bf946c0a124e6e2', + 'setuptools-0.6c4-py2.5.egg': '8f5a052e32cdb9c72bcf4b5526f28afc', + 'setuptools-0.6c5-py2.3.egg': 'ee9fd80965da04f2f3e6b3576e9d8167', + 'setuptools-0.6c5-py2.4.egg': 'afe2adf1c01701ee841761f5bcd8aa64', + 'setuptools-0.6c5-py2.5.egg': 'a8d3f61494ccaa8714dfed37bccd3d5d', +} + +import sys, os + +def _validate_md5(egg_name, data): + if egg_name in md5_data: + from md5 import md5 + digest = md5(data).hexdigest() + if digest != md5_data[egg_name]: + print >>sys.stderr, ( + "md5 validation of %s failed! (Possible download problem?)" + % egg_name + ) + sys.exit(2) + return data + + +def use_setuptools( + version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir, + download_delay=15 +): + """Automatically find/download setuptools and make it available on sys.path + + `version` should be a valid setuptools version number that is available + as an egg for download under the `download_base` URL (which should end with + a '/'). `to_dir` is the directory where setuptools will be downloaded, if + it is not already available. If `download_delay` is specified, it should + be the number of seconds that will be paused before initiating a download, + should one be required. If an older version of setuptools is installed, + this routine will print a message to ``sys.stderr`` and raise SystemExit in + an attempt to abort the calling script. + """ + try: + import setuptools + if setuptools.__version__ == '0.0.1': + print >>sys.stderr, ( + "You have an obsolete version of setuptools installed. Please\n" + "remove it from your system entirely before rerunning this script." + ) + sys.exit(2) + except ImportError: + egg = download_setuptools(version, download_base, to_dir, download_delay) + sys.path.insert(0, egg) + import setuptools; setuptools.bootstrap_install_from = egg + + import pkg_resources + try: + pkg_resources.require("setuptools>="+version) + + except pkg_resources.VersionConflict, e: + # XXX could we install in a subprocess here? + print >>sys.stderr, ( + "The required version of setuptools (>=%s) is not available, and\n" + "can't be installed while this script is running. Please install\n" + " a more recent version first.\n\n(Currently using %r)" + ) % (version, e.args[0]) + sys.exit(2) + +def download_setuptools( + version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir, + delay = 15 +): + """Download setuptools from a specified location and return its filename + + `version` should be a valid setuptools version number that is available + as an egg for download under the `download_base` URL (which should end + with a '/'). `to_dir` is the directory where the egg will be downloaded. + `delay` is the number of seconds to pause before an actual download attempt. + """ + import urllib2, shutil + egg_name = "setuptools-%s-py%s.egg" % (version,sys.version[:3]) + url = download_base + egg_name + saveto = os.path.join(to_dir, egg_name) + src = dst = None + if not os.path.exists(saveto): # Avoid repeated downloads + try: + from distutils import log + if delay: + log.warn(""" +--------------------------------------------------------------------------- +This script requires setuptools version %s to run (even to display +help). I will attempt to download it for you (from +%s), but +you may need to enable firewall access for this script first. +I will start the download in %d seconds. + +(Note: if this machine does not have network access, please obtain the file + + %s + +and place it in this directory before rerunning this script.) +---------------------------------------------------------------------------""", + version, download_base, delay, url + ); from time import sleep; sleep(delay) + log.warn("Downloading %s", url) + src = urllib2.urlopen(url) + # Read/write all in one block, so we don't create a corrupt file + # if the download is interrupted. + data = _validate_md5(egg_name, src.read()) + dst = open(saveto,"wb"); dst.write(data) + finally: + if src: src.close() + if dst: dst.close() + return os.path.realpath(saveto) + +def main(argv, version=DEFAULT_VERSION): + """Install or upgrade setuptools and EasyInstall""" + + try: + import setuptools + except ImportError: + egg = None + try: + egg = download_setuptools(version, delay=0) + sys.path.insert(0,egg) + from setuptools.command.easy_install import main + return main(list(argv)+[egg]) # we're done here + finally: + if egg and os.path.exists(egg): + os.unlink(egg) + else: + if setuptools.__version__ == '0.0.1': + # tell the user to uninstall obsolete version + use_setuptools(version) + + req = "setuptools>="+version + import pkg_resources + try: + pkg_resources.require(req) + except pkg_resources.VersionConflict: + try: + from setuptools.command.easy_install import main + except ImportError: + from easy_install import main + main(list(argv)+[download_setuptools(delay=0)]) + sys.exit(0) # try to force an exit + else: + if argv: + from setuptools.command.easy_install import main + main(argv) + else: + print "Setuptools version",version,"or greater has been installed." + print '(Run "ez_setup.py -U setuptools" to reinstall or upgrade.)' + + + +def update_md5(filenames): + """Update our built-in md5 registry""" + + import re + from md5 import md5 + + for name in filenames: + base = os.path.basename(name) + f = open(name,'rb') + md5_data[base] = md5(f.read()).hexdigest() + f.close() + + data = [" %r: %r,\n" % it for it in md5_data.items()] + data.sort() + repl = "".join(data) + + import inspect + srcfile = inspect.getsourcefile(sys.modules[__name__]) + f = open(srcfile, 'rb'); src = f.read(); f.close() + + match = re.search("\nmd5_data = {\n([^}]+)}", src) + if not match: + print >>sys.stderr, "Internal error!" + sys.exit(2) + + src = src[:match.start(1)] + repl + src[match.end(1):] + f = open(srcfile,'w') + f.write(src) + f.close() + + +if __name__=='__main__': + if len(sys.argv)>2 and sys.argv[1]=='--md5update': + update_md5(sys.argv[2:]) + else: + main(sys.argv[1:]) + + + + + Property changes on: enumprocess/trunk/ez_setup.py ___________________________________________________________________ Name: svn:executable + Added: enumprocess/trunk/setup.py =================================================================== --- enumprocess/trunk/setup.py (rev 0) +++ enumprocess/trunk/setup.py 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,64 @@ +#!/usr/bin/env python +"""enumprocess: list pids and processes + +TODO: long description +""" +__author__ = "Christian Heimes" +__version__ = "$Id$" +__revision__ = "$Revision$" + +import sys + +if sys.version_info < (2,4): + raise RuntimeError("Python 2.4+ required:\n%s" % sys.version) + +# boot strap easy setup +SETUPTOOLS_VERSION = "0.6c1" +from ez_setup import use_setuptools +use_setuptools(version=SETUPTOOLS_VERSION) + +# import the rest +from setuptools import setup +from setuptools import find_packages + +from distutils.core import setup + +VERSION = "0.1" + +me = "Christian Heimes" +email = "chr...@ch..." + +setup_infos = dict( + name = "enumprocess", + version = VERSION, + description = __doc__[:__doc__.find('\n')].strip(), + long_description = '\n'.join([line + for line in __doc__.split('\n')[1:]]), + author = "Christian Heimes", + author_email = "chr...@ch...", + maintainer = "Christian Heimes", + maintainer_email = "chr...@ch...", + url = "http://sourceforge.net/projects/pymoul/", + download_url= "http://cheeseshop.python.org/pypi/enumprocess", + license = "BSD License", + keywords = ["pid", "process", "enum", "proc", "ps", "psapi"], + platforms=['Independant'], + classifiers = ( + 'Development Status :: 4 - Beta', + 'Intended Audience :: Developers', + 'License :: OSI Approved :: BSD License', + 'Natural Language :: English', + 'Operating System :: Microsoft :: Windows', + 'Operating System :: POSIX', + 'Operating System :: POSIX :: Linux', + 'Programming Language :: Python', + 'Topic :: System', + 'Topic :: Software Development :: Libraries' + ), + setup_requires = ["setuptools>="+SETUPTOOLS_VERSION,], + packages = ['enumprocess'], + package_dir = {'' : 'src'}, + zip_safe = True, +) + +setup(**setup_infos) Property changes on: enumprocess/trunk/setup.py ___________________________________________________________________ Name: svn:executable + Added: enumprocess/trunk/src/enumprocess/__init__.py =================================================================== --- enumprocess/trunk/src/enumprocess/__init__.py (rev 0) +++ enumprocess/trunk/src/enumprocess/__init__.py 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,32 @@ +# Copyright (c) 2007 Christian Heimes. All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are +# met: +# +# 1. Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# +# 2. Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in +# the documentation and/or other materials provided with the +# distribution. +# +# 3. Neither the name of Infrae nor the names of its contributors may +# be used to endorse or promote products derived from this software +# without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL INFRAE OR +# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, +# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, +# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR +# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF +# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +"""enum processes +""" +from processinfo import * Added: enumprocess/trunk/src/enumprocess/processinfo.py =================================================================== --- enumprocess/trunk/src/enumprocess/processinfo.py (rev 0) +++ enumprocess/trunk/src/enumprocess/processinfo.py 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,477 @@ +# Copyright (c) 2007 Christian Heimes. All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are +# met: +# +# 1. Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# +# 2. Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in +# the documentation and/or other materials provided with the +# distribution. +# +# 3. Neither the name of Infrae nor the names of its contributors may +# be used to endorse or promote products derived from this software +# without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL INFRAE OR +# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, +# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, +# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR +# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF +# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +"""Get process informations + +The module contains for implementations: + + - an Unsupported implementation that raises UnsupportedError + - a Linux implementation that read the data from /proc + - a Unix/POSIX implementation that parses the output of ps + - a Windows implementation that uses ctypes to get the infos from psapi.dll + +API +=== + getPids() - list of ints/longs + getPidNames() - mapping pid -> name + getPidDetails(pid) - detailed informations about a process + getPidDetails('self') - detailed informations about current process + +>>> cur = os.getpid() +>>> exe = sys.executable +>>> pids = getPids() +>>> pids > 1 +True +>>> isinstance(pids[0], (int, long)) +True +>>> cur in pids +True + +>>> mapping = getPidNames() +>>> cur in mapping +True +>>> mapping[cur].lower() in sys.executable +True + +>>> getPidDetails('self')['name'] == getPidDetails(cur)['name'] +True +>>> getPidDetails(cur)['name'] == mapping[cur] +True + +>>> for impl in (WinEnumProcesses(), LinuxProcReader(), PsParser()): +... if impl.supported(): +... isinstance(impl.getPids(), list) and None +... isinstance(impl.getPidNames(), dict) and None +""" +__author__ = "Christian Heimes" +__version__ = "$Id$" +__revision__ = "$Revision$" +__all__ = ['getPids', 'getPidNames', 'getPidDetails', 'supportedOS', + 'UnsupportedError'] + +import logging +import os +import os.path +import sys + + +LOG = logging.getLogger("processinfo") +NULL = "\x00" + +_plat = sys.platform.startswith +if _plat('win') or _plat('cygwin'): + from ctypes import windll, c_ulong, sizeof, c_buffer, byref + PSAPI = windll.psapi + KERNEL = windll.kernel32 + PROCESS_QUERY_INFORMATION = 0x0400 + PROCESS_VM_READ = 0x0010 + PROCESS_FLAGS = PROCESS_QUERY_INFORMATION | PROCESS_VM_READ +else: + from subprocess import Popen + from subprocess import PIPE + +class UnsupportedError(OSError): + """OS or command not supported error + """ + pass + +class Unsupported(object): + """Unsupported OS implementation + + Raises L{UnsupportedError} + """ + __slots__ = () + + @classmethod + def supported(cls): + """Supported flag property""" + return False + + @classmethod + def log(cls): + """Log information about the implementation""" + LOG.warning("Unsupported OS. Neither proc filesystem nor 'ps' works.") + + @classmethod + def getPids(cls): + """Get a list of pids + + @return: a list of pid numbers + @rtype: list(int) + """ + raise UnsupportedError + + @classmethod + def getPidNames(cls): + """Get a mapping of pid -> name + + @return: mapping of pid -> name + @rtype: dict(int:str) + """ + raise UnsupportedError + + @classmethod + def getPidDetails(pid): + """Get detailed informations about a process + + @param pid: pid number or 'self' + @type pid: int or basestring + @return: detailed information about a process + @rtype: dict(str:data) + """ + raise UnsupportedError + +class PsParser(object): + """Parse the output of the ps command + """ + __slots__ = () + + CMD = "ps -e --no-header --cols=1024" + PIDNAMES = "%s --format=pid,ucmd" % CMD + PIDS = "%s --format=pid" % CMD + + @classmethod + def supported(cls): + """Supported flag property""" + try: + cls._exec(cls.PIDS) + except Exception: # catch all! + return False + else: + return True + + @classmethod + def log(cls): + """Log information about the implementation""" + LOG.debug("Using the 'ps' command on POSIX os") + + @classmethod + def getPids(cls): + """Get a list of pids + """ + stdout = cls._exec(cls.PIDS) + if stdout is None: + return None + pids = [] + for line in stdout: + try: + pid = int(line.strip()) + except ValueError: + pass + else: + pids.append(pid) + return pids + + @classmethod + def getPidNames(cls): + """Get a list of pid -> name + """ + stdout = cls._exec(cls.PIDNAMES) + if stdout is None: + return None + mapping = {} + for line in stdout: + line = line.strip() + idx = line.find(' ') + pid, name = line[:idx], line[idx+1:] + try: + pid = int(pid) + except ValueError: + pass + else: + mapping[pid] = name + return mapping + + @classmethod + def getPidDetails(cls, pid): + """Get detailed informations about a process + + TODO + """ + if pid == 'self': + pid = os.getpid() + raise UnsupportedError("not implemented yet") + + @staticmethod + def _exec(cmd): + """Execute command cmd + + The method waits until the command has finished. It returns None of + something went wrong. + + @param cmd: Command to execute + @type cmd: str + @return: None or stdin as file like object + """ + try: + popen = Popen(cmd, shell=True, bufsize=-1, stdout=PIPE, + env = {'LC_ALL' : 'C'}) + rc = popen.wait() + except (OSError, ValueError): + LOG.exception("Failed to execute '%s'" % cmd) + return None + else: + if rc != 0: + LOG.error("'%s' returned with error code %i" % (cmd, rc)) + return None + else: + return popen.stdout + +class LinuxProcReader(object): + """Get process informations under Linux by reading /proc + + Tested under Linux, may work on other POSIX os with /proc, too. + """ + __slots__ = () + + PROC = "/proc" + + @classmethod + def supported(cls): + return os.path.isfile("%s/self/status" % cls.PROC) + + @classmethod + def log(cls): + """Log information about the implementation""" + LOG.debug("Using the proc filesystem on Linux") + + @classmethod + def getPids(cls): + """Get a list of pids + """ + pids = [] + for name in os.listdir(cls.PROC): + if os.path.isdir(cls.PROC + '/' + name): + try: + pids.append(int(name)) + except ValueError: + pass + return pids + + @classmethod + def getPidNames(cls): + """Get a list of pid -> name + """ + mapping = {} + for pid in cls.getPids(): + name = cls._readProcStatus(pid, searchkey='name') + if name is not None: + mapping[pid] = name + return mapping + + def getPidDetails(cls, pid): + """Get detailed informations about a process + """ + details = cls._readProcStatus(pid) + if details is None: + return None + details.update(cls._readProcOther(pid)) + details['cmdline'] = cls._readProcCmdline(pid) + return details + + @classmethod + def _readProcStatus(cls, pid, searchkey=None): + """Read and parse status informations for PID pid + + pid - pid as long or int or 'self' + + If searchkey is None the method returns a mapping of lower keys + to values (stripped). + If searchkey is given than the method immediatly returns the value + or returns None if they key can't be found. + """ + mapping = {} + status = '%s/%s/status' % (cls.PROC, pid) + try: + # read entiry file to avoid race conditions + lines = open(status, 'r').readlines() + except IOError: + LOG.exception("%s not found" % status) + return None + for line in lines: + try: + key, value = line.split(':\t') + except ValueError: + continue + key = key.lower() + value = value.strip() + #if value.endswith(' kB'): + # value = value[:-3] + if key == searchkey: + return value + mapping[key.lower()] = value + if searchkey is not None: + return None + return mapping + + @classmethod + def _readProcCmdline(cls, pid): + """Read cmdline informations for pid and returns a list similar to sys.argv + """ + try: + # read entiry file to avoid race conditions + data = open('%s/%s/cmdline' % (cls.PROC, pid), 'r').read() + except IOError: + return None + return data.split(NULL) + + @classmethod + def _readProcOther(cls, pid): + """Read other possible useful things + + cwd -> current work directory (may not exist) + exe -> path to executable (may not exist) + """ + return { + 'cwd' : os.path.realpath('%s/%s/cwd' % (cls.PROC, pid)), + 'exe' : os.path.realpath('%s/%s/exe' % (cls.PROC, pid)), + } + +class WinEnumProcesses(object): + """""Get process informations under Win32 with psapi.dll + + Based on enumprocesses from Eric Koome + http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/305279 + """ + __slots__ = () + + @classmethod + def supported(cls): + try: + cls.getPids() + except Exception: # catch all! + return False + else: + return True + + @classmethod + def log(cls): + """Log information about the implementation""" + LOG.debug("Using ctypes on Windows") + + @classmethod + def getPids(cls): + """Get a list of pids + """ + arr = c_ulong * 256 + lpidProcess = arr() + cb = sizeof(lpidProcess) + cbNeeded = c_ulong() + + # Call Enumprocesses to get hold of process id's + PSAPI.EnumProcesses(byref(lpidProcess), cb, byref(cbNeeded)) + nReturned = cbNeeded.value/sizeof(c_ulong()) # number of processes returned + return [pid for pid in lpidProcess][:nReturned] + + @classmethod + def getPidNames(cls): + """Get a list of pid -> name + """ + hModule = c_ulong() + count = c_ulong() + modname = c_buffer(51) + mapping = {} + for pid in cls.getPids(): + # Get handle to the process based on PID + hProcess = KERNEL.OpenProcess(PROCESS_FLAGS, False, pid) + if hProcess: + PSAPI.EnumProcessModules(hProcess, byref(hModule), + sizeof(hModule), byref(count)) + PSAPI.GetModuleBaseNameA(hProcess, hModule.value, modname, + sizeof(modname)) + try: + name = u"".join([c for c in modname if c != NULL]) + except UnicodeError, msg: + LOG.exception("Can't decode name of pid %s" % pid) + else: + mapping[pid] = name + modname[:] = sizeof(modname) * NULL + KERNEL.CloseHandle(hProcess) + + return mapping + + @classmethod + def getPidDetails(cls, pid): + """Get detailed informations about a process + """ + if pid == 'self': + pid = os.getpid() + + hModule = c_ulong() + count = c_ulong() + modname = c_buffer(51) + hProcess = KERNEL.OpenProcess(PROCESS_FLAGS, False, pid) + if not hProcess: + return None + + PSAPI.EnumProcessModules(hProcess, byref(hModule), + sizeof(hModule), byref(count)) + PSAPI.GetModuleBaseNameA(hProcess, hModule.value, modname, + sizeof(modname)) + try: + name = u"".join([c for c in modname if c != NULL]) + except UnicodeError, msg: + LOG.exception("Can't decode name of pid %s" % pid) + else: + name = None + + KERNEL.CloseHandle(hProcess) + return {'name' : name} + + +# Initialize global methods +_enumProcesses = None +for impl in (WinEnumProcesses(), LinuxProcReader(), PsParser()): + if impl.supported(): + impl.log() + _enumProcesses = impl + break + +if _enumProcesses is None: + LOG.error("System %s is not supported" % sys.platform) + _enumProcesses = Unsupported() + +getPids = _enumProcesses.getPids +getPidNames = _enumProcesses.getPidNames +getPidDetails = _enumProcesses.getPidDetails +supportedOS = _enumProcesses.supported() + +def test_suite(): + import unittest + from doctest import DocTestSuite + return unittest.TestSuite(( + DocTestSuite() + )) + +if __name__ == '__main__': + import unittest + logging.basicConfig() + unittest.main(defaultTest="test_suite") + print getPids() + print getPidNames() Added: enumprocess/trunk/src/enumprocess.egg-info/PKG-INFO =================================================================== --- enumprocess/trunk/src/enumprocess.egg-info/PKG-INFO (rev 0) +++ enumprocess/trunk/src/enumprocess.egg-info/PKG-INFO 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,24 @@ +Metadata-Version: 1.0 +Name: enumprocess +Version: 0.1 +Summary: enumprocess: list pids and processes +Home-page: http://sourceforge.net/projects/pymoul/ +Author: Christian Heimes +Author-email: chr...@ch... +License: BSD License +Download-URL: http://cheeseshop.python.org/pypi/enumprocess +Description: + TODO: long description + +Keywords: pid,process,enum,proc,ps,psapi +Platform: Independant +Classifier: Development Status :: 4 - Beta +Classifier: Intended Audience :: Developers +Classifier: License :: OSI Approved :: BSD License +Classifier: Natural Language :: English +Classifier: Operating System :: Microsoft :: Windows +Classifier: Operating System :: POSIX +Classifier: Operating System :: POSIX :: Linux +Classifier: Programming Language :: Python +Classifier: Topic :: System +Classifier: Topic :: Software Development :: Libraries Added: enumprocess/trunk/src/enumprocess.egg-info/SOURCES.txt =================================================================== --- enumprocess/trunk/src/enumprocess.egg-info/SOURCES.txt (rev 0) +++ enumprocess/trunk/src/enumprocess.egg-info/SOURCES.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1,16 @@ +CHANGES.txt +INSTALL.txt +LICENSE.txt +MANIFEST.in +Makefile +README.txt +ez_setup.py +setup.py +version.txt +src/enumprocess/__init__.py +src/enumprocess/processinfo.py +src/enumprocess.egg-info/PKG-INFO +src/enumprocess.egg-info/SOURCES.txt +src/enumprocess.egg-info/dependency_links.txt +src/enumprocess.egg-info/top_level.txt +src/enumprocess.egg-info/zip-safe Added: enumprocess/trunk/src/enumprocess.egg-info/dependency_links.txt =================================================================== --- enumprocess/trunk/src/enumprocess.egg-info/dependency_links.txt (rev 0) +++ enumprocess/trunk/src/enumprocess.egg-info/dependency_links.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1 @@ + Added: enumprocess/trunk/src/enumprocess.egg-info/top_level.txt =================================================================== --- enumprocess/trunk/src/enumprocess.egg-info/top_level.txt (rev 0) +++ enumprocess/trunk/src/enumprocess.egg-info/top_level.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1 @@ +enumprocess Added: enumprocess/trunk/src/enumprocess.egg-info/zip-safe =================================================================== --- enumprocess/trunk/src/enumprocess.egg-info/zip-safe (rev 0) +++ enumprocess/trunk/src/enumprocess.egg-info/zip-safe 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1 @@ + Added: enumprocess/trunk/version.txt =================================================================== --- enumprocess/trunk/version.txt (rev 0) +++ enumprocess/trunk/version.txt 2007-02-27 14:29:15 UTC (rev 202) @@ -0,0 +1 @@ +0.1 \ No newline at end of file This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site. |