File size: 836 Bytes
c481f8a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | 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
|