modify

main.py

@@ -39,6 +39,7 @@ app.add_middleware(
 class TraitementDB(Base):
     __tablename__ = "traitements"
     id = Column(Integer, primary_key=True, index=True)
+    uid = Column(String, nullable=False)  # Ajout du champ UID
     date_traitement = Column(TIMESTAMP)
     type_traitement = Column(String)
     produit_utilise = Column(String)
@@ -55,9 +56,6 @@ def ensure_table_exists():
     else:
         print("✅ Table 'traitements' déjà présente.")
 
-# # Appel de la fonction pour garantir la présence de la table
-# ensure_table_exists()
-
 # Dépendance à injecter dans les routes
 def get_db():
     db = SessionLocal()
@@ -69,6 +67,7 @@ def get_db():
 # Schéma de réponse
 class TraitementResponse(BaseModel):
     id: int
+    uid: str  # Ajout du champ UID
     date_traitement: datetime
     type_traitement: str
     produit_utilise: str
@@ -77,10 +76,11 @@ class TraitementResponse(BaseModel):
     is_deleted: int
 
     class Config:
-        from_attributes = True  # ✅ Pydantic v2+ (remplace orm_mode = True)
+        from_attributes = True
 
 # Schéma de création
 class TraitementCreate(BaseModel):
+    uid: str  # Ajout du champ UID
     date_traitement: str  # ISO format attendu côté client
     type_traitement: str
     produit_utilise: str
@@ -96,13 +96,18 @@ def convert_db_to_pydantic(db_obj):
     return TraitementResponse.from_orm(db_obj)
 
 @app.get("/traitements", response_model=List[TraitementResponse])
-def get_traitements(db: Session = Depends(get_db)):
-    db_traitements = db.query(TraitementDB).filter(TraitementDB.is_deleted == 0).all()
+def get_traitements(uid: str, db: Session = Depends(get_db)):
+    db_traitements = db.query(TraitementDB).filter(
+        TraitementDB.is_deleted == 0,
+        TraitementDB.uid == uid
+    ).all()
     return [convert_db_to_pydantic(t) for t in db_traitements]
 
 @app.get("/traitements/all", response_model=List[TraitementResponse])
-def get_all_traitements(db: Session = Depends(get_db)):
-    db_traitements = db.query(TraitementDB).all()
+def get_all_traitements(uid: str, db: Session = Depends(get_db)):
+    db_traitements = db.query(TraitementDB).filter(
+        TraitementDB.uid == uid
+    ).all()
     return [convert_db_to_pydantic(t) for t in db_traitements]
 
 @app.post("/traitements", response_model=TraitementResponse)
@@ -124,10 +129,13 @@ def create_traitement(traitement: TraitementCreate, db: Session = Depends(get_db
     return convert_db_to_pydantic(db_traitement)
 
 @app.patch("/traitements/{treatment_id}/soft-delete", response_model=TraitementResponse)
-def soft_delete_treatment(treatment_id: int, db: Session = Depends(get_db)):
-    traitement = db.query(TraitementDB).filter(TraitementDB.id == treatment_id).first()
+def soft_delete_treatment(treatment_id: int, uid: str, db: Session = Depends(get_db)):
+    traitement = db.query(TraitementDB).filter(
+        TraitementDB.id == treatment_id,
+        TraitementDB.uid == uid
+    ).first()
     if not traitement:
         raise HTTPException(status_code=404, detail="Traitement introuvable")
     traitement.is_deleted = 1
     db.commit()
-    return convert_db_to_pydantic(traitement)
+    return convert_db_to_pydantic(traitement)