Shows all ambigram dates between two given years

ambigram_dates.py

'''
Dados os dois anos abaixo, este programa exibe todas as datas 
que são tanto palíndromas (capicuas) como ambigramas (rotacionais).
'''

ANO_INI = 2000
ANO_FIM = 2100

import datetime as dt

# representação em display de 7 segmentos
seg_rep = {
    '0': ('###', '# #', '# #', '# #', '###'),
    '1': ('  #', '  #', '  #', '  #', '  #'),
    '2': ('###', '  #', '###', '#  ', '###'),
    '3': ('###', '  #', '###', '  #', '###'),
    '4': ('# #', '# #', '###', '  #', '  #'),
    '5': ('###', '#  ', '###', '  #', '###'),
    '6': ('###', '#  ', '###', '# #', '###'),
    '7': ('###', '  #', '  #', '  #', '  #'),
    '8': ('###', '# #', '###', '# #', '###'),
    '9': ('###', '# #', '###', '  #', '###'),
}

# representação espelho em display de 7 segmentos
rev_seg_rep = {
    '0': ('###', '# #', '# #', '# #', '###'),
    '1': ('  #', '  #', '  #', '  #', '  #'),
    '2': ('###', '  #', '###', '#  ', '###'),
    '3': ('###', '#  ', '###', '#  ', '###'),
    '4': ('#  ', '#  ', '###', '# #', '# #'),
    '5': ('###', '#  ', '###', '  #', '###'),
    '6': ('###', '# #', '###', '  #', '###'),
    '7': ('#  ', '#  ', '#  ', '#  ', '###'),
    '8': ('###', '# #', '###', '# #', '###'),
    '9': ('###', '#  ', '###', '# #', '###'),
}

# exibe na tela a representação em 7 segmentos
def led_display(num, rev=False):
    if rev:
        digits = [rev_seg_rep[digit] for digit in reversed(str(num))]
    else:
        digits = [seg_rep[digit] for digit in str(num)]
    ret = ''
    for i in range(5):
        ret += "  ".join(s[i] for s in digits) + "\n"
    return ret

    
def ambigrama(num):
    return led_display(num) == led_display(num, True)

# onde tudo começa...
dt_ini = dt.date(ANO_INI,1,1)
dt_fim = dt.date(ANO_FIM,12,31)
delta = dt.timedelta(days=1)

while dt_ini <= dt_fim:
    dia = dt_ini.strftime("%d%m%Y")
    if ambigrama(dia):
        print("{} é um capicua e um ambigrama".format(dt_ini.strftime("%d/%m/%Y")))
        print(led_display(dia))
    dt_ini = dt_ini + delta