EVOLUTION-NINJA

Edit File: 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)