ryuko-ng/helpers/userlogs.py

import json
import time

userlog_event_types = {"warns": "Warn",
                       "bans": "Ban",
                       "kicks": "Kick",
                       "mutes": "Mute",
                       "notes": "Note"}


def get_userlog():
    with open("data/userlog.json", "r") as f:
        return json.load(f)


def set_userlog(contents):
    with open("data/userlog.json", "w") as f:
        f.write(contents)


def userlog(uid, issuer, reason, event_type, uname: str = ""):
        userlogs = get_userlog()
        uid = str(uid)
        if uid not in userlogs:
            userlogs[uid] = {"warns": [],
                             "mutes": [],
                             "kicks": [],
                             "bans": [],
                             "notes": [],
                             "watch": False,
                             "name": "n/a"}
        if uname:
            userlogs[uid]["name"] = uname
        timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
        log_data = {"issuer_id": issuer.id,
                    "issuer_name": f"{issuer}",
                    "reason": reason,
                    "timestamp": timestamp}
        if event_type not in userlogs[uid]:
            userlogs[uid][event_type] = []
        userlogs[uid][event_type].append(log_data)
        set_userlog(json.dumps(userlogs))
        return len(userlogs[uid][event_type])