Tryag File Manager - Edit By Mr.SaHr

//lib/fm-agent/plugins/users.py

import agent_util
import logging

logger = logging.getLogger(__name__)

class UsersPlugin(agent_util.Plugin):
    textkey = "users"
    label = "Users"

@classmethod
    def get_metadata(self, config):
        status = agent_util.SUPPORTED
        msg = None

options = []

data = {
            "users.unique_login_count": {
                "label": "Users unique login count",
                "options": options,
                "status": status,
                "error_message": msg
            },
            "users.total_login_count": {
                "label": "Users total login count",
                "options": options,
                "status": status,
                "error_message": msg
            },
        }
        return data

def check(self, textkey, data, config):
        query = ''
        if (textkey == 'users.unique_login_count'):
            query = 'who |cut -c 1-9 |sort -u |wc -l'

if (textkey == 'users.total_login_count'):
            query = 'who | wc -l'

ret, output = agent_util.execute_command('%s' % query)
        return int(output)