import re
import urllib.request
global tablica
def wypluj():
    flag=input("Czy chcesz poznac literki województw?")
    if flag==("Tak" or "TAK" or "y" or "Y" or "T" or "t"):
        flag=True
        return flag
    else:
        flag=False
        return flag

def wpisywanie(zrodlo):
    tablica=input("Podaj tablicę rejestracyjną")
    tablica.upper()
    dl_tab=len(tablica)
    if(dl_tab== 6 or dl_tab== 7):
        if re.match("^[A-Z]+.*", tablica):
            if re.match("U+.*", tablica):
                print("pojazd sił zbrojnych")
            else:
                if re.match("[A-Z]{2,3}.[0-9]*$", tablica):
                    print("Normalna tablica")
                    flag=wypluj()
                    print(re.match("[A-Z]{2,3}.[0-9]*$", tablica).group())
                    pattern=re.compile(r'<td>([A-Z])</td>\r\n<td>(\w+-?\w+)</td>')
                    wojewodztwa=pattern.findall(zrodlo)
                    for i in range(len(wojewodztwa)):
                        if wojewodztwa[i][0]==tablica[0]:
                            print("Pojazd jest z wojewodztwa ",wojewodztwa[i][1])
                    pattern=re.compile(r'<td>([A-Z]{2,3})</td>\r\n<td>(\w+\s?-?\s?\w+)</td>')
                    miasta=pattern.findall(zrodlo)
                    for i in range(len(miasta)):
                        if miasta[i][0]==tablica[:2] or miasta[i][0]==tablica[:3]:
                            print("Pojazd jest z miasta/powiatu ", miasta[i][1])
                    if(flag==True):
                        for i in range(len(wojewodztwa)):
                            print("Literka ",wojewodztwa[i][0], " Wojewodztwo ", wojewodztwa[i][1])

        else:
            print("Podano zły format rejestracji (musi zaczynać się literką)")
            wpisywanie()
    else :
        print("Podano za dużo lub mało znaków (musi być 6 lub 7)")
        wpisywanie()


def main():
    URL="http://tablicerejestracyjne.pl/polska/2000"
    zrodlo=urllib.request.urlopen(URL).read().decode()
    wpisywanie(zrodlo)

if __name__ == '__main__':
    main()