from os import getcwd import requests from bs4 import BeautifulSoup from src.configs.api_config import wenku8_username, wenku8_password # 登录页面的URL login_url = 'https://www.wenku8.net/login.php?jumpurl=http%3A%2F%2Fwww.wenku8.net%2Findex.php' index_url = 'https://www.wenku8.net/index.php' headers = { 'Accept-Language': 'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2', 'Connection': 'keep-alive', 'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:60.0) Gecko/20100101 Firefox/60.0', 'Upgrade-Insecure-Requests': '1' } # 登录表单数据 login_data = { 'username': wenku8_username, 'password': wenku8_password, 'usecookie': '0', 'action': 'login' } async def login(): # 发送登录请求 with requests.Session() as session: # 注意:这里使用了Session对象来保持会话状态 login_response = session.post(login_url, data=login_data, headers=headers) # 检查登录是否成功(根据实际需求调整) if login_response.status_code == 200: # 登录成功后,Session对象已经自动保存了Cookie # 可以直接使用该Session对象访问受保护的页面 print("登录成功!") # 获取 Cookie cookies = session.cookies # 保存 Cookie 到文件 with open('wenku8.cookie', 'w') as f: for cookie in cookies: f.write(f"{cookie.name}={cookie.value}; ") print("Cookie 保存成功!") else: print("登录失败,状态码:", login_response.status_code) async def get_books(): with open('wenku8.cookie', 'r') as f: cookie = f.read() headers1 = { 'Connection': 'keep-alive', 'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:60.0) Gecko/20100101 Firefox/60.0', 'Upgrade-Insecure-Requests': '1', 'Cookie': cookie } response = requests.get(index_url, headers=headers1) print(response) html = response.content.decode('gbk') soup = BeautifulSoup(html, 'html.parser') orders = soup.find_all(name='div', class_='block') head = """