Python多网站登录流转功能开发求助：新手遇登录切换瓶颈

嘿，作为刚上手Python的新手，碰到多站点登录流转的问题太正常了！我来帮你捋清楚怎么解决——核心问题其实是每个站点的登录状态要隔离，不能用同一个会话去处理多个站点，不然cookie和登录信息会混在一起，导致切换站点时状态混乱。下面给你一套可落地的解决方案，结合代码示例一步步来：

每个网站的登录状态（比如Cookie、会话标识）都是独立的，所以我们要给每个站点单独创建一个requests.Session()对象，确保它们的登录信息互不干扰。同时把每个站点的配置、登录、操作、下载逻辑拆分开，方便循环处理。

第一步：整理站点配置 #

先把所有需要处理的站点信息统一存成列表字典，后续循环遍历就行，方便维护：

import requests
from bs4 import BeautifulSoup  # 假设你用BeautifulSoup做网页解析

# 每个站点的配置，根据你的实际需求修改字段
sites = [
    {
        "name": "站点A",
        "login_url": "https://site-a.com/login",
        "username": "your_user_a",
        "password": "your_pass_a",
        "action_page": "https://site-a.com/do-something",
        "report_url": "https://site-a.com/download-report",
        "save_path": "./reports/site_a_report.pdf"
    },
    {
        "name": "站点B",
        "login_url": "https://site-b.com/login",
        "username": "your_user_b",
        "password": "your_pass_b",
        "action_page": "https://site-b.com/do-something",
        "report_url": "https://site-b.com/download-report",
        "save_path": "./reports/site_b_report.pdf"
    }
]

第二步：封装各环节函数 #

把登录、执行操作、下载报告拆成独立函数，代码更清晰，也方便单独调试：

1. 登录函数（处理单个站点登录） #

def login(session, site_config):
    """给指定站点执行登录操作"""
    # 先获取登录页面，提取可能需要的CSRF Token（很多站点都需要这个）
    login_page = session.get(site_config["login_url"])
    soup = BeautifulSoup(login_page.text, "html.parser")
    # 注意：这里的CSRF Token字段名要根据实际页面调整，比如有的叫csrf_token
    csrf_token = soup.find("input", {"name": "csrfmiddlewaretoken"})["value"]

    # 构造登录表单数据
    login_form = {
        "username": site_config["username"],
        "password": site_config["password"],
        "csrfmiddlewaretoken": csrf_token
    }

    # 发送登录请求
    response = session.post(site_config["login_url"], data=login_form)
    # 验证登录是否成功（比如检查页面是否包含"欢迎"这类关键词，或者响应状态码）
    if "欢迎回来" in response.text or response.status_code == 302:
        print(f"✅ 成功登录 {site_config['name']}")
        return True
    else:
        print(f"❌ {site_config['name']} 登录失败，请检查账号密码或表单字段")
        return False

2. 执行操作函数（登录后要做的事情） #

def run_actions(session, site_config):
    """执行站点的后续操作（比如导航、提交表单等）"""
    # 示例：访问操作页面，这里可以根据你的需求添加更多逻辑
    action_response = session.get(site_config["action_page"])
    # 比如你原来的网页抓取代码就可以放在这里
    # soup = BeautifulSoup(action_response.text, "html.parser")
    # ... 你的抓取/操作逻辑 ...

    if action_response.status_code == 200:
        print(f"✅ {site_config['name']} 操作执行完成")
        return True
    else:
        print(f"❌ {site_config['name']} 操作执行失败")
        return False

3. 下载报告函数 #

def download_report(session, site_config):
    """下载并保存站点报告"""
    # 用stream=True处理大文件下载
    report_response = session.get(site_config["report_url"], stream=True)
    if report_response.status_code == 200:
        with open(site_config["save_path"], "wb") as f:
            for chunk in report_response.iter_content(chunk_size=1024):
                f.write(chunk)
        print(f"✅ {site_config['name']} 报告已保存到：{site_config['save_path']}")
        return True
    else:
        print(f"❌ {site_config['name']} 报告下载失败")
        return False

第三步：主循环处理所有站点 #

最后用循环遍历每个站点，给每个站点创建独立会话，依次执行登录、操作、下载：

def main():
    for site in sites:
        print(f"\n===== 开始处理 {site['name']} =====")
        # 每个站点用独立的Session，自动隔离登录状态
        with requests.Session() as session:
            # 模拟浏览器请求头，避免被反爬拦截
            session.headers.update({
                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36"
            })
            # 登录失败就跳过后续操作
            if not login(session, site):
                continue
            # 操作失败也跳过下载
            if not run_actions(session, site):
                continue
            # 下载报告
            download_report(session, site)
        print(f"===== {site['name']} 处理完成 =====\n")

if __name__ == "__main__":
    main()

几个关键提醒 #

• 不同站点的登录表单字段可能不一样，比如有的用email代替username，CSRF Token的字段名也可能不同，一定要用浏览器开发者工具查看实际登录请求的参数。
• 如果碰到动态渲染的页面（比如用JavaScript加载内容），requests可能搞不定，这时候可以换成selenium，同样给每个站点创建独立的浏览器实例就行。
• 记得处理异常情况，比如网络超时、页面加载失败等，可以给函数加try-except块。

内容的提问来源于stack exchange，提问作者Paulo Costa