Python新版本中能否将类名设为变量？如何简化分支查询代码？

1. Python中是否可将类名作为变量使用 #

无论Python新版本还是旧版本，类本身都是第一类对象，完全可以将类名赋值给变量，之后通过该变量直接访问类的属性与方法（包括你需要的objects.all()）。

示例代码：

class Course:
    class Objects:
        def all(self):
            return ["Course 1", "Course 2"]
    objects = Objects()

# 将类名赋值给变量
selected_class = Course
# 通过变量调用类的属性方法
print(selected_class.objects.all())  # 输出: ["Course 1", "Course 2"]

2. 简化多分支条件代码 #

你可以用字典映射的方式，把target值和对应的类做绑定，直接通过target取值后调用方法，彻底替代多分支判断逻辑。

原代码示例（假设结构）： #

def get_paginated_objects(target, start, end):
    if target == 'course':
        return Course.objects.all()[start:end]
    elif target == 'subject':
        return Subject.objects.all()[start:end]
    elif target == 'teacher':
        return Teacher.objects.all()[start:end]
    else:
        raise ValueError("无效的target值")

简化后的代码： #

# 预先定义target与类的映射关系
TARGET_CLASS_MAP = {
    'course': Course,
    'subject': Subject,
    'teacher': Teacher
}

def get_paginated_objects(target, start, end):
    try:
        # 直接通过target获取对应类，调用目标方法
        return TARGET_CLASS_MAP[target].objects.all()[start:end]
    except KeyError:
        raise ValueError(f"无效的target值: {target}")

这种方式不仅代码更简洁，后续新增其他类时，只需在TARGET_CLASS_MAP中添加键值对即可，无需修改函数核心逻辑，维护成本更低。

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