diff --git a/run.py b/run.py index 3c59c2c..b3d2d5f 100644 --- a/run.py +++ b/run.py @@ -1,28 +1,26 @@ -import os -import sys -from dotenv import load_dotenv - - -# 환경 변수 로드 -load_dotenv() - +import os, sys # 프로젝트 루트 기준 경로 추가 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'lib'))) -# 실행 모드 확인 -mode = os.getenv("MODE", "").strip().lower() +from conf import config -if mode == "biz": - from lib.biz_crawler import NaverReviewCollector - print("[INFO] 비즈니스 리뷰 수집기 실행") - collector = NaverReviewCollector() - collector.run() +def main(): + mode = os.getenv("MODE", "").strip().lower() -elif mode == "map": - from lib.naver_review_crawler import NaverMapReviewCollector - print("[INFO] 지도 리뷰 수집기 실행") - collector = NaverMapReviewCollector() - collector.run() + if mode == "biz": + from lib.biz_crawler import NaverReviewCollector + print("[INFO] 비즈니스 리뷰 수집기 실행") + collector = NaverReviewCollector() + collector.run() -else: - print("[ERROR] .env 파일에서 MODE 값을 설정해주세요. (biz 또는 map)") + elif mode == "map": + from lib.naver_review_crawler import NaverMapReviewCollector + print("[INFO] 지도 리뷰 수집기 실행") + collector = NaverMapReviewCollector() + collector.run() + + else: + print(f"[ERROR] .env 파일에서 MODE 값을 설정해주세요. (biz 또는 map) 현재값: '{mode}'") + +if __name__ == "__main__": + main() \ No newline at end of file