smart-top/smart/cmd/__init__.py

"""
This file contains class and functions that extract data from running processes like top and stores them into a data store of the calling codes choice
dependencies:
    - top (on the os)
@TODO:
    Test this thing on windows to see if it works
"""
import pandas as pd
import numpy as np
import subprocess
import os
import datetime
# from transport import factory
import sys
import hashlib
import re
from io import StringIO
import importlib as IL
import importlib.util

class plugin :
    """
    This is a context specific decorator to be used for generic commands. 
    This decorator should be placed on parsing functions
    """
    def __init__(self,**_args):
        """
        :name   name of the plugin
        :mode   restrict to reader/writer
        :about  tell what the function is about    
        """
        self._cmd = _args['cmd'] 
        self._alias = _args['alias'] if 'alias' in _args else None
        
    def __call__(self,pointer,**kwargs):
        def wrapper(_args,**kwargs):
            return pointer(_args,**kwargs)
        #
        # @TODO:
        # add attributes to the wrapper object
        #
        # self._name = pointer.__name__ if not self._name else self._name
        # setattr(wrapper,'context',True)
        setattr(wrapper,'cmd',self._cmd)
        setattr(wrapper,'alias',self._alias)
        
        return wrapper
      
def read(**args) :
    """
    This function will perform the actual reads of process informations. 
    @return {user,pid,start,status, name, args, mem,cpu}
    """
    # cmd     = args['cmd'] #"ps -eo pid,user,pmem,pcpu,stat,etime,args|awk 'OFS=\";\" {$1=$1; if($5 > 9) print }'"
    xchar   = ";"
    
    try:
        _parser = args['parser'] if 'parser' in args else None
        stream = None
        cmd = None
        if _parser and hasattr(_parser,'cmd'):
            cmd = getattr(_parser,'cmd')
        elif 'cmd' in args :
            cmd = args['cmd']
        if cmd :
            handler = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE)			
            stream = str(handler.communicate()[0]).replace("b'",'')
            
        return _parser (stream) if _parser else stream

        # return stream
        #
        # At this point we need to load the parser or return the output as is
        # if 'parser' in args :
        #     _parser = args['parser']
            
        #     return _parser(stream)
        # else:
        #     return stream

    except Exception as e:
        print (e)
        pass
    
# if __name__ == '__main__' :
#     #
#     # Being directly called (external use of the )
#     print(read())