import pickle

class Ledger():
    def __init__(self) -> None:
        self.data = [] # [ID, wins, losses, money]
        self.indexes = {} # ID: Index
    
    def readLedger(self, ID):
        data = None
        index = self.indexes.get(ID, None)
        if index is not None:
            data = self.data[ID]
        return data
    
    def writeLedger(self, ID, data):
        newData = self.readLedger(ID)

        if newData is not None:
            _, wins, losses, money = newData
            _, newWins, newLosses, newMoney = data
            wins += newWins
            losses += newLosses
            money += newMoney
            data = [ID, wins, losses, money]
            index = self.indexes[ID]
            self.data[index] = data
        else:
            self.data.append(data)
            index = len(self.data) - 1
            self.indexes.update({ID: index})