base.py 440 B

123456789101112131415161718192021
  1. from loguru import logger
  2. from magic_pdf.libs.drop_reason import DropReason
  3. def get_data_source(jso: dict):
  4. data_source = jso.get("data_source")
  5. if data_source is None:
  6. data_source = jso.get("file_source")
  7. return data_source
  8. def exception_handler(jso: dict, e):
  9. logger.exception(e)
  10. jso["need_drop"] = True
  11. jso["drop_reason"] = DropReason.Exception
  12. jso["exception"] = f"ERROR: {e}"
  13. return jso