39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import logging
|
|
|
|
import convert_obj_to_md
|
|
from redis_client import redis_client
|
|
from rq import Worker
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def process_task(task):
|
|
"""
|
|
태스크 정보를 기반으로 파일 확장자에 맞는 변환 함수를 호출합니다.
|
|
"""
|
|
file_ext = task.get("extension")
|
|
input_path = task.get("input_path")
|
|
output_path = task.get("output_path")
|
|
|
|
logger.info(f"작업 처리 중: {task}")
|
|
|
|
try:
|
|
if file_ext == "hwp":
|
|
convert_obj_to_md.convert_hwp_to_md(input_path, output_path)
|
|
else:
|
|
convert_obj_to_md.convert_to_md(input_path, output_path)
|
|
logger.info(f"변환 완료: {task}")
|
|
return {"status": "success", "output_path": output_path}
|
|
|
|
except Exception as e:
|
|
logger.error(f"작업 처리 중 에러: {e}", exc_info=True)
|
|
return {"status": "error", "message": str(e)}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
listen = ["task_queue1"]
|
|
|
|
worker = Worker(listen, connection=redis_client) # 최신 방식
|
|
worker.work()
|