Delete ar prompt

colin-api/src/colin_api/resources/filing.py

@@ -296,3 +296,34 @@ def post(identifier, **kwargs):
         except Exception as err:  # pylint: disable=broad-except
             current_app.logger.error(f'Error updating AR status for {identifier}: {str(err)}')
             return jsonify({'message': 'Error updating AR status.'}), HTTPStatus.INTERNAL_SERVER_ERROR
+
+
+@cors_preflight('POST')
+@API.route('/<string:legal_type>/<string:identifier>/filings/arprompt')
+# pylint: disable=too-few-public-methods
+class DeleteARPrompt(Resource):
+    """Delete AR Prompt for corporation."""
+
+    @staticmethod
+    @cors.crossdomain(origin='*')
+    @jwt.requires_roles([COLIN_SVC_ROLE])
+    def post(identifier, **kwargs):
+        """Clean up data in Colin for the corporation."""
+        # pylint: disable=unused-argument
+        try:
+            with DB.connection as con:
+                with con.cursor() as cursor:
+                    # Delete from AR prompt for the given corporation
+                    delete_ar_prompt = """
+                        DELETE FROM AR_PROMPT  
+                        WHERE corp_num = :identifier
+                    """
+                    cursor.execute(delete_ar_prompt, {'identifier': identifier})
+                    # Commit the transaction
+                    con.commit()
+                    current_app.logger.info(f'Successfully deleted AR prompt for corporation {identifier}.')
+                    return jsonify({'message': f'AR prompt deleted for corporation {identifier}.'}), HTTPStatus.OK
+
+        except Exception as err:  # pylint: disable=broad-except
+            current_app.logger.error(f'Error Deleteing AR status for {identifier}: {str(err)}')
+            return jsonify({'message': 'Error Deleteing AR status.'}), HTTPStatus.INTERNAL_SERVER_ERROR