From 0019a3d925315c7836814ee03df2ebdc8be0fb68 Mon Sep 17 00:00:00 2001 From: rarmitag <35737789+rarmitag@users.noreply.github.com> Date: Wed, 11 Sep 2024 15:13:49 -0700 Subject: [PATCH] 319_delete_ar_prompt (#2981) * 319_delete_ar_prompt * 319_delete_AR_PROMPT_FIX --- colin-api/src/colin_api/resources/filing.py | 32 +++++++++++++++++++++ 1 file changed, 32 insertions(+) diff --git a/colin-api/src/colin_api/resources/filing.py b/colin-api/src/colin_api/resources/filing.py index 147712531..4d3ff5d05 100644 --- a/colin-api/src/colin_api/resources/filing.py +++ b/colin-api/src/colin_api/resources/filing.py @@ -296,3 +296,35 @@ 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:identifier/filings') +# 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 \ No newline at end of file