22 lines
779 B
Python
22 lines
779 B
Python
import json
|
|
import os
|
|
|
|
COOKIE_PATH = os.path.join(os.path.dirname(__file__), '..', 'conf', 'cookies.json')
|
|
|
|
def save_cookies(driver):
|
|
cookies = driver.get_cookies()
|
|
with open(COOKIE_PATH, 'w', encoding='utf-8') as f:
|
|
json.dump(cookies, f, indent=2)
|
|
|
|
def load_cookies(driver, url='https://asp.upsolution.co.kr/'):
|
|
if not os.path.exists(COOKIE_PATH):
|
|
raise FileNotFoundError("쿠키 파일이 존재하지 않습니다. 먼저 수동 로그인 후 쿠키를 저장해주세요.")
|
|
|
|
driver.get(url)
|
|
with open(COOKIE_PATH, 'r', encoding='utf-8') as f:
|
|
cookies = json.load(f)
|
|
for cookie in cookies:
|
|
# domain 필드는 문제가 될 수 있으므로 제거
|
|
cookie.pop('domain', None)
|
|
driver.add_cookie(cookie)
|