MarioWeb/IDManager.py

import random
import csv
class idManager():
    levelNum = 200
    timeMin = 2
    tutorialMax = 3
    ip_dic = {}
    ip_recent = {}
    ip_control = {}
    ip_time = {}
    ip_tutorial = {}
    def __int__(self):
        self.levelNum = 200


    def getLevels(self,ip):
        if ip not in self.ip_dic.keys():
            self.ip_dic[ip] = []

        levels = [random.randint(1,self.levelNum), random.randint(1,self.levelNum)]
        while levels[0] in self.ip_dic[ip]:
            levels[0] = random.randint(1,self.levelNum)
        self.ip_dic[ip].append(levels[0])
        while levels[1] in self.ip_dic[ip]:
            levels[1] = random.randint(1, self.levelNum)
        self.ip_dic[ip].append(levels[1])

        self.ip_recent[ip] = [levels[0],levels[1]]
        return ["lvl"+str(levels[0]),"lvl"+str(levels[1])]
        #return ["test1","test2"]

    def getRecent(self,ip):
        return self.ip_recent[ip]
    def setControl(self,ip,content):
        if content == "A":
            self.ip_control[ip] = 0
        else:
            self.ip_control[ip] = 1
    def getTimes(self,ip):
        print(self.ip_time[ip])
        if ip not in self.ip_time.keys():
            return 0
        else:
            if self.ip_time[ip]>=self.timeMin:
                return 1
            else:
                return 0
    def addTimes(self,ip):
        if ip not in self.ip_time.keys():
            self.ip_time[ip] = 0
        self.ip_time[ip] = self.ip_time[ip]+1
        return self.ip_time[ip]

    def addTutorial(self,ip):

        if ip not in self.ip_tutorial.keys():
            self.ip_tutorial[ip] = 1
            return 1
        else:
            if self.ip_tutorial[ip]>self.tutorialMax:
                return 1
            else:
                self.ip_tutorial[ip] = self.ip_tutorial[ip] + 1
                return self.ip_tutorial[ip]

    def hasNextTutorial(self,ip):
        if self.ip_tutorial[ip] == self.tutorialMax:
            return 0
        else:
            return 1
    def getControl(self,ip):
        return self.ip_control[ip]
    def write_csv(self,path, data):
        with open(path, 'a+', newline='') as f:
            csv_write = csv.writer(f)
            csv_write.writerow([*data, ''])