path = 'assets/data/'

class Partida(object):
    
    def __init__(self, aCharacter=None):
        self.mCharacter = aCharacter
        area1 = []
        for i in range(6):
            area1.append(0)
        area2 = []
        for i in range(6):
            area2.append(0)
        area3 = []
        for i in range(6):
            area3.append(0)
        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 getPartidasFromFile():
    par = {}
    try:
        f = open(path + 'data_partidas', 'r')
    except:
        print 'Error al abrir 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.strip('\n')
        aux = str(aux)
        #print 'linea', 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
    #print 'se levantaron estas partidas', par.keys()
    return par
    
def aux1(line):
    line = line.strip('\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 al guardar 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.strip('[')
                    s = s.strip(']')
                    f.write(s + '\n')
        else:
            f.write('None\n')
    f.close()
    print 'partidas guardadas'