Correct update comptage (#2)

Reviewed-on: #2
This commit is contained in:
Dryusdan 2022-07-03 15:39:37 +02:00
parent a09d665e18
commit d08a0e5602

View file

@ -58,9 +58,9 @@ def get_comptage_by_date_and_boucle(date, id_boucle, unaccounted_table):
def update_comptage(comptage: schemas.ComptageBase, unaccounted_table):
with Session() as session:
if unaccounted_table:
stmt = update(models.unaccounted_Comptage).where(models.unaccounted_Comptage.id == comptage.id).values(count=comptage.count)
stmt = update(models.unaccounted_Comptage).where(models.unaccounted_Comptage.id == comptage.id).values(holiday=comptage.holiday, week_day=comptage.week_day, count=comptage.count)
else:
stmt = update(models.Comptage).where(models.Comptage.id == comptage.id).values(count=comptage.count)
stmt = update(models.Comptage).where(models.Comptage.id == comptage.id).values(holiday=comptage.holiday, week_day=comptage.week_day, count=comptage.count)
session.execute(stmt)
session.commit()
return True