"""Functions for getting memory usage of Windows processes."""
__all__ = ['get_current_process', 'get_memory_info', 'get_memory_usage']importctypesfrom ctypes importwintypes
GetCurrentProcess=ctypes.windll.kernel32.GetCurrentProcess
GetCurrentProcess.argtypes=[]
GetCurrentProcess.restype=wintypes.HANDLE
SIZE_T=ctypes.c_size_tclassPROCESS_MEMORY_COUNTERS_EX(ctypes.Structure):
_fields_=[
('cb', wintypes.DWORD),
('PageFaultCount', wintypes.DWORD),
('PeakWorkingSetSize', SIZE_T),
('WorkingSetSize', SIZE_T),
('QuotaPeakPagedPoolUsage', SIZE_T),
('QuotaPagedPoolUsage', SIZE_T),
('QuotaPeakNonPagedPoolUsage', SIZE_T),
('QuotaNonPagedPoolUsage', SIZE_T),
('PagefileUsage', SIZE_T),
('PeakPagefileUsage', SIZE_T),
('PrivateUsage', SIZE_T),
]
GetProcessMemoryInfo=ctypes.windll.psapi.GetProcessMemoryInfo
GetProcessMemoryInfo.argtypes=[
wintypes.HANDLE,
ctypes.POINTER(PROCESS_MEMORY_COUNTERS_EX),
wintypes.DWORD,
]
GetProcessMemoryInfo.restype=wintypes.BOOLdefget_current_process():"""Return handle to current process."""
returnGetCurrentProcess()def get_memory_info(process=None):"""Return Win32 process memory counters structure as a dict."""
if process isNone:
process=get_current_process()
counters=PROCESS_MEMORY_COUNTERS_EX()
ret=GetProcessMemoryInfo(process, ctypes.byref(counters),
ctypes.sizeof(counters))if notret:raisectypes.WinError()
info=dict((name, getattr(counters, name))for name, _ incounters._fields_)returninfodef get_memory_usage(process=None):"""Return this process's memory usage in bytes."""info= get_memory_info(process=process)return info['PrivateUsage']if __name__ == '__main__':importpprint
pprint.pprint(get_memory_info())