How to get current CPU and RAM usage in Python?

前端 未结 16 1485
攒了一身酷
攒了一身酷 2020-11-22 04:09

What\'s your preferred way of getting current system status (current CPU, RAM, free disk space, etc.) in Python? Bonus points for *nix and Windows platforms.

There s

相关标签:
16条回答
  • 2020-11-22 05:06

    Below codes, without external libraries worked for me. I tested at Python 2.7.9

    CPU Usage

    import os
    
        CPU_Pct=str(round(float(os.popen('''grep 'cpu ' /proc/stat | awk '{usage=($2+$4)*100/($2+$4+$5)} END {print usage }' ''').readline()),2))
    
        #print results
        print("CPU Usage = " + CPU_Pct)
    

    And Ram Usage, Total, Used and Free

    import os
    mem=str(os.popen('free -t -m').readlines())
    """
    Get a whole line of memory output, it will be something like below
    ['             total       used       free     shared    buffers     cached\n', 
    'Mem:           925        591        334         14         30        355\n', 
    '-/+ buffers/cache:        205        719\n', 
    'Swap:           99          0         99\n', 
    'Total:        1025        591        434\n']
     So, we need total memory, usage and free memory.
     We should find the index of capital T which is unique at this string
    """
    T_ind=mem.index('T')
    """
    Than, we can recreate the string with this information. After T we have,
    "Total:        " which has 14 characters, so we can start from index of T +14
    and last 4 characters are also not necessary.
    We can create a new sub-string using this information
    """
    mem_G=mem[T_ind+14:-4]
    """
    The result will be like
    1025        603        422
    we need to find first index of the first space, and we can start our substring
    from from 0 to this index number, this will give us the string of total memory
    """
    S1_ind=mem_G.index(' ')
    mem_T=mem_G[0:S1_ind]
    """
    Similarly we will create a new sub-string, which will start at the second value. 
    The resulting string will be like
    603        422
    Again, we should find the index of first space and than the 
    take the Used Memory and Free memory.
    """
    mem_G1=mem_G[S1_ind+8:]
    S2_ind=mem_G1.index(' ')
    mem_U=mem_G1[0:S2_ind]
    
    mem_F=mem_G1[S2_ind+8:]
    print 'Summary = ' + mem_G
    print 'Total Memory = ' + mem_T +' MB'
    print 'Used Memory = ' + mem_U +' MB'
    print 'Free Memory = ' + mem_F +' MB'
    
    0 讨论(0)
  • 2020-11-22 05:11

    "... current system status (current CPU, RAM, free disk space, etc.)" And "*nix and Windows platforms" can be a difficult combination to achieve.

    The operating systems are fundamentally different in the way they manage these resources. Indeed, they differ in core concepts like defining what counts as system and what counts as application time.

    "Free disk space"? What counts as "disk space?" All partitions of all devices? What about foreign partitions in a multi-boot environment?

    I don't think there's a clear enough consensus between Windows and *nix that makes this possible. Indeed, there may not even be any consensus between the various operating systems called Windows. Is there a single Windows API that works for both XP and Vista?

    0 讨论(0)
  • Taken feedback from first response and done small changes

                #!/usr/bin/env python
                #Execute commond on windows machine to install psutil>>>>python -m pip install psutil
                import psutil
    
                print ('                                                                   ')
                print ('----------------------CPU Information summary----------------------')
                print ('                                                                   ')
    
                # gives a single float value
                vcc=psutil.cpu_count()
                print ('Total number of CPUs :',vcc)
    
                vcpu=psutil.cpu_percent()
                print ('Total CPUs utilized percentage :',vcpu,'%')
    
                print ('                                                                   ')
                print ('----------------------RAM Information summary----------------------')
                print ('                                                                   ')
                # you can convert that object to a dictionary 
                #print(dict(psutil.virtual_memory()._asdict()))
                # gives an object with many fields
                vvm=psutil.virtual_memory()
    
                x=dict(psutil.virtual_memory()._asdict())
    
                def forloop():
                    for i in x:
                        print (i,"--",x[i]/1024/1024/1024)#Output will be printed in GBs
    
                forloop()
                print ('                                                                   ')
                print ('----------------------RAM Utilization summary----------------------')
                print ('                                                                   ')
                # you can have the percentage of used RAM
                print('Percentage of used RAM :',psutil.virtual_memory().percent,'%')
                #79.2
                # you can calculate percentage of available memory
                print('Percentage of available RAM :',psutil.virtual_memory().available * 100 / psutil.virtual_memory().total,'%')
                #20.8
    
    0 讨论(0)
  • 2020-11-22 05:13

    Here's something I put together a while ago, it's windows only but may help you get part of what you need done.

    Derived from: "for sys available mem" http://msdn2.microsoft.com/en-us/library/aa455130.aspx

    "individual process information and python script examples" http://www.microsoft.com/technet/scriptcenter/scripts/default.mspx?mfr=true

    NOTE: the WMI interface/process is also available for performing similar tasks I'm not using it here because the current method covers my needs, but if someday it's needed to extend or improve this, then may want to investigate the WMI tools a vailable.

    WMI for python:

    http://tgolden.sc.sabren.com/python/wmi.html

    The code:

    '''
    Monitor window processes
    
    derived from:
    >for sys available mem
    http://msdn2.microsoft.com/en-us/library/aa455130.aspx
    
    > individual process information and python script examples
    http://www.microsoft.com/technet/scriptcenter/scripts/default.mspx?mfr=true
    
    NOTE: the WMI interface/process is also available for performing similar tasks
            I'm not using it here because the current method covers my needs, but if someday it's needed
            to extend or improve this module, then may want to investigate the WMI tools available.
            WMI for python:
            http://tgolden.sc.sabren.com/python/wmi.html
    '''
    
    __revision__ = 3
    
    import win32com.client
    from ctypes import *
    from ctypes.wintypes import *
    import pythoncom
    import pywintypes
    import datetime
    
    
    class MEMORYSTATUS(Structure):
        _fields_ = [
                    ('dwLength', DWORD),
                    ('dwMemoryLoad', DWORD),
                    ('dwTotalPhys', DWORD),
                    ('dwAvailPhys', DWORD),
                    ('dwTotalPageFile', DWORD),
                    ('dwAvailPageFile', DWORD),
                    ('dwTotalVirtual', DWORD),
                    ('dwAvailVirtual', DWORD),
                    ]
    
    
    def winmem():
        x = MEMORYSTATUS() # create the structure
        windll.kernel32.GlobalMemoryStatus(byref(x)) # from cytypes.wintypes
        return x    
    
    
    class process_stats:
        '''process_stats is able to provide counters of (all?) the items available in perfmon.
        Refer to the self.supported_types keys for the currently supported 'Performance Objects'
    
        To add logging support for other data you can derive the necessary data from perfmon:
        ---------
        perfmon can be run from windows 'run' menu by entering 'perfmon' and enter.
        Clicking on the '+' will open the 'add counters' menu,
        From the 'Add Counters' dialog, the 'Performance object' is the self.support_types key.
        --> Where spaces are removed and symbols are entered as text (Ex. # == Number, % == Percent)
        For the items you wish to log add the proper attribute name in the list in the self.supported_types dictionary,
        keyed by the 'Performance Object' name as mentioned above.
        ---------
    
        NOTE: The 'NETFramework_NETCLRMemory' key does not seem to log dotnet 2.0 properly.
    
        Initially the python implementation was derived from:
        http://www.microsoft.com/technet/scriptcenter/scripts/default.mspx?mfr=true
        '''
        def __init__(self,process_name_list=[],perf_object_list=[],filter_list=[]):
            '''process_names_list == the list of all processes to log (if empty log all)
            perf_object_list == list of process counters to log
            filter_list == list of text to filter
            print_results == boolean, output to stdout
            '''
            pythoncom.CoInitialize() # Needed when run by the same process in a thread
    
            self.process_name_list = process_name_list
            self.perf_object_list = perf_object_list
            self.filter_list = filter_list
    
            self.win32_perf_base = 'Win32_PerfFormattedData_'
    
            # Define new datatypes here!
            self.supported_types = {
                                        'NETFramework_NETCLRMemory':    [
                                                                            'Name',
                                                                            'NumberTotalCommittedBytes',
                                                                            'NumberTotalReservedBytes',
                                                                            'NumberInducedGC',    
                                                                            'NumberGen0Collections',
                                                                            'NumberGen1Collections',
                                                                            'NumberGen2Collections',
                                                                            'PromotedMemoryFromGen0',
                                                                            'PromotedMemoryFromGen1',
                                                                            'PercentTimeInGC',
                                                                            'LargeObjectHeapSize'
                                                                         ],
    
                                        'PerfProc_Process':              [
                                                                              'Name',
                                                                              'PrivateBytes',
                                                                              'ElapsedTime',
                                                                              'IDProcess',# pid
                                                                              'Caption',
                                                                              'CreatingProcessID',
                                                                              'Description',
                                                                              'IODataBytesPersec',
                                                                              'IODataOperationsPersec',
                                                                              'IOOtherBytesPersec',
                                                                              'IOOtherOperationsPersec',
                                                                              'IOReadBytesPersec',
                                                                              'IOReadOperationsPersec',
                                                                              'IOWriteBytesPersec',
                                                                              'IOWriteOperationsPersec'     
                                                                          ]
                                    }
    
        def get_pid_stats(self, pid):
            this_proc_dict = {}
    
            pythoncom.CoInitialize() # Needed when run by the same process in a thread
            if not self.perf_object_list:
                perf_object_list = self.supported_types.keys()
    
            for counter_type in perf_object_list:
                strComputer = "."
                objWMIService = win32com.client.Dispatch("WbemScripting.SWbemLocator")
                objSWbemServices = objWMIService.ConnectServer(strComputer,"root\cimv2")
    
                query_str = '''Select * from %s%s''' % (self.win32_perf_base,counter_type)
                colItems = objSWbemServices.ExecQuery(query_str) # "Select * from Win32_PerfFormattedData_PerfProc_Process")# changed from Win32_Thread        
    
                if len(colItems) > 0:        
                    for objItem in colItems:
                        if hasattr(objItem, 'IDProcess') and pid == objItem.IDProcess:
    
                                for attribute in self.supported_types[counter_type]:
                                    eval_str = 'objItem.%s' % (attribute)
                                    this_proc_dict[attribute] = eval(eval_str)
    
                                this_proc_dict['TimeStamp'] = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.') + str(datetime.datetime.now().microsecond)[:3]
                                break
    
            return this_proc_dict      
    
    
        def get_stats(self):
            '''
            Show process stats for all processes in given list, if none given return all processes   
            If filter list is defined return only the items that match or contained in the list
            Returns a list of result dictionaries
            '''    
            pythoncom.CoInitialize() # Needed when run by the same process in a thread
            proc_results_list = []
            if not self.perf_object_list:
                perf_object_list = self.supported_types.keys()
    
            for counter_type in perf_object_list:
                strComputer = "."
                objWMIService = win32com.client.Dispatch("WbemScripting.SWbemLocator")
                objSWbemServices = objWMIService.ConnectServer(strComputer,"root\cimv2")
    
                query_str = '''Select * from %s%s''' % (self.win32_perf_base,counter_type)
                colItems = objSWbemServices.ExecQuery(query_str) # "Select * from Win32_PerfFormattedData_PerfProc_Process")# changed from Win32_Thread
    
                try:  
                    if len(colItems) > 0:
                        for objItem in colItems:
                            found_flag = False
                            this_proc_dict = {}
    
                            if not self.process_name_list:
                                found_flag = True
                            else:
                                # Check if process name is in the process name list, allow print if it is
                                for proc_name in self.process_name_list:
                                    obj_name = objItem.Name
                                    if proc_name.lower() in obj_name.lower(): # will log if contains name
                                        found_flag = True
                                        break
    
                            if found_flag:
                                for attribute in self.supported_types[counter_type]:
                                    eval_str = 'objItem.%s' % (attribute)
                                    this_proc_dict[attribute] = eval(eval_str)
    
                                this_proc_dict['TimeStamp'] = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.') + str(datetime.datetime.now().microsecond)[:3]
                                proc_results_list.append(this_proc_dict)
    
                except pywintypes.com_error, err_msg:
                    # Ignore and continue (proc_mem_logger calls this function once per second)
                    continue
            return proc_results_list     
    
    
    def get_sys_stats():
        ''' Returns a dictionary of the system stats'''
        pythoncom.CoInitialize() # Needed when run by the same process in a thread
        x = winmem()
    
        sys_dict = { 
                        'dwAvailPhys': x.dwAvailPhys,
                        'dwAvailVirtual':x.dwAvailVirtual
                    }
        return sys_dict
    
    
    if __name__ == '__main__':
        # This area used for testing only
        sys_dict = get_sys_stats()
    
        stats_processor = process_stats(process_name_list=['process2watch'],perf_object_list=[],filter_list=[])
        proc_results = stats_processor.get_stats()
    
        for result_dict in proc_results:
            print result_dict
    
        import os
        this_pid = os.getpid()
        this_proc_results = stats_processor.get_pid_stats(this_pid)
    
        print 'this proc results:'
        print this_proc_results
    

    http://monkut.webfactional.com/blog/archive/2009/1/21/windows-process-memory-logging-python

    0 讨论(0)
提交回复
热议问题