本文探讨如何在python中通过重构`person`类及其相关的排序逻辑，实现代码的简洁性与功能完整性。重点介绍如何利用字典优化`__getitem__`方法中的属性访问，以及如何结合`getattr`动态处理排序键，并引入`functools.total_ordering`装饰器简化比较方法的实现，从而遵循dry原则，提升代码的可读性和可维护性。
在软件开发中，“Don't Repeat Yourself”（DRY）原则是提高代码质量和可维护性的核心指导思想。在处理数据类（如Person类）时，我们经常会遇到需要实现多种魔术方法（magic methods）和辅助函数的情况。本文将以一个Person类的示例为基础，详细讲解如何通过Python的特性来优化代码，使其更加简洁、高效，并易于扩展。
1. 简化类属性访问：优化 __getitem__ 方法
__getitem__ 魔术方法允许我们通过索引访问类的实例属性，使其行为类似于列表或字典。原始实现中，通过一系列 if-elif 语句来根据索引返回不同的属性，这种模式在属性增多时会变得冗长且难以维护。
原始 __getitem__ 方法示例：
class Person:
def __init__(self, name, age, height, weight):
self._name = name
self._age = age
self._height = height
self._weight = weight
def __getitem__(self, index):
if index == 0:
return self._name
elif index == 1:
return self._age
elif index == 2:
return self._weight
elif index == 3:
return self._height
else:
raise IndexError("Index out of range")优化方案：使用字典映射
立即学习“Python免费学习笔记（深入）”；
我们可以将索引与对应属性的映射关系存储在一个字典中，从而避免重复的 if-elif 结构。
class Person:
# ... (其他方法和属性定义)
def __getitem__(self, index):
"""
通过索引访问Person对象的属性。
使用字典映射索引和属性，提高代码简洁性和可维护性。
"""
attributes_map = {
0: self._name,
1: self._age,
2: self._weight,
3: self._height
}
if index in attributes_map:
return attributes_map[index]
raise IndexError("Index out of range")优化优势：
- 简洁性： 代码量显著减少，逻辑更清晰。
- 可维护性： 当需要添加或修改可索引的属性时，只需更新 attributes_map 字典，而无需修改多个 if-elif 分支。
- 可读性： 字典映射直观地展示了索引与属性的对应关系。
2. 动态化排序键选择：重构 create_persons_list 函数
在对 Person 对象列表进行排序时，根据不同的键（如姓名、年龄、身高、体重）进行排序是很常见的需求。原始实现中，通过 if-elif 链来判断 sort_key 并为 mergesort 函数提供不同的 lambda 表达式作为 key 参数，这种方式同样存在重复代码。
原始 create_persons_list 函数示例：
def create_persons_list(n=10, sort_key='height'):
person_objects = [...] # 创建Person对象列表
if sort_key == 'name':
return mergesort(person_objects, key=lambda x: x.get_name())
elif sort_key == 'age':
return mergesort(person_objects, key=lambda x: x.get_age())
elif sort_key == 'height':
return mergesort(person_objects, key=lambda x: x.get_height())
elif sort_key == 'weight':
return mergesort(person_objects, key=lambda x: x.get_weight())
else:
return mergesort(person_objects) # 默认行为，依赖Person.__lt__优化方案：利用字典和 getattr 动态调用方法
我们可以将 sort_key 与 Person 对象中对应的 getter 方法名进行映射。然后，使用内置的 getattr() 函数动态地获取并调用这些方法。
import numpy as np
import load_names
# from merge_sort import mergesort # 假设 mergesort 已导入
NAMES = load_names.load_names() # 假设 NAMES 已加载
class Person:
# ... (__init__, __repr__, __str__, __getitem__等方法)
# 原始的getter方法
def get_name(self):
return self._name
def get_age(self):
return self._age
def get_height(self):
return self._height
def get_weight(self):
return self._weight
# 为了保持原始的默认排序行为 (age, height, weight)
# 可以在类中添加一个方法，或者在排序函数中直接构建元组
def _get_default_sort_key_tuple(self):
return (self._age, self._height, self._weight)
def create_persons_list(n=10, sort_key='height'):
"""
创建Person对象列表并根据指定键排序。
利用字典映射和getattr动态选择排序键，精简代码。
"""
person_objects = [Person(np.random.choice(NAMES), np.random.randint(18, 101),
np.random.randint(150, 201), np.random.randint(45, 101))
for _ in range(n)]
key_method_map = {
'name': 'get_name',
'age': 'get_age',
'height': 'get_height',
'weight': 'get_weight'
}
# 检查sort_key是否在映射中
method_name = key_method_map.get(sort_key)
if method_name:
# 如果找到对应的getter方法名，则动态构建key函数
key_func = lambda p: getattr(p, method_name)()
else:
# 如果sort_key未指定或无效，则使用原始的默认排序逻辑
# 原始的Person.__lt__方法是基于 (age, height, weight) 进行比较的
key_func = lambda p: p._get_default_sort_key_tuple() # 假设Person类有此辅助方法
return mergesort(person_objects, key=key_func) 
