def hw_remove_reserva(id):
session = SessionLocal()
reserva = session.query(Reserva).filter_by(id=id).first()
session.delete(reserva)
session.commit()
session.close()
return {"message": f"Reserva {reserva.id} was deleted"}
def hw_remove_aeroporto(id):
session = SessionLocal()
aeroporto = session.query(Aeroporto).filter_by(id=id).first()
session.delete(aeroporto)
session.commit()
session.close()
return {"message": f"Aeroporto {aeroporto.id} was deleted"}
def hw_remove_voo(id):
session = SessionLocal()
voo = session.query(Voo).filter_by(id=id).first()
session.delete(voo)
session.commit()
session.close()
return {"message": f"Voo {voo.id} was deleted"}
def hw_update_aeroporto(request):
session = SessionLocal()
aeroporto_old = session.query(Aeroporto).filter_by(
id=request["id"]).first()
session.delete(aeroporto_old)
aeroporto = Aeroporto(id=request["id"],
nome=request["nome"],
cidade=request["cidade"])
session.add(aeroporto)
session.commit()
aeroporto_json = hw_get_aeroporto(aeroporto.id)
session.close()
return aeroporto_json
def hw_update_reserva(request):
session = SessionLocal()
reserva_old = session.query(Reserva).filter_by(id=request["id"]).first()
session.delete(reserva_old)
reserva = Reserva(
id_voo=request["id_voo"],
id_cadastro=request["id_cadastro"],
e_ticket=request["e_ticket"],
)
session.add(reserva)
session.commit()
reserva_json = hw_get_reserva(reserva.id)
session.close()
return reserva_json
def delete_video_by_oid(oid):
"""
查询bilibiliVideos表中是否已存在该视频
"""
try:
session = SessionLocal()
# 查询是否存在
del_video = session.query(BilibiliVideo).filter(
BilibiliVideo.oid == oid).first()
session.delete(del_video)
except Exception as e:
print(e)
session.close()
return False
else:
session.commit()
return True
def hw_update_voo(request):
session = SessionLocal()
voo_old = session.query(Voo).filter_by(id=request["id"]).first()
session.delete(voo_old)
voo = Voo(
id=request["id"],
data=datetime.strptime(request["data"], "%d/%m/%Y %H:%M:%S"),
destino=request["destino"],
companhia=request["companhia"],
capacidade=request["capacidade"],
ocupacao=request["ocupacao"],
preco=request["preco"],
)
session.add(voo)
session.commit()
voo_json = hw_get_voo(voo.id)
session.close()
return voo_json
