path = 'assets/data/'

class Partida(object):
    
    def __init__(self, aCharacter=None):
        self.mCharacter = aCharacter
        area1 = [0] * 6
        area2 = [0] * 6
        area3 = [0] * 6
        self.mAreas = {1: area1, 2: area2, 3: area3}
        
    def isEmpty(self):
        return (self.mCharacter == None)
    
    def setCharacter(self, aCharacter):
        self.mCharacter = aCharacter
        
    def getCharacter(self):
        return self.mCharacter
    
    def getAreas(self):
        return self.mAreas
    
    def getArea(self, n):
        return self.mAreas[n]
    
    def setGame(self, a, j, s):
        self.mAreas[a][j] = s
        
def getPartidasFromFile():
    par = {}
    try:
        f = open(path + 'data_partidas', 'r')
    except:
        print 'Error opening partidas'
        p1 = Partida()
        p2 = Partida()
        p3 = Partida()
        par = {1: p1, 2: p2, 3:p3}
        return par
    
    for i in [1, 2, 3]:
        l = f.readline()
        aux = l.replace('\n', '')
        aux = str(aux)
        if not((aux == 'None') or (aux == '')):
            p = Partida(aux)
            for j in [1, 2, 3]:
                l = f.readline()
                p.mAreas[j] = aux1(l)
        else:
            p = Partida()
            
        par[i] = p

    return par
    
def aux1(line):
    line = line.replace('\n', '')
    line = line.split(', ')
    line_int = []
    for e in line:
        try:
            n = int(e)
        except:
            n = 0
        line_int.append(n)    
    return line_int

def setPartidasToFile(par):
    try:
        f = open(path + 'data_partidas', 'w')
    except:
        print 'Error saving partidas'
        return

    for i in [1, 2, 3]:
        if par.has_key(i):
            p = par[i]
            if p.mCharacter == None:
                f.write('None\n')
            else:
                f.write(p.mCharacter + '\n')
                for j in p.mAreas.keys():
                    a = p.mAreas[j]
                    s = str(a)
                    s = s.replace('[', '')
                    s = s.replace(']', '')
                    f.write(s + '\n')
        else:
            f.write('None\n')
    f.close()
    print 'saving partidas'