save database

khiemauto · Jan 12, 2021 · 68cb254 · 68cb254
1 parent dd59de6
commit 68cb254
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/rest.py b/api/rest.py
@@ -152,6 +152,8 @@ async def add_images_database(user_name: str, files: List[UploadFile] = File(...
             for face_data in face_datas:
                 face_dicts[face_data["StaffCode"]] = face_data
             share_param.face_infos = face_dicts
+
+            self.system.save_database(self.db_folder_path)
 
             return JSONResponse(content, status_code=status.HTTP_201_CREATED)
 
@@ -206,6 +208,8 @@ async def delete_user(user_name: str):
                 face_dicts[face_data["StaffCode"]] = face_data
             share_param.face_infos = face_dicts
 
+            self.system.save_database(self.db_folder_path)
+
             if ret:
                 return PlainTextResponse(f"Removed {nbphoto} photos of {user_name}", status_code=status.HTTP_200_OK)
             else: