2020-04-code2.py

import string 

with open("2020-04-input.txt") as arquivo:
    lista_linhas = arquivo.read()

lista_linhas = lista_linhas.replace('\n\n','AAAAA')
lista_linhas = lista_linhas.replace('\n',' ')
lista_linhas = lista_linhas.replace('AAAAA','\n').splitlines()

contador_validos = 0



for linha in lista_linhas:
    total_itens = len(linha.split(' '))
    if total_itens in range(7,9):
        lista_itens = linha.strip().split(' ')
        dicionario = {}
        lista_itens_validar = ['byr', 'eyr', 'iyr', 'hgt', 'hcl', 'ecl', 'pid']
        lista_cores_validas = ['amb','blu','brn','gry','grn','hzl','oth']
        for itens in lista_itens:
            chave,valor = itens.strip().split(':')
            dicionario[chave] = valor
        
        check = all(item in dicionario.keys() for item in lista_itens_validar)
        if check:
            if (int(dicionario['byr']) in range(1920,2003)) and (int(dicionario['iyr']) in range(2010,2021)) and (int(dicionario['eyr']) in range(2020,2031)):
                if (('in' in dicionario['hgt']) and (int(dicionario['hgt'][:-2]) in range(59,77))) or (('cm' in dicionario['hgt']) and (int(dicionario['hgt'][:-2]) in range(150,194))):
                    if(('#' in dicionario['hcl']) and all(caracteres in string.hexdigits for caracteres in dicionario['hcl'][1:])):
                        if dicionario['ecl'] in lista_cores_validas:
                            if len(dicionario['pid']) == 9:
                                contador_validos += 1
                                print(dicionario, 'OK',dicionario['pid'],contador_validos)
                            else:
                                print(dicionario,'pid zoado')
                        else:
                            print(dicionario,'ecl zoada')
                    else:
                        print(dicionario,'hcl zoado')
                else:
                    print(dicionario,'hgt zoado')
            else:
                print(dicionario,'valor ruim byr,iyr ou eyr')
        else:
            print(dicionario,'NOK')