Olá Glauber, tudo bem com você?
Meus parabéns pela implementação do código, fico feliz em ver que você foi além do proposto pela atividade, adicionando uma validação da entrada do usuário e fazendo o uso da biblioteca datetime, a dedicação e empenho é muito importante na consolidação do aprendizado.
Lembre-se que pode contar sempre com o fórum Alura, seja para compartilhar o seu desenvolvimento ou para tirar suas dúvidas.
Fico à disposição.
Abraços e bons estudos.