Revize 9cf9a19d

Re #8572 - Added the `delete_certificate(id)` method to the Controller.

src/controllers/certificates_controller.py

@@ -10 +10 @@
     SSL_ID, SIGNATURE_ID, AUTHENTICATION_ID, \
     DATETIME_FORMAT, ROOT_CA_ID, INTERMEDIATE_CA_ID, CERTIFICATE_ID  # TODO DATABASE_FILE - not the Controller's
 from src.exceptions.database_exception import DatabaseException
+from src.exceptions.unknown_exception import UnknownException
 from src.model.subject import Subject
 from src.services.certificate_service import CertificateService, RevocationReasonInvalidException, \
     CertificateStatusInvalidException, CertificateNotFoundException, CertificateAlreadyRevokedException
@@ -161 +162 @@
         cert = self.certificate_service.get_certificate(v)
 
         if cert is None:
-            return E_NO_CERTIFICATES_FOUND, C_NO_DATA                               
+            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
         else:
             return {"success": True, "data": cert.pem_data}, C_SUCCESS
 
@@ -183 +184 @@
         cert = self.certificate_service.get_certificate(v)
 
         if cert is None:
-            return E_NO_CERTIFICATES_FOUND, C_NO_DATA                               
+            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
         else:
             data = self.cert_to_dict_full(cert)
             if data is None:
@@ -239 +240 @@
         if certs is None:
             return E_GENERAL_ERROR, C_INTERNAL_SERVER_ERROR
         elif len(certs) == 0:
-            return E_NO_CERTIFICATES_FOUND, C_NO_DATA                               
+            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
         else:
             ret = []
             for c in certs:
@@ -296 +297 @@
         cert = self.certificate_service.get_certificate(v)
 
         if cert is None:
-            return E_NO_CERTIFICATES_FOUND, C_NO_DATA                               
+            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
 
         if cert.parent_id is None:
             return E_NO_CERTIFICATES_FOUND, C_NO_DATA
@@ -448 +449 @@
             return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
         else:
             return {"success": True, "data": self.certificate_service.get_public_key_from_certificate(cert)}, C_SUCCESS
+
+    def delete_certificate(self, id):
+        """
+        Deletes a certificate identified by ID, including its corresponding subtree (all descendants).
+        :param id: target certificate ID
+        :rtype: DeleteResponse
+        """
+        try:
+            v = int(id)
+        except ValueError:
+            return E_WRONG_PARAMETERS, C_BAD_REQUEST
+
+        try:
+            self.certificate_service.delete_certificate(v)
+        except CertificateNotFoundException:
+            return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
+        except DatabaseException:
+            return E_CORRUPTED_DATABASE, C_INTERNAL_SERVER_ERROR
+        except CertificateStatusInvalidException or RevocationReasonInvalidException or UnknownException:
+            return E_GENERAL_ERROR, C_INTERNAL_SERVER_ERROR
+
+        return {"success": True, "data": "The certificate and its descendants have been successfully deleted."}