fix: clinic approval flow api

services/clinicServices.py

@@ -3,7 +3,7 @@ from sqlalchemy.orm import Session, joinedload
 from models import Clinics
 from schemas.UpdateSchemas import ClinicStatusUpdate, ClinicUpdate
 from schemas.ResponseSchemas import Clinic
-from typing import List, Literal, Union
+from typing import List, Literal, Optional, Union
 from exceptions import ResourceNotFoundException
 from enums.enums import ClinicStatus, UserType
 from exceptions.unauthorized_exception import UnauthorizedException
@@ -191,26 +191,63 @@ class ClinicServices:
         return counts
         
     
-    def update_clinic_status(self, user, clinic_id: int, status: ClinicStatus):
+    def update_clinic_status(self, user, clinic_id: int, status: ClinicStatus, documentStatus: Optional[dict] = None, rejection_reason: Optional[str] = None):
+        try:
+            if user["userType"] != UserType.SUPER_ADMIN:
+                raise UnauthorizedException("You are not authorized to update clinic status")
 
-        if user["userType"] != UserType.SUPER_ADMIN:
-            raise UnauthorizedException("You are not authorized to update clinic status")
+            clinic = self.db.query(Clinics).filter(Clinics.id == clinic_id).first()
 
-        clinic = self.db.query(Clinics).filter(Clinics.id == clinic_id).first()
+            if clinic is None:
+                raise ResourceNotFoundException("Clinic not found")
 
-        if clinic is None:
-            raise ResourceNotFoundException("Clinic not found")
+            clinic.status = status
+            self.db.add(clinic)
+            self.db.commit()
+            self.db.refresh(clinic)
 
-        clinic.status = status
-        self.db.add(clinic)
-        self.db.commit()
-        self.db.refresh(clinic)
+            if clinic.status == ClinicStatus.ACTIVE:
+                clinic_file_verification = self.db.query(ClinicFileVerifications).filter(ClinicFileVerifications.clinic_id == clinic_id).first()
 
-        clinic_response = Clinic(**clinic.__dict__.copy())
+                clinic_file_verification.logo_is_verified = True
+                clinic_file_verification.abn_doc_is_verified = True
+                clinic_file_verification.contract_doc_is_verified = True
 
-        # if rejected then email to clinic creator
-        if clinic.status == ClinicStatus.REJECTED:
-            pass
+                self.db.add(clinic_file_verification)
+                self.db.commit()
 
-        return clinic_response
+
+            if clinic.status == ClinicStatus.REJECTED or clinic.status == ClinicStatus.UNDER_REVIEW:
+                clinic_file_verification = self.db.query(ClinicFileVerifications).filter(ClinicFileVerifications.clinic_id == clinic_id).first()
+
+
+                if documentStatus and "LOGO" in documentStatus:
+                    clinic_file_verification.logo_is_verified = documentStatus.get("LOGO")
+                    clinic_file_verification.rejection_reason = rejection_reason
+
+                if documentStatus and "ABN" in documentStatus:
+                    clinic_file_verification.abn_doc_is_verified = documentStatus.get("ABN")
+                    clinic_file_verification.rejection_reason = rejection_reason
+
+                if documentStatus and "CONTRACT" in documentStatus:
+                    clinic_file_verification.contract_doc_is_verified = documentStatus.get("CONTRACT")
+                    clinic_file_verification.rejection_reason = rejection_reason
+
+
+                self.db.add(clinic_file_verification)
+                self.db.commit()
+
+
+            # if rejected or under review then email to clinic creator
+            if clinic.status == ClinicStatus.REJECTED or clinic.status == ClinicStatus.UNDER_REVIEW:
+                pass
+
+            # handle inactive status
+            if clinic.status == ClinicStatus.INACTIVE:
+                pass
+
+            return 
+        except Exception as e:
+            print(e)
+            raise Exception(e)