escrever um bom python é simples. tipos + docstrings + limites de linha [80 caracteres] Veja como esta função é bonita.