yuuki/libs/data.py

#!/usr/bin/python3
# coding=UTF-8

import os, time, json
from .core.ttypes import OpType

class Yuuki_Data:
    def __init__(self):

        # Data

        self.Data = {}

        self.DataType = {
            "Global":{
                "LastResetLimitTime":None,
            },
            "Group": {},
            "LimitInfo":{},
            "BlackList":[]
        }

        self.GroupType = {
            "SEGroup":None,
            "Ext_Admin":[],
            "GroupTicket":{}
        }

        self.LimitType = {
            "KickLimit":{},
            "CancelLimit":{}
        }

        self.SEGrouptype = {
            OpType.NOTIFIED_UPDATE_GROUP:False,
            OpType.NOTIFIED_INVITE_INTO_GROUP:False,
            OpType.NOTIFIED_ACCEPT_GROUP_INVITATION:False,
            OpType.NOTIFIED_KICKOUT_FROM_GROUP:False
        }

        self.initType = {
            "Group":self.GroupType,
            "LimitInfo":self.LimitType,
            "SEGroup":self.SEGrouptype
        }

        self.DataPath = "data/"
        self.DataName = "{}.json"

        if not os.path.isdir(self.DataPath):
            os.mkdir(self.DataPath)

        for Type in self.DataType:
            name = self.DataPath + self.DataName.format(Type)
            if not os.path.isfile(name):
                with open(name, "w") as f:
                    f.write("")
                Type = 0
            else:
                with open(name, "r") as f:
                    try:
                        json.loads(f.read())
                        Type = 0
                    except ValueError:
                        Type = 1
            assert Type == 0, "{}\nJson Test Error".format(name)

        # Data Initialize

        for Type in self.DataType:
            name = self.DataPath + self.DataName.format(Type)
            with open(name, "r+") as f:
                text = f.read()
                if text != "":
                    self.Data[Type] = json.loads(text)
                else:
                    self.Data[Type] = self.DataType[Type]
                    f.write(json.dumps(self.Data[Type]))

        # Log

        self.LogType = {
            "JoinGroup":"<li>%s: %s(%s) -> Inviter: %s</li>",
            "KickEvent":"<li>%s: %s(%s) -(%s)> Kicker: %s | Kicked: %s | Status: %s</li>",
            "CancelEvent":"<li>%s: %s(%s) -(%s)> Inviter: %s | Canceled: %s</li>",
            "BlackList":"<li>%s: %s(%s)</li>"
        }

        self.LogPath = "logs/"
        self.LogName = "{}.html"

        self.initHeader = "<title>{} - SYB</title>" \
                          "<meta charset='utf-8' />"

        if not os.path.isdir(self.LogPath):
            os.mkdir(self.LogPath)

        for Type in self.LogType:
            name = self.LogPath + self.LogName.format(Type)
            if not os.path.isfile(name):
                with open(name, "w") as f:
                    f.write(self.initHeader.format(Type))

    def file(self, Type, Mode, Format):
        if Format == "Data":
            return open(self.DataPath + self.DataName.format(Type), Mode)
        elif Format == "Log":
            return open(self.LogPath + self.LogName.format(Type), Mode)

    def syncData(self):
        for Type in self.DataType:
            with self.file(Type, "w", "Data") as f:
                f.write(json.dumps(self.Data[Type]))

    def updateData(self, Object, Input, Data):
        if type(Object) == list:
            if Input:
                Object.append(Data)
            else:
                Object.remove(Data)
        elif type(Object) == dict:
            Object[Input] = Data
        self.syncData()

    def updateLog(self, Type, Data):
        with self.file(Type, "a", "Log") as f:
            f.write(self.LogType[Type] % Data)

    def getTime(self, format="%b %d %Y %H:%M:%S %Z"):
        Time = time.localtime(time.time())
        return time.strftime(format, Time)

    def getData(self, Type, Query=None):
        if type(Query) == list:
            if len(Query) == 1:
                if Query not in self.Data[Type]:
                    if Type in self.initType:
                        self.Data[Type][Query] = self.initType[Type]
                    else:
                        self.Data[Type][Query] = {}
                return self.Data[Type][Query]
            elif len(Query) == 2:
                if Query[0] not in self.Data[Type]:
                    if Type in self.initType:
                        self.Data[Type][Query[0]] = self.initType[Type]
                    else:
                        self.Data[Type][Query[0]] = self.initType[Query]
                if Query[1] not in self.Data[Type][Query]:
                    if Type in self.initType:
                        self.Data[Type][Query[0]][Query[1]] = self.initType[Query[0]]
                    else:
                        self.Data[Type][Query[0]][Query[1]] = {}
                return self.Data[Type][Query[0]][Query[1]]
            else:
                assert "Error Query Level"
        else:
            return self.Data[Type]

    def getLimit(self, Type):
        if Type == "Kick":
            Limit = {}
            for userId in self.getData("LimitInfo", "KickLimit"):
                Limit[userId] = int(self.getData("LimitInfo", ["KickLimit", userId], 3))
        elif Type == "Cancel":
            Limit = {}
            for userId in self.getData("LimitInfo", "CancelLimit"):
                Limit[userId] = int(self.getData("LimitInfo", ["CancelLimit", userId], 3))
        else:
            Limit = None
        return Limit

    def getSEGroup(self, GroupID):
        SEMode = self.getData("Group", GroupID)["SEGroup"]
        if SEMode == None:
            return None
        SEMode_ = {}
        for Mode in SEMode:
            SEMode_[int(Mode)] = SEMode[Mode]
        return SEMode_