Delete notes of a removed patient (#131)

* Delete notes of a removed patient

* Fix flake8 issues

Author: tschaffter

server/openapi_server/controllers/note_controller.py

@@ -83,8 +83,23 @@ def delete_note(dataset_id, fhir_store_id, note_id):  # noqa: E501
     res = None
     status = None
     try:
-        db_note = DbNote.objects.get(id=note_id)
-        db_note.delete()
+        DbNote.objects.get(id=note_id).delete()
+        res = {}
+        status = 200
+    except DoesNotExist:
+        status = 404
+        res = Error("The specified resource was not found", status)
+    except Exception as error:
+        status = 500
+        res = Error("Internal error", status, str(error))
+    return res, status
+
+
+def delete_notes_by_patient(patientId):
+    res = None
+    status = None
+    try:
+        DbNote.objects(patientId=patientId).delete()
         res = {}
         status = 200
     except DoesNotExist:

server/openapi_server/controllers/patient_controller.py

@@ -9,6 +9,7 @@
 from openapi_server.models.patient_create_request import PatientCreateRequest  # noqa: E501
 from openapi_server.models.patient_create_response import PatientCreateResponse  # noqa: E501
 from openapi_server.config import Config
+from openapi_server.controllers.note_controller import delete_notes_by_patient  # noqa: E501
 
 
 def create_patient(dataset_id, fhir_store_id):  # noqa: E501
@@ -73,6 +74,7 @@ def delete_patient(dataset_id, fhir_store_id, patient_id):  # noqa: E501
     res = None
     status = None
     try:
+        delete_notes_by_patient(patient_id)
         db_patient = DbPatient.objects.get(id=patient_id)
         db_patient.delete()
         res = {}