[python] Requests 라이브러리를 이용해 세션 쿠키를 데이터베이스에 저장하는 방법은?

22 Nov 2023

python

먼저 Requests 라이브러리를 설치합니다. pip을 사용하여 설치할 수 있습니다.

pip install requests

세션을 관리하기 위해 Session 객체를 생성합니다.

import requests

session = requests.Session()

로그인 요청을 보내고 세션 쿠키를 받아옵니다.

login_data = {
    'username': 'your_username',
    'password': 'your_password'
}

response = session.post('https://example.com/login', data=login_data)

세션 쿠키를 데이터베이스에 저장합니다. 예를 들어, SQLite3 데이터베이스를 사용한다면 다음과 같은 코드를 사용할 수 있습니다.

import sqlite3

# SQLite3 데이터베이스에 연결
conn = sqlite3.connect('mydatabase.db')
cursor = conn.cursor()

# 세션 쿠키를 텍스트 형태로 변환하여 저장합니다.
cookie = session.cookies.get_dict()
cookie_text = str(cookie)

cursor.execute("INSERT INTO cookies (cookie_text) VALUES (?)", (cookie_text,))

# 변경 내용을 데이터베이스에 반영
conn.commit()

# 데이터베이스 연결 종료
conn.close()

필요할 때 세션을 복구하기 위해 데이터베이스에서 쿠키를 가져옵니다.

# SQLite3 데이터베이스에 연결
conn = sqlite3.connect('mydatabase.db')
cursor = conn.cursor()

# 데이터베이스에서 쿠키를 가져옵니다.
cursor.execute("SELECT cookie_text FROM cookies LIMIT 1")
cookie_text = cursor.fetchone()[0]

# 쿠키를 세션에 적용합니다.
session.cookies.update(eval(cookie_text))

# 데이터베이스 연결 종료
conn.close()

이제 세션 쿠키를 데이터베이스에 저장하고 필요할 때마다 복구할 수 있습니다. 데이터베이스에 저장된 쿠키를 가져오는 방법은 데이터베이스 종류에 따라 다를 수 있으니 참고해 주세요.