| import os | |
| from loguru import logger | |
| from dotenv import load_dotenv | |
| from xhs_utils.session_manager import SessionManager | |
| def load_env(): | |
| load_dotenv() | |
| cookies_str = os.getenv('COOKIES') | |
| return cookies_str | |
| def init(): | |
| media_base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../datas/media_datas')) | |
| excel_base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../datas/excel_datas')) | |
| for base_path in [media_base_path, excel_base_path]: | |
| if not os.path.exists(base_path): | |
| os.makedirs(base_path) | |
| logger.info(f'创建目录 {base_path}') | |
| sm = SessionManager() | |
| cookies_str = sm.load_cookies(mode="both") or load_env() | |
| base_path = { | |
| 'media': media_base_path, | |
| 'excel': excel_base_path, | |
| } | |
| return cookies_str, base_path | |