Python OOP：如何合理拆分含复杂登录逻辑的Client类

优先选择组合模式，完全不建议用继承实现。 #

为什么继承不可行 #

继承的核心语义是is-a（是一个），如果让LogIn继承Client，等价于逻辑上认定「登录流程是一种客户端」，完全不符合实际业务关系，还会带来两个严重问题：

• 职责混乱：LogIn子类会继承到get_images、get_table_data这类和登录毫无关系的业务方法，类的边界完全模糊
• 扩展性差：后续如果要加短信登录、扫码登录、滑块验证等其他鉴权逻辑，会直接面对多继承的钻石问题，代码维护成本指数上升

组合方案的优化实现 #

你初步写的组合思路方向完全正确，只是有几个细节可以优化：requests.Session是可变对象，登录流程中对Session的Cookie、Header修改会直接作用在原对象上，不需要登录方法返回Session再重新赋值，也不需要每次调用登录都新建登录处理器实例。
优化后的代码如下：

登录逻辑独立模块（login.py） #

所有登录相关的步骤方法统一收敛到这个类里，内部辅助方法加单下划线标记为私有，不对外暴露：

class SiteLogInHandler:
    def __init__(self, session):
        self.session = session

    def _step1_get_landing_cookie(self, url, headers):
        self.session.get(url, headers=headers)

    def _step2_get_validate_cookie(self, url, headers):
        self.session.get(url, headers=headers)

    # 省略中间3-9步的前置Cookie收集方法，每个方法只做单一步骤
    def _step10_get_auth_pre_cookie(self, url, headers):
        self.session.get(url, headers=headers)

    def do_login(self, base_url, headers, username, password):
        # 按顺序执行所有前置步骤
        self._step1_get_landing_cookie(f"{base_url}/landing", headers)
        self._step2_get_validate_cookie(f"{base_url}/validate_init", headers)
        # 省略中间步骤调用
        self._step10_get_auth_pre_cookie(f"{base_url}/auth_pre", headers)
        # 提交账号密码完成登录
        resp = self.session.post(
            f"{base_url}/submit_login",
            data={"user": username, "password": password},
            headers=headers
        )
        # 遇到4xx/5xx状态码直接抛异常，比手动判断200更鲁棒
        resp.raise_for_status()
        print("Successfully logged in!")

主Client类实现 #

客户端初始化时就完成登录处理器的实例化，复用同一个Session对象：

import requests
from login import SiteLogInHandler

class Client:
    def __init__(self):
        self.session = requests.Session()
        # 把当前会话传给登录处理器，后续直接复用
        self._login_handler = SiteLogInHandler(self.session)

    def login(self, base_url, headers, username, password):
        self._login_handler.do_login(base_url, headers, username, password)

    def get_images(self, url, headers):
        # 业务图片获取逻辑，直接用self.session发请求即可，已自动携带登录态Cookie
        pass

    def get_table_data(self, url, headers):
        # 表格数据获取逻辑
        pass

额外优化建议 #

• 10个登录步骤的辅助函数不要用login_func1这种无意义命名，最好根据每个步骤的实际作用命名，后续排查问题不用挨个点进去看逻辑
• 提交账号密码的接口常规是POST请求，你原有代码里写的GET可以根据站点实际情况调整
• 后续如果要扩展其他鉴权方式，只要新增对应Handler类即可，不需要修改原有Client的业务代码，完全符合开闭原则

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