U (cG@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlmZddlmZddlZddlm Z ddlm!Z!dd lm"Z"dd lm#Z#dd lm$Z$dd lm%Z%dd lm&Z&ddl'm(Z(ddl'm)Z)ddl'm*Z*ddl'm+Z+ddl,m-Z-ddl,m.Z.ddl,m/Z/ddl,m0Z0ddl,m1Z1ddl,m2Z2ddl,m3Z3ddl,m4Z4zddlm5Z5Wn<e6k r4e7e8dddl5Z5W5QRXYnXej9dkrLddl:Z:ndZ:e$rbddl;md?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcddgGZ=deej>kZ?de j@kZAdfe j@kpdge j@kZBeAp&eBZCejDdhkZEe)didjZFe#rZeFdkkZGeFdlkZHndmZGdmZHdnZIdoZJdpZKdqZLeCreIdr9ZIeLdr9ZLeJds9ZJeKdr9ZKe jMdtkrdue NZOn dve NZOe2dwZPe-rdxQdydzZRnd{ZReSTd|kZUe jVWe jVXe jVYeZd}d}Z[e jVXe[d~Z\e jVWe jVYeZZ]e$o:e% Z^e_ej`dZae_edZbe_edZce_ej`dZde_ej`dZee_ej`dZfe_edZge_ej`dZhe_ej`dZie_ej`dZje_edZkzekoelemZnWneok rdmZnYnXe_edZpe_edZqe_ej`dZre#se o e sdkZtddZueuZvewe jxdZyezeyj{dde|eDZ}e~edeZeZeZGdddejZddZeddd@ZeddBZddAZedd=ZeddZejeLfdd>Zddd?ZddUZddTZGdddeZeejdeLddddWZee/efdeLdddddXZeedeLddddVZddOZddZejddNZdddPZdddSZGdddejZee_GddHdHeZee?dGddGdGeZddKZddZGddIdIZGddJdJZddZeIfddFZdddDZdddEZdddZZeedfdd[Zejfdd\Zddd]Zdd^Zejdd_ZddYZddɄZdd`ZddaZddbZdddZe$rejdddcZnejdddcZejzddфZe$reejddӄdS)z Test utilities. )print_functionN)AF_INET)AF_INET6) SOCK_STREAM)AIX)FREEBSD)LINUX)MACOS)POSIX)SUNOS)WINDOWS) bytes2human)memoize) print_color) supports_ipv6)PY3)FileExistsError)FileNotFoundError)range)super)u)unicode)which)mockignore)wait_pidAPPVEYORDEVNULLGLOBAL_TIMEOUTTOLERANCE_SYS_MEM NO_RETRIESPYPY PYTHON_EXEROOT_DIR SCRIPTS_DIR TESTFN_PREFIXUNICODE_SUFFIXINVALID_UNICODE_SUFFIX CI_TESTINGVALID_PROC_STATUSESTOLERANCE_DISK_USAGEIS_64BITHAS_CPU_AFFINITY HAS_CPU_FREQ HAS_ENVIRONHAS_PROC_IO_COUNTERS HAS_IONICEHAS_MEMORY_MAPSHAS_PROC_CPU_NUM HAS_RLIMITHAS_SENSORS_BATTERY HAS_BATTERYHAS_SENSORS_FANSHAS_SENSORS_TEMPERATURESZHAS_MEMORY_FULL_INFO MACOS_11PLUS MACOS_12PLUSpyrun terminate reap_childrenspawn_testproc spawn_zombiespawn_children_pairZThreadTaskunittestskip_on_access_deniedskip_on_not_implementedretry_on_failureTestMemoryLeakPsutilTestCaseprocess_namespacesystem_namespace print_sysinfoZ install_pipZinstall_test_depschdir safe_rmpath create_exeZ decode_pathZ encode_path get_testfn get_winverkernel_version call_until wait_for_pid wait_for_filecheck_net_address get_free_port bind_socketbind_unix_sockettcp_socketpairunix_socketpaircreate_sockets reload_moduleimport_module_by_pathwarncopyload_shared_lib is_namedtupleZ__pypy__GITHUB_ACTIONSZ CIBUILDWHEELlcCsntd}ttt|ddd}|dkrjtjtj dddgdd id d }ttt|ddd}|S) Nr.) z-sS-cz-import platform; print(platform.mac_ver()[0])ZSYSTEM_VERSION_COMPAT0T)envuniversal_newlines) platformmac_vertuplemapintsplit subprocess check_outputsys executable) version_strversionru9/tmp/pip-unpacked-wheel-l83rk7k5/psutil/tests/__init__.py macos_versions  rw)rc) rFrciPirrjavaz $psutil-%s-z @psutil-%s-u-ƒőősfutf8surrogateescapeufÀ€)asciizus-asciiz..scripts cpu_affinitycpu_freq getloadavgenvironionice memory_mapsnet_io_counterscpu_num io_countersrlimitsensors_battery sensors_fanssensors_temperaturesthreadscCsdd}tr@tr$trtdStdStr6tjtj StdSnt r|tj p|tjtj p|tdtj ddp|t }|std|Stjtj }tj|st||SdS) NcSs<ztj|dgtjtjdWntk r2YdSX|SdS)Nz-V)stdoutstderr)ro check_callPIPE Exception)exerururvattempts z_get_py_exe..attemptZpypy3pypypythonz python%s.%srbz"can't find python exe real abspath)r`r$rrrospathrealpathrqrrr version_infopsutilProcessr ValueErrorexistsAssertionError)rrrururv _get_py_exes*   rzr+cCs g|]}|drtt|qS)ZSTATUS_ startswithgetattrr.0xrururv s rAF_UNIXcsPeZdZdZfddZddZddZdd Zd d Zd d Z ddZ Z S) ThreadTaskz6A thread task which does nothing expect staying alive.cs$td|_d|_t|_dS)NFMbP?)r__init___running _interval threadingEvent_flagself __class__rurvrs zThreadTask.__init__cCs|jj}d||jt|fS)Nz<%s running=%s at %#x>)r__name__rid)rnamerururv__repr__szThreadTask.__repr__cCs ||SN)startrrururv __enter__#szThreadTask.__enter__cOs |dSr)stop)rargskwargsrururv__exit__'szThreadTask.__exit__cCs(|jrtdtj||jdS)zStart thread and keep it running until an explicit stop() request. Polls for shutdown every 'timeout' seconds. zalready startedN)rrrThreadrrwaitrrururvr*s zThreadTask.startcCs(d|_|j|jr$t|jqdS)NT)rrsettimesleeprrrururvrun3s zThreadTask.runcCs |jstdd|_|dS)z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)rrjoinrrururvr9szThreadTask.stop) r __module__ __qualname____doc__rrrrrrr __classcell__rururrvrs  rcstfdd}|S)Ncs.z ||WStk r(tYnXdSr)rr?rrfunrurvwrapperGs  z&_reap_children_on_err..wrapper functoolswrapsrrrurrv_reap_children_on_errFsrcKs|dt|dt|dt|dtjtrJd}|d||dkrt}zDt|d|}td |g}t j |f|}t |t |d d d W5t|Xn"t j |f|}t |t|j|S) a~Creates a python subprocess which does nothing for 60 secs and return it as a subprocess.Popen instance. If "cmd" is specified that is used instead of python. By default stdin and stdout are redirected to /dev/null. It also attempts to make sure the process is in a reasonably initialized state. The process is registered for cleanup on reap_children(). stdinrcwdrg creationflagsNz:from time import sleep;open(r'%s', 'w').close();sleep(60);reTdeleteempty) setdefaultr rgetcwdrr rNrLr%roPopen_subprocesses_startedaddrSrRpid)cmdkwdsZCREATE_NO_WINDOWtestfnZpylinesprocrururvr@Qs.        cCsd}ttd}zztdtj|tf}t rDt |dd\}}n t |\}}t |j }tt|ddd}t|t |}||fWSt||dk rt|XdS) aCreate a subprocess which creates another one as in: A (us) -> B (child) -> C (grandchild). Return a (child, grandchild) tuple. The 2 processes are fully initialized and will live for 60 secs and are registered for cleanup on reap_children(). Ndira; import subprocess, os, sys, time s = "import os, time;" s += "f = open('%s', 'w');" s += "f.write(str(os.getpid()));" s += "f.close();" s += "time.sleep(60);" p = subprocess.Popen([r'%s', '-c', s]) p.wait() r)rTFr)rNrrrLtextwrapdedentrbasenamer%r r=rrrrmrS _pids_startedr)tfilerssubpchildZgrandchild_pidZ grandchildrururvrBys"       cstjs tt}td|}d}t|}z| t t |\}}| \}}zZt |gggt t|d}t|t|tfddd|fWWS|XW5|t||dk rt|XdS)zCreate a zombie process and return a (parent, zombie) process tuple. In order to kill the zombie parent must be terminate()d first, then zombie must be wait()ed on. a import os, sys, time, socket, contextlib child_pid = os.fork() if child_pid > 0: time.sleep(3000) else: # this is the zombie process s = socket.socket(socket.AF_UNIX) with contextlib.closing(s): s.connect('%s') if sys.version_info < (3, ): pid = str(os.getpid()) else: pid = bytes(str(os.getpid()), 'ascii') s.sendall(pid) NcsSr)statusruzombierurvzspawn_zombie..zret == psutil.STATUS_ZOMBIE)rr rrNrrrWcloserL settimeoutr!r=acceptselectfilenormrecvrrrrQ)Z unix_filesrcrsockparentconn_ZzpidrurrvrAs.      c Ks|dd|ddt}zFt|d}||W5QRXtt|jgf|}t|j||fWSt k rt |YnXdS)zRun python 'src' code string in a separate interpreter. Returns a subprocess.Popen instance and the test file where the source code was written. rNrwt) rrNopenwriter@r%rrRrrrL)rrsrcfilefrrururvr=s     cKstrdnd}|dtj|dtj|dd|d|t|trTt|}tj|f|}t |t r|j t d\}}n | \}}|jdkrt||rt||d r|d d }|S) zUrun cmd in a subprocess and return its output. raises RuntimeError on error. rrrrrhTrtimeout N)r rror isinstancestrshlexrnrrrr communicater! returncode RuntimeErrorr]endswith)rrflagsprrrururvshs&          rc sddddfdd}fddfd d }d d }|}zbt|trb|||WLSt|tjtjfr||W,St|tjr|||WSt d |W5t|tjtjfr||t|tr|n|j}t|rt|XdS)aTerminate a process and wait() for it. Process can be a PID or an instance of psutil.Process(), subprocess.Popen() or psutil.Popen(). If it's a subprocess.Popen() or psutil.Popen() instance also closes its stdin / stdout / stderr fds. PID is wait()ed even if the process is already gone (kills zombies). Does nothing if the process does not exist. Return process exit status. cSsdt|tjrts|n ||tr`t|tjr`zt|j|WStj k r^YnXdSr) r rorrrr rrr NoSuchProcessprocrrururvrs  zterminate..waitcSs6trtrtj}tr(|tjkr(|tj||dSr)r r`signalSIGKILLr send_signalSIGCONT)rsigrururvsendsigs  zterminate..sendsigc sZz|Wn@tk rN}z"tr0|jdkr0n|jtjkr>W5d}~XYnX||S)N)OSErrorr winerrorerrnoZESRCH)rrerrrrrrurvterm_subprocess_procs z'terminate..term_subprocess_proccs0z|Wntjk r$YnX||Sr)rrrr!rurvterm_psutil_proc's z#terminate..term_psutil_proccsFzt|}Wn(tjk r6tr2t||YSYn X||SdSr)rrrr r)rrr)r#rurvterm_pid.s zterminate..term_pidcSs4|jr|j|jr |j|jr0|jdSr)rrrr)rrururv flush_popen8s   zterminate..flush_popenz wrong type %rN) r rorrrmr pid_existsrr TypeError)Z proc_or_pidr wait_timeoutr"r$r%rrru)rrr#rrvr>s&        cCstj|d}tr&t}t|qtrr wait_procsr!r]rr)r*r+rrrgonealiverururvr?Rs    cCsts tdd}td}|D]"}|s4|dkr>||7}q qDq |sTtd|d}d}|d}t|d}t|dkrt|d}t|dkrt|d}|||fS) z"Return a tuple such as (2, 6, 36).z not POSIXrbrazcan't parse %rrr) r NotImplementedErrorrunameisdigitrrnrmlen)rr3cminormicronumsmajorrururvrPws&         cCsdts tdt}t|dr*|jp&d}n(td|d}|rNt| d}nd}|d|d|fS)Nz not WINDOWSservice_pack_majorrz\s\d$rr1) r r2rqgetwindowsversionhasattrr;researchrmgroup)ZwvsprrururvrOs  c@s<eZdZdZeddddfddZddZdd Zd d ZdS) retryzA retry decorator.NrcCs2|r|rtd||_||_||_||_||_dS)Nz/timeout and retries args are mutually exclusive)r exceptionrretriesintervallogfun)rrDrrErFrGrururvrszretry.__init__ccsT|jr*t|j}t|krPdVqn&|jrHt|jD] }dVq:ndVqHdSr)rrrEr)rstop_atrrururv__iter__s   zretry.__iter__cCs|jdk rt|jdSr)rFrrrrururvrs z retry.sleepcs"tfdd}|_|S)Nc s|d}D]b}z||WSjk rh}z,|}jdk rH|WYqW5d}~XYqXqtrv|ndSr)rDrGrr)rrexcrrrrurvrs  zretry.__call__..wrapper)rr decorator)rrrrurKrv__call__szretry.__call__) rrrrrrrIrrMrurururvrCs  rCr)rDrGrrFcCst|trtddS)zWait for pid to show up in the process list then return. Used in the test suite to give time the sub process to initialize. {Gz?N)rrr rr)rrururvrRs Tc Cs:t|d}|}W5QRX|s*|s*t|r6t||S)z8Wait for a file to be written on disk with some content.rb)rreadrrL)fnamerrrdatarururvrSs cCs|}t|st|S)zVKeep calling function for timeout secs and exit if eval() expression is True. )evalr)rexprretrururvrQs cCsndd}zLt|}t|jr0ttj|}nttj|}t rJ|n||Wnt k rhYnXdS)z?Convenience function for removing temporary test files or dirs.c Ss~tt}t|krvz|WStk r4Yn6tk rh}z|}tdt|W5d}~XYnXtdq |dS)Nz ignoring %srN)rr!rZ WindowsErrorr]r r)rrHrr rururv retry_funs  " zsafe_rmpath..retry_funN) rstatS_ISDIRst_moderpartialshutilrmtreeremover r)rrVstrrururvrLs   cCs(zt|Wntk r"YnXdS)z.Convenience function for creating a directory.N)rmkdirrrrururv safe_mkdir'sr`c cs.t}zt|dVW5t|XdS)z@Context manager which temporarily changes the current directory.N)rrrK)dirnamecurdirrururvrK/s   c Cstj|rt||rtds(tdt|tr int main() { pause(); return 1; } z.csuffixrz-oN)rrrrrrr boolrrr rrNrrLrrorr[copyfiler%r rWchmodrYS_IEXEC)ZoutpathZc_coderr^rururvrM:s$   r0cCs.tjt||d}tj|stj|SqdS)zReturn an absolute pathname of a file or dir that did not exist at the time this call is made. Also schedule it for safe deletion at interpreter exit. It's technically racy but probably not really due to the time variant. )prefixrerN)tempfilemktempr(rrrr)rerrrururvrNXs c@sBeZdZddZeejds$ejjZe s>ddZ e j ddZ dS) TestCasecCs.|jj}|dsd|}d||jj|jfS)Nzpsutil.z psutil.tests.z%s.%s.%s)rrrrZ_testMethodName)rZfqmodrururv__str__ms zTestCase.__str__assertRaisesRegexcCsdSrrurrururvrunTest{szTestCase.runTestcos dVdSrru)rrkwrururvsubTest~szTestCase.subTestN)rrrrnr=unittestrmZassertRaisesRegexprorrp contextlibcontextmanagerrrrurururvrmis rmc@sBeZdZdZdddZddZdd Zd d Zd d ZddZ dS)rGzUTest class providing auto-cleanup wrappers on top of process test utilities. r0NcCst||d}|t||S)N)rer)rN addCleanuprL)rrerrQrururvrNs  zPsutilTestCase.get_testfncOst||}|t||Sr)r@rvr>)rrrrrururvr@s  zPsutilTestCase.spawn_testproccCs*t\}}|t||t|||fSr)rBrvr>)rZchild1Zchild2rururvrBs   z"PsutilTestCase.spawn_children_paircCs*t\}}|t||t|||fSr)rArvr>)rrrrururvrAs   zPsutilTestCase.spawn_zombiecOs*t||\}}|t||t||Sr)r=rvrLr>)rrrrrrururvr=s  zPsutilTestCase.pyruncCs|tjtj|jt|tjtjfrn|r2tz | }Wntjk rTYnXtd||j ddt |jrt|j| |jt dS)Nz3Process.status() didn't raise exception (status=%s)rr) assertRaisesrrrrr r is_runningrrrr&Z assertNotInpids)rrrrururvassertProcessGones   z PsutilTestCase.assertProcessGone)r0N) rrrrrNr@rBrAr=rzrurururvrGs zunreliable on PYPYc@seZdZdZdZdZdZer dndZdZ e Z e eddZedd Zed d Zd d ZddZddZddZddZddZddZdddZddZdS) rFaTest framework class for detecting function memory leaks, typically functions implemented in C which forgot to free() memory from the heap. It does so by checking whether the process memory usage increased before and after calling the function many times. Note that this is hard (probably impossible) to do reliably, due to how the OS handles memory, the GC and so on (memory can even decrease!). In order to avoid false positives, in case of failure (mem > 0) we retry the test for up to 5 times, increasing call repetitions each time. If the memory keeps increasing then it's a failure. If available (Linux, OSX, Windows), USS memory is used for comparison, since it's supposed to be more precise, see: https://gmpy.dev/blog/2016/real-process-memory-and-environ-in-python If not, RSS memory is used. mallinfo() on Linux and _heapwalk() on Windows may give even more precision, but at the moment are not implemented. PyPy appears to be completely unstable for this framework, probably because of its JIT, so tests on PYPY are skipped. Usage: class TestLeaks(psutil.tests.TestMemoryLeak): def test_fun(self): self.execute(some_function) rcrrzTZ PSUTIL_DEBUGcCstddS)NF)r _set_debugclsrururv setUpClassszTestMemoryLeak.setUpClasscCst|jdSr)rr|_psutil_debug_origr}rururv tearDownClassszTestMemoryLeak.tearDownClasscCs|j}t|d|jS)NZuss) _thisprocmemory_full_inforZrss)rmemrururv_get_mems zTestMemoryLeak._get_memcCstr|jS|jSdSr)r rnum_fds num_handlesrrururv _get_num_fdss zTestMemoryLeak._get_num_fdscCs|jrt|dtjddS)NZyellow)colorfile)verboserrqr)rmsgrururv_logszTestMemoryLeak._logcCsx|}|||}||}|dkr8|d||dkrttrHdnd}|dkr\|d7}d|||f}||dS) zMakes sure num_fds() (POSIX) or num_handles() (Windows) does not increase after calling a function. Used to discover forgotten close(2) and CloseHandle syscalls. rzHnegative diff %r (gc probably collected a resource from a previous test)fdhandler1rz%s unclosed %s after calling %rN)rcallfailr )rrbeforeafterdifftype_rrururv _check_fdss  zTestMemoryLeak._check_fdscCs^tjdd|}t|D]}||}~~qtjdd|}|tjg||}|S)zGet 2 distinct memory samples, before and after having called fun repeatedly, and return the memory difference. r1)Z generation)gcZcollectrrr assertEqualgarbage)rrtimesZmem1rrUZmem2rrururv _call_ntimes s    zTestMemoryLeak._call_ntimesc Csg}d}|}td|dD]} |||} d| t| t| ||f} || | |kp^| |k} | r|| dkrv|| dS| dkrt|| ||7}| }q|d|dS)Nrr1z,Run #%s: extra-mem=%s, per-call=%s, calls=%sz. )rrr appendrprintrr) rrr warmup_timesrE tolerancemessagesZprev_memZincreaseidxrrsuccessrururv _check_mems.     zTestMemoryLeak._check_memcCs|Srru)rrrururvr3szTestMemoryLeak.callNc Cs|dk r |n|j}|dk r|n|j}|dk r0|n|j}|dk rB|n|j}zD|dksZtd|dksjtd|dksztd|dkstdWn.tk r}ztt|W5d}~XYnX||||||j |||||ddS) zTest a callable.Nr1ztimes must be >= 1rzwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)rrrEr) rrrErrrr rrr)rrrrrErr rururvexecute6s&    zTestMemoryLeak.executec s"fdd}j|f|dS)znConvenience method to test a callable while making sure it raises an exception on every call. csdSr)rwrurJrrrurvrOsz*TestMemoryLeak.execute_w_exc..callN)r)rrJrrrrurrv execute_w_excKszTestMemoryLeak.execute_w_exc)NNNN)rrrrrrrr+rErrrrrfrgetenvr classmethodrrrrrrrrrrrrurururvrFs.    cCsddl}ddl}ddl}ddl}ddl}ddl}z ddl}Wntk rTd}YnXz ddl}Wntk rzd}YnX| }t j rt drt d|d<n~t jrd|d|d<nbt jrddtt||d<t|dr|dd |7<nd ||f|d<d t||g|d <t jrV|d |d <d ||| g|d<t!|dd|d<|dk r|dd|j"7<t jrt drt ddg} t| #dd|d<nd|d<|$d} | r| |d<t%&|d<|'} d| d| df|d<|j(t )*d|d<|j+*d|d<|,|d<t-j./d|d <t-0|d!<t1|d"<|2|d#<t-3|d$<t 4|d%<d&t5d'd(t 6D|d)<t 7} d*t8| j9t:| j;t:| j<f|d+<t =} d*t8| j9t:| j;t:| j<f|d,<t>t ?|d-<t @A}|Bd.d|C||d/<tDd0t%jEd1|FD]$\}}tDd2|d3|ft%jEd1qntDd0t%jEd1t%jGHdS)4Nr lsb_releasezlsb_release -d -sZOSz Darwin %szWindows  win32_editionz, z%s %sarchrbkernelr __version__z not installedpipz (wheel=%s)rcz --versionrr1glibcz fs-encodingz%s, %slangz%Y-%m-%d %H:%M:%Sz boot-timeruser~homerZpyexehostnameZPIDZcpusz%.1f%%, %.1f%%, %.1f%%cSsg|]}|tdqS)d)r cpu_countrrururvrsz!print_sysinfo..Zloadavgz%s%%, used=%s, total=%sZmemoryswapryrrzF======================================================================rz%-17s %s:)I collectionsdatetimegetpasslocaleripprintr ImportErrorwheel OrderedDictrrrrZOSXrjr rrlr  win32_verr=rsystemrtlist architecturemachiner r3python_implementationpython_versionpython_compilerrrrnlibc_verrqgetfilesystemencoding getlocale fromtimestamp boot_timestrftimenowgetuserrr expanduserrr%nodegetpidrrkrvirtual_memoryrmpercentr usedtotal swap_memoryr5ryras_dictr,pformatrritemsrflush)rrrrrirrrinfooutrrrrZpinfokvrururvrJUs                     cCs6t}t|dr|St|dr2t|SdS)Nrrr)rrr=rrandomchoicer)rrururv_get_eligible_cpus   rc@sBeZdZdZddifddifgZddifddddifd difd difd difd difd difddifddifg Zddifddddifddifddifddifddifddifddifddifddifddifddifddifd difd!difd"difd#difgZerFed$difg7Zed%difg7Zed&difg7Zed'difg7Zer\ed(difg7Ze rred)difg7Ze red*e j fifg7Ze red+difg7Zered,difg7Zered-difg7Zered.difg7Zered/dd0d1ifg7ZgZereddifg7Znede jfifg7Ze rJed*e j d2fifg7Ze rerned)e jd3fifg7Zned)e jfifg7Ze red+egfifg7Zd4ejfifd5difd6difd7difd8difgZered4ejfifg7Zed4ejfifg7ZeeeeZd9d:ZdDd;d<Zd=d>Z e!d?d@Z"e!dAdBZ#dCS)ErHaA container that lists all Process class method names + some reasonable parameters to be called with. Utility methods (parent(), children(), ...) are excluded. >>> ns = process_namespace(psutil.Process()) >>> for fun, name in ns.iter(ns.getters): ... fun() cpu_percentruZmemory_percentrr+r*TrxZmemory_info_exZoneshotrparentsrrrZcmdline connectionskindall cpu_timesZ create_timerrrZ memory_inforniceZnum_ctx_switchesZ num_threadsZ open_filesZppidrrusernameZuidsZgidsZterminalrrrrrrrrrZgroupedF)rirrZsuspendresumer>killcCs ||_dSr)_proc)rrrururvrszprocess_namespace.__init__ccsZt|}t||D]>\}}}|r,|t|j|}tj|f||}||fVqdSz_Given a list of tuples yields a set of (fun, fun_name) tuples in random order. N)rrshuffle clear_cacherrrrZ)rlsrfun_namerrrrururviters  zprocess_namespace.itercCs|jj|jjdddS)z&Clear the cache of a Process instance.T)Z _ignore_nspN)r_initrrrururvr(szprocess_namespace.clear_cachecCs>|D]4\}}}d|}t||sd|jj|f}t|qdS)z}Given a TestCase instance and a list of tuples checks that the class defines the required test method names. Ztest_z$%r class should define a '%s' methodN)r=rrAttributeError)r~Z test_classrrr meth_namerrururvtest_class_coverage,s z%process_namespace.test_class_coveragecCs`tdd|jD}tdd|jD}tddttjD}||B|A}|r\td|dS)NcSsg|] }|dqSrrurrururvr:sz*process_namespace.test..cSsg|] }|dqSrrurrururvr;scSsg|]}|ddkr|qS)rrrurrururvr<s z!uncovered Process class names: %r)rrignoredrrrr)r~thisrklassZleftoutrururvtest8s  zprocess_namespace.testN)T)$rrrrutilsrgettersr r2r3r6rZ RLIMIT_NOFILEr/r5r1r r4ZsettersZNORMAL_PRIORITY_CLASSrZIOPRIO_CLASS_NONEZ IOPRIO_NORMALrrSIGTERMZkillersZ CTRL_C_EVENTZCTRL_BREAK_EVENTrrrrrrrrurururvrHs       c@seZdZdZddifddddifddddifddifd dd difd dd difd dd difd dddifdefifddddifddifddifddddifdefifddifddifddifddifgZereddd difg7Ze reddifg7Ze reddifg7Ze r$eddifg7Ze r:eddifg7Ze r`ed difg7Zed!d"ifg7Zd#difd$egfifd%difd&difgZeZed'd(ZejZd)S)*rIzA container that lists all the module-level, system-related APIs. Utilities such as cpu_percent() are excluded. Usage: >>> ns = system_namespace >>> for fun, name in ns.iter(ns.getters): ... fun() rrurlogicalFTZ cpu_statsrZpercpuZdisk_io_countersZperdiskZdisk_partitionsr disk_usageZnet_connectionsrZ net_if_addrsZ net_if_statsrZpernicr&ryrZusersrrrrrrZwin_service_iterZwin_service_get)ZalgZ process_iterr-rZcpu_times_percentccsLt|}t||D]0\}}}tt|}tj|f||}||fVqdSr)rrrrrrrZ)rrrrrrururvrus   zsystem_namespace.iterN)rrrrrrrrr0HAS_GETLOADAVGr:r9r7r rrrr staticmethodrrHrrurururvrIBsT         cCst|st|d|_|S)zA decorator to mark a TestCase class. When running parallel tests, class' unit tests will be run serially (1 process). T)inspectisclassrZ _serialrun)rrururv serialrunsr cCsdd}ttd||dS)zZDecorator which runs a test function and retries N times before actually failing. cSstd|tjddS)Nz %r, retryingr)rrqr)rJrururvrGsz retry_on_failure..logfunN)rDrrErG)rCr)rErGrururvrEscsfdd}|S)z,Decorator to Ignore AccessDenied exceptions.cstfdd}|S)Ncs@z ||WStjk r:dk r,s,tdYnXdS)Nzraises AccessDenied)rZ AccessDeniedrsSkipTestrronly_ifrurvrs z9skip_on_access_denied..decorator..wrapperrrr rrvrLsz(skip_on_access_denied..decoratorrur rLrur rvrCs csfdd}|S)z3Decorator to Ignore NotImplementedError exceptions.cstfdd}|S)NcsHz ||WStk rBdk r*s*dj}t|YnXdS)Nz4%r was skipped because it raised NotImplementedError)r2rrsr )rrrr rurvrs z;skip_on_not_implemented..decorator..wrapperrrr rrvrLs z*skip_on_not_implemented..decoratorrurrur rvrDs 127.0.0.1c Cs@tt(}||df|dW5QRSQRXdS)z6Return an unused TCP port. Subject to race conditions.rr1N)rtclosingsocketbind getsockname)hostrrururvrUscCs|dkr|ttfkrd}t||}z@tjdkrB|tjtjd|||tj kr`| d|WSt k r| YnXdS)zBinds a generic socket.Nr0r)ntcygwinr1rz) rrrrr setsockopt SOL_SOCKET SO_REUSEADDRrrlistenrr)familytypeaddrrrururvrVs     cCsrtjs ttj|rt|ttj|}z"|||tj krL| dWnt k rl| YnX|S)zBind a UNIX socket.rz) rr rrrrrrrrrrr)rrrrururvrWs   rc Cstt|t}|||d|}t|t}zJ|||}|\}}||krz||fWW5QRS| qPWnt k r| YnXW5QRXdS)z^Build a pair of TCP sockets connected to each other. Return a (server, client) tuple. rzN) rtrrrrrrconnectrrr)rrZllr6ZcaddrarururvrXs     cCstjs td}}z@t|tjd}|dttjtj}|d||Wn6t k r|dk rr| |dk r| YnX||fS)zBuild a pair of UNIX sockets connected to each other through the same UNIX file name. Return a (server, client) tuple. Nrr) rr rrWrr setblockingrrrr)rserverclientrururvrYs   c csg}d}}z|ttjtj|ttjtjtrd|ttj tj|ttj tjt rt rt }t }t |\}}t|tjd}|||fD]}||q|VW5|D] }|q||fD]}|dk rt|qXdS)z1Open as many socket families / types as possible.Nr!)rrLrrVrrr SOCK_DGRAMrrr HAS_CONNECTIONS_UNIXrNrYrW)socksZfname1Zfname2rrQs1s2Zs3rururvrZs*     cCsddl}tr(tr(ts(t|tjs(t||tjkrdd| dD}t |dksZt||D]"}d|krvdks^nt|q^tst |}| |nb|tj krt|tst|tst |}||n.|tjkrtd|dk st|n td |dS) z[Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. rNcSsg|] }t|qSru)rmrrururvr9sz%check_net_address..rarz([a-fA-F0-9]{2}[:|\-]?){6}zunknown family %r) ipaddressenumrr$r IntEnumrrrrnr5r IPv4Addressrr  IPv6AddressrZAF_LINKr>matchr)rrr+ZoctsnumrururvrT1s&       cCsTdd}dd}dd}dd}d d }||||||||||d S) z*Check validity of a connection namedtuple.cSst|dk}t|dks$tt||d|jks.check_ntuplec Ss|jtttfkst|jtdk r:t|jtjsNt|nt|jtsNt||jtkrt |j|j }t |Rz| |jddfWn2t jk r}z|jtjkrW5d}~XYnXW5QRXn |jtkr|jtjkst|jdS)Nr)rrrrrr,r r-rmrrrtrrr3errorrZ EADDRNOTAVAILrr CONN_NONE)rrr rururv check_familyYs     z-check_connection_ntuple..check_familycSsttdt}|jtjtj|fks,t|jtdk rLt|jtj s`t|nt|jt s`t||jtjkr|j t j kst|j dS)NSOCK_SEQPACKET)rrobjectrrr%rr,r r-rmrrr7)rr9rururv check_typens  z+check_connection_ntuple..check_typecSs|j|jfD]}|jttfkrt|ts4tt||s:q t|j t sTtt|j d|j krjdksvnt|j t |j |jq |jt kr t|ts tt|q dS)Nri)r3r4rrrr rkrrportrmrTiprr )rrrururv check_addrszs" z,check_connection_ntuple..check_addrscSst|jtst|jddttD}|j|ks.check_status..) r rr rrrrrrrrr7)rZvalidsrururv check_statuss z-check_connection_ntuple..check_statusNru)rr5r8r;r>r?rururvcheck_connection_ntupleKs    r@cCsPzddl}t|dstWn&tk r@ddl}||YSX||SdS)z,Backport of importlib.reload of Python 3.3+.rNreload) importlibr=rimprA)modulerBrCrururvr[s cCstjtj|d}tjddkr:ddl}|||Stjdddkrfddlm }||| Sddl }|j ||}|j |}|j||SdS)Nrrbr)SourceFileLoader)rrsplitextrrqrrCZ load_sourceZimportlib.machineryrE load_moduleimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module)rrrCrErBspecmodrururvr\s    cCst|tdS)zRaise a warning msg.N)warningsr] UserWarning)rrururvr]scCsVt|}|j}t|dks&|dtkr*dSt|dd}t|tsDdStdd|DS)z-Check if object is an instance of namedtuple.r1rF_fieldsNcss|]}t|tkVqdSr)rr )rnrururv sz is_namedtuple..)r __bases__r5rkrr r)rtbrrururvr_s  c#sttrdnddt|d}fddtD}t|}t||zt ||VW5t |XdS)zCtx manager which picks up a random shared CO lib used by this process, copies it in another location and loads it in memory via ctypes. Return the new absolutized path. rrz.sordcs6g|].}tj|jdkr|jkr|jqS)r1)rrrFlowerrrextrurvrs'copyload_shared_lib..N) r$rNrrrrrr[rgrLctypesCDLL)redstlibsrrurYrvr^s     c #sddlm}ddlm}dt|d}fddtD}trb|sbddtD}t |}t ||d }zt|}|VW5|d k rtj j j}|jg|_||j}|dkr|t|Xd S) zCtx manager which picks up a random shared DLL lib used by this process, copies it in another location and loads it in memory via ctypes. Return the new absolutized, normcased path. r)WinError)wintypesz.dllrdcsFg|]>}|jrdtj|jkrd|jkr|jqS)rZwow64)rrXrrrrrZrurvrs r[cSs(g|] }dtj|jkr|jqS)r)rrrrXrrururvrsN)r\r`rarNrrrr$rrr[rgwindllkernel32 FreeLibraryZHMODULEargtypes_handlerLZWinDLL) rer`rar^r_rcfilererUrurbrvr^s(         cCstdddS)NTr))r?rurururvcleanup_test_procssricCs t|Sr)rqexit)rframerururvrrr)N)F)TF)N)r0N)N)N)r)r)r0)r0)r __future__ratexitrtr\rrrrrrirr>rr r[rrrWrorqrkrrrrsrPrrrrrrrr r r r Zpsutil._commonr rrrZpsutil._compatrrrrrrrrrrcatch_warnings simplefilterrr,Zpsutil._psposixr__all__builtin_module_namesr$rrr`r+maxsizer.rwr;r<r#r"r-r!rrr(r)decoder*rrXZASCII_FSrrrra__file__r&r'ZHEREr&r=rr/r0rr1r3r4ZHAS_NET_IO_COUNTERSr5r2r6r7rfrr8rr9r:Z HAS_THREADSgetuidZ SKIP_SYSCONSrr%rdevnullr registerrrr,rr:rrrrrrrr@rBrAr=rrr>r?rPrOrCrrRrrSrQrLr`rurKrMrNrmrGZskipIfrFrJrrHrIr rErCrDrUrVrWrXrYrZrTr@r[r\r]r_r^rirurururvs                            .                            0  ' &-  U %:    !  1 [  B     R   ,