Escreva um programa Python para ler as notas n1 e n2 de um aluno e calcule a sua média aritmética - Exercícios Resolvidos de Python

Informe a nota N1: 9.4
Informe a nota N2: 11
Nota inválida. Informe a nota N2: 8.3
A média do aluno é: 8.85
O aluno foi aprovado

Informe a nota N1: 4.2
Informe a nota N2: 1.8
A média do aluno é: 3.0
O aluno reprovou direto.

Informe a nota N1: 5.7
Informe a nota N2: 6
A média do aluno é: 5.85
Informe a nota do exame: 7
A média do aluno com o exame é: 6.425
O aluno foi aprovado após o exame.

# método principal
def main():
  # vamos ler a primeira nota
  n1 = float(input("Informe a nota N1: "))
  # a nota é válida?
  while ((n1 < 0) or (n1 > 10)):
    n1 = float(input("Nota inválida. Informe a nota N1: "))
    
  # vamos ler a segunda nota
  n2 = float(input("Informe a nota N2: "))
  # a nota é válida?
  while ((n2 < 0) or (n2 > 10)):
    n2 = float(input("Nota inválida. Informe a nota N2: "))
    
  # vamos calcular a média do aluno
  media = (n1 + n2) / 2.0
  print("A média do aluno é: {0}".format(media))
    
  # o aluno foi aprovado?
  if (media >= 7.0):
    print("O aluno foi aprovado")
  # o aluno ficou de exame?
  elif ((media >= 3.5) and (media < 7.0)):
    # vamos ler a nota do exame
    exame = float(input("Informe a nota do exame: "))
    # a nota é válida?
    while ((exame < 0) or (exame > 10)):
      print("Nota inválida. Informe a nota do exame: ")
      exame = float(input("Nota inválida. Informe a nota do exame: "))
      
    # calcula a nota média
    media = (media + exame) / 2.0
    print("A média do aluno com o exame é: {0}".format(media))
      
    # o aluno foi aprovado após o exame?
    if (media >= 5.0):
      print("O aluno foi aprovado após o exame.")
    else:
      print("O aluno foi reprovado após o exame.")
  # reprovou direto
  else:
    print("O aluno reprovou direto.")
  
if __name__== "__main__":
  main()