import datetime
import calendar
def adicionar_dias(data, dias):
return data + datetime.timedelta(days=dias)
def subtrair_dias(data, dias):
return data - datetime.timedelta(days=dias)
def adicionar_semanas(data, semanas):
return data + datetime.timedelta(weeks=semanas)
def subtrair_semanas(data, semanas):
return data - datetime.timedelta(weeks=semanas)
def adicionar_meses(data, meses):
ano = data.year + (data.month + meses - 1) // 12
mes = (data.month + meses - 1) % 12 + 1
dia = min(data.day, calendar.monthrange(ano, mes)[1])
return datetime.date(ano, mes, dia)
def subtrair_meses(data, meses):
ano = data.year + (data.month - meses - 1) // 12
mes = (data.month - meses - 1) % 12 + 1
dia = min(data.day, calendar.monthrange(ano, mes)[1])
return datetime.date(ano, mes, dia)
while True:
print("Escolha uma operação:")
print("1. Adicionar Dias")
print("2. Subtrair Dias")
print("3. Adicionar Semanas")
print("4. Subtrair Semanas")
print("5. Adicionar Meses")
print("6. Subtrair Meses")
print("7. Sair")
escolha = input("Digite o número da operação ou '7' para sair: ")
data = input("Digite a data (no formato YYYY-MM-DD): ")
data = datetime.datetime.strptime(data, '%Y-%m-%d').date()
if escolha in ('1', '2', '3', '4', '5', '6','7'):
quantidade = int(input("Digite a quantidade: "))
if escolha == '1':
resultado = adicionar_dias(data, quantidade)
elif escolha == '2':
resultado = subtrair_dias(data, quantidade)
elif escolha == '3':
resultado = adicionar_semanas(data, quantidade)
elif escolha == '4':
resultado = subtrair_semanas(data, quantidade)
elif escolha == '5':
resultado = adicionar_meses(data, quantidade)
elif escolha == '6':
resultado = subtrair_meses(data, quantidade)
elif escolha == '7':
print("Fim do Programa")
break
print("Resultado:", resultado)
else:
print("Escolha inválida. Tente novamente.")
