sbt-idp/cope2n-ai-fi/celery_worker/mock_process_tasks_fi.py

from celery_worker.worker_fi import app
from celery_worker.client_connector_fi import CeleryConnector
from common.process_pdf import compile_output_sbt

@app.task(name="process_fi_invoice")
def process_invoice(rq_id, list_url):
    from celery_worker.client_connector_fi import CeleryConnector
    from common.process_pdf import compile_output_fi
    
    c_connector = CeleryConnector()
    try:
        result = compile_output_fi(list_url)
        hoadon = {"status": 200, "content": result, "message": "Success"}
        print(hoadon)
        c_connector.process_fi_invoice_result((rq_id, hoadon))
        return {"rq_id": rq_id}
    except Exception as e:
        print(e)
        hoadon = {"status": 404, "content": {}}
        c_connector.process_fi_invoice_result((rq_id, hoadon))
        return {"rq_id": rq_id}
    

@app.task(name="process_sap_invoice")
def process_sap_invoice(rq_id, list_url):
    from celery_worker.client_connector_fi import CeleryConnector
    from common.process_pdf import compile_output
    
    print(list_url)
    c_connector = CeleryConnector()
    try:
        result = compile_output(list_url)
        hoadon = {"status": 200, "content": result, "message": "Success"}
        c_connector.process_sap_invoice_result((rq_id, hoadon))
        return {"rq_id": rq_id}
    except Exception as e:
        print(e)
        hoadon = {"status": 404, "content": {}}
        c_connector.process_sap_invoice_result((rq_id, hoadon))
        return {"rq_id": rq_id}
    
@app.task(name="process_manulife_invoice")
def process_manulife_invoice(rq_id, list_url):
    from celery_worker.client_connector_fi import CeleryConnector
    from common.process_pdf import compile_output_manulife
    # TODO: simply returning 200 and 404 doesn't make any sense
    c_connector = CeleryConnector()
    try:
        result = compile_output_manulife(list_url)
        hoadon = {"status": 200, "content": result, "message": "Success"}
        print(hoadon)
        c_connector.process_manulife_invoice_result((rq_id, hoadon))
        return {"rq_id": rq_id}
    except Exception as e:
        print(e)
        hoadon = {"status": 404, "content": {}}
        c_connector.process_manulife_invoice_result((rq_id, hoadon))
        return {"rq_id": rq_id}

@app.task(name="process_sbt_invoice")
def process_sbt_invoice(rq_id, list_url, metadata):
    # TODO: simply returning 200 and 404 doesn't make any sense
    c_connector = CeleryConnector()
    try:
        result = compile_output_sbt(list_url)
        metadata['ai_inference_profile'] = result.pop("inference_profile")
        hoadon = {"status": 200, "content": result, "message": "Success"}
        print(hoadon)
        c_connector.process_sbt_invoice_result((rq_id, hoadon, metadata))
        return {"rq_id": rq_id}
    except Exception as e:
        print(f"[ERROR]: Failed to extract invoice: {e}")
        print(e)
        hoadon = {"status": 404, "content": {}}
        c_connector.process_sbt_invoice_result((rq_id, hoadon, metadata))
        return {"rq_id": rq_id}