From 42bf4bfad44d5aa3d23ac182adc6ad0afdd153e7 Mon Sep 17 00:00:00 2001 From: Aymeric Chaumont <aymeric.chaumont@student-cs.fr> Date: Wed, 13 Jul 2022 14:16:11 +0200 Subject: [PATCH] linting --- backend/main.py | 4 ++-- backend/video-capture.py | 10 +++++----- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/backend/main.py b/backend/main.py index cddefa0..7ab2713 100644 --- a/backend/main.py +++ b/backend/main.py @@ -47,7 +47,7 @@ async def get_records(place: str, db: Session = Depends(get_db)): @app.post('/api/records', response_model=schemas.Record) -async def stats(record: schemas.RecordBase, db: Session = Depends(get_db)): +async def post_records(record: schemas.RecordBase, db: Session = Depends(get_db)): db_record = models.Records(**record.dict()) db.add(db_record) db.commit() @@ -56,7 +56,7 @@ async def stats(record: schemas.RecordBase, db: Session = Depends(get_db)): @app.delete('/api/records', response_model=None) -async def stats(id: int, db: Session = Depends(get_db)): +async def del_records(id: int, db: Session = Depends(get_db)): if id == 0: db.query(models.Records).delete() else: diff --git a/backend/video-capture.py b/backend/video-capture.py index 9437c1b..530f4e6 100644 --- a/backend/video-capture.py +++ b/backend/video-capture.py @@ -3,7 +3,7 @@ from datetime import datetime, timedelta import numpy as np import keras from utils.preprocessing import fix_singular_shape, norm_by_imagenet -from db import models, database, schemas +from db import models from db.database import SessionLocal @@ -24,13 +24,13 @@ while(cap.isOpened()): count_prediction = np.sum(pred_map) waiting_time = timedelta(seconds=120 + count_prediction * 30) record = {"place": "local", - "date": current_time, - "density": count_prediction, - "waiting_time": waiting_time} + "date": current_time, + "density": count_prediction, + "waiting_time": waiting_time} db_record = models.Records(**record) db.add(db_record) db.commit() db.refresh(db_record) count += 1 cap.release() -cv2.destroyAllWindows() \ No newline at end of file +cv2.destroyAllWindows() -- GitLab