31 lines
968 B
Python
31 lines
968 B
Python
# common.py
|
|
import os, yaml
|
|
import logging
|
|
import time
|
|
import glob
|
|
|
|
def load_config():
|
|
"""
|
|
conf/config.yaml 파일을 UTF-8로 읽어 파이썬 dict로 반환
|
|
"""
|
|
path = os.path.join(os.path.dirname(__file__), '..', 'conf', 'config.yaml')
|
|
with open(path, encoding='utf-8') as f:
|
|
return yaml.safe_load(f)
|
|
|
|
def get_logger(name):
|
|
logger = logging.getLogger(name)
|
|
if not logger.handlers:
|
|
handler = logging.StreamHandler()
|
|
formatter = logging.Formatter('[%(asctime)s] %(levelname)s: %(message)s')
|
|
handler.setFormatter(formatter)
|
|
logger.addHandler(handler)
|
|
logger.setLevel(logging.INFO)
|
|
return logger
|
|
|
|
def wait_download_complete(download_dir, ext, timeout=60):
|
|
for _ in range(timeout):
|
|
files = glob.glob(os.path.join(download_dir, f"*.{ext.strip('.')}"))
|
|
if files:
|
|
return files[0]
|
|
time.sleep(1)
|
|
raise TimeoutError("다운로드 대기 시간 초과") |