#现代 Web 爬虫技术：Session + Cookie 认证的模拟登录

import requests

# 定义常量（好习惯！别把账号密码写死在代码里，生产环境可以用环境变量）
LOGIN_URL = "https://login2.scrape.center/login"
INDEX_URL = "https://login2.scrape.center/"
USERNAME = "admin"
PASSWORD = "admin"

# 1. 单独发 POST 请求，禁止自动重定向（不然拿不到 Set-Cookie）
login_resp = requests.post(
    LOGIN_URL,
    data={"username": USERNAME, "password": PASSWORD},
    allow_redirects=False
)

# 2. 手动提取 Response 里的 Cookie
cookies = login_resp.cookies

# 3. 单独发 GET 请求，手动塞 Cookie
index_resp = requests.get(INDEX_URL, cookies=cookies)

# 简单验证：看看页面里有没有“欢迎回来，admin”（用 bs4 查）
from bs4 import BeautifulSoup
soup = BeautifulSoup(index_resp.text, "html.parser")
print(soup.find("h2", class_="mb-3"))  # 成功的话会打印 <h2>欢迎回来，admin</h2>

import requests
from bs4 import BeautifulSoup

LOGIN_URL = "https://login2.scrape.center/login"
INDEX_URL = "https://login2.scrape.center/"
USERNAME = "admin"
PASSWORD = "admin"

# 1. 创建一个 Session 会话对象
session = requests.Session()

# 2. （可选但推荐）先加个通用的 User-Agent，模拟真实浏览器，避免被反爬
session.headers.update({
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36"
})

# 3. 用 Session 发 POST 登录请求（不需要禁止重定向，Session 会自动处理）
login_resp = session.post(
    LOGIN_URL,
    data={"username": USERNAME, "password": PASSWORD}
)
# 简单验证登录状态码（POST 重定向后的首页状态码应该是 200）
print("登录后首页状态码：", login_resp.status_code)

# 4. 用同一个 Session 发后续所有请求（自动携带 Cookie，太爽了！）
index_resp = session.get(INDEX_URL)
soup = BeautifulSoup(index_resp.text, "html.parser")
print(soup.find("h2", class_="mb-3"))

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager
import requests
from bs4 import BeautifulSoup
import time

LOGIN_URL = "https://login2.scrape.center/login"
INDEX_URL = "https://login2.scrape.center/"
USERNAME = "admin"
PASSWORD = "admin"

# -------------------------- Selenium 模拟登录 + 拿 Cookie --------------------------
# 1. 初始化 Chrome 浏览器（webdriver-manager 自动下载对应版本的驱动）
driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()))
# （可选）窗口最大化，避免元素找不到
driver.maximize_window()

try:
    # 2. 打开登录页
    driver.get(LOGIN_URL)
    # （必加！新手常犯的错：页面还没加载完就找元素，会报错 NoSuchElementException）
    time.sleep(1)

    # 3. 模拟输入账号密码、点击登录
    driver.find_element(By.CSS_SELECTOR, 'input[name="username"]').send_keys(USERNAME)
    driver.find_element(By.CSS_SELECTOR, 'input[name="password"]').send_keys(PASSWORD)
    driver.find_element(By.CSS_SELECTOR, 'button[type="submit"]').click()
    # 等跳转完成
    time.sleep(2)

    # 4. 验证是否登录成功（如果当前 URL 是首页，说明成功）
    if driver.current_url == INDEX_URL:
        print("Selenium 模拟登录成功！")
        # 5. 获取登录后的所有 Cookie
        selenium_cookies = driver.get_cookies()
    else:
        print("Selenium 模拟登录失败！")
        selenium_cookies = []
finally:
    # 6. 关闭浏览器（用完就关，节省资源）
    driver.quit()

# -------------------------- Requests 拿 Cookie 继续爬取 --------------------------
if selenium_cookies:
    session = requests.Session()
    session.headers.update({
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36"
    })

    # 把 Selenium 的 Cookie 逐个塞到 requests 的 Session 里
    for cookie in selenium_cookies:
        session.cookies.set(
            name=cookie["name"],
            value=cookie["value"],
            domain=cookie["domain"],
            path=cookie["path"]
        )

    # 验证 + 爬取首页前 3 个电影标题
    index_resp = session.get(INDEX_URL)
    soup = BeautifulSoup(index_resp.text, "html.parser")
    print("首页前3个电影：")
    for i, movie in enumerate(soup.find_all("h5", class_="card-title"), 1):
        print(f"{i}. {movie.text.strip()}")