下面给出基于 Django 的重构版本，并逐条标注改动与收益。示例使用类视图（CBV）、Django ORM、分页、JSON/CSV 双输出、类型标注、结构化日志与异常处理等现代特性。
重构后的代码
from typing import Iterable, Iterator, List, Dict
import logging
import csv
import io
from pathlib import Path
from django.contrib.auth import get_user_model
from django.core.paginator import Paginator, EmptyPage
from django.http import JsonResponse, StreamingHttpResponse, HttpRequest, HttpResponse
from django.views import View
logger = logging.getLogger(name)
新式工具函数：类型标注与内置 sum，避免手写循环
def compute(x: Iterable[int]) -> int:
return sum(x)
class UserListView(View):
default_page_size: int = 20
max_page_size: int = 100
def get(self, request: HttpRequest) -> HttpResponse:
fmt = (request.GET.get('format') or 'json').lower()
page_str = request.GET.get('page', '1')
per_page_str = request.GET.get('per_page', str(self.default_page_size))
# 参数校验与默认值处理
try:
page = max(1, int(page_str))
except ValueError:
return JsonResponse({'error': 'Invalid page parameter'}, status=400)
try:
per_page = max(1, int(per_page_str))
per_page = min(per_page, self.max_page_size)
except ValueError:
per_page = self.default_page_size
if fmt not in {'json', 'csv'}:
return JsonResponse({'error': 'Invalid format parameter'}, status=400)
try:
User = get_user_model()
# ORM + values：仅取必要字段，避免加载不需要的列
qs = (
User.objects
.values('id', 'email', 'first_name', 'last_name', 'username')
.order_by('id')
)
paginator = Paginator(qs, per_page)
try:
page_obj = paginator.page(page)
except EmptyPage:
return JsonResponse(
{'error': 'Page out of range', 'total_pages': paginator.num_pages},
status=404
)
# 列表推导 + 轻量序列化
data: List[Dict] = [self._serialize_user(u) for u in page_obj.object_list]
logger.info(
'User list requested',
extra={'page': page, 'per_page': per_page, 'format': fmt, 'count': len(data)}
)
if fmt == 'csv':
# Streaming CSV，避免一次性拼接大字符串造成内存压力
filename = Path(f'users_page_{page}.csv').name
response = StreamingHttpResponse(
self._csv_row_generator(data),
content_type='text/csv; charset=utf-8'
)
response['Content-Disposition'] = f'attachment; filename="{filename}"'
return response
# 标准 JSON 响应
return JsonResponse(
{
'data': data,
'count': paginator.count,
'page': page_obj.number,
'total_pages': paginator.num_pages,
'per_page': per_page,
},
json_dumps_params={'ensure_ascii': False}
)
except Exception as exc:
# 结构化日志 + 统一错误响应
logger.exception('Failed to fetch user list', extra={'page': page, 'per_page': per_page, 'format': fmt})
return JsonResponse({'error': 'Internal server error'}, status=500)
@staticmethod
def _serialize_user(u: Dict) -> Dict:
full_name = f"{u.get('first_name', '')} {u.get('last_name', '')}".strip()
name = full_name or u.get('username', '')
return {'id': u['id'], 'name': name, 'email': u.get('email', '')}
@staticmethod
def _csv_row_generator(users: List[Dict]) -> Iterator[str]:
# 使用 csv.writer 保证正确的 CSV 转义；with 上下文管理器管理临时缓冲
yield 'id,name,email\n'
for u in users:
with io.StringIO() as s:
writer = csv.writer(s)
writer.writerow([u['id'], u['name'], u['email']])
yield s.getvalue()
更改点与带来的好处
- 将函数视图重构为类视图（CBV，UserListView）
- 好处：更易扩展（支持不同 HTTP 方法）、可复用性强，符合 Django 最佳实践。
- 使用 Django ORM 替代原生 SQL（get_user_model + values + order_by）
- 好处：防止 SQL 注入、统一数据访问、只取必要字段提升性能与内存占用。
- 使用 values/select_related/prefetch_related 的思想优化查询（此处 values 即足够）
- 好处：减少不必要的字段/关系加载；在有外键场景可继续扩展 select_related/prefetch_related。
- 引入 Paginator 分页，参数校验和默认值处理（page/per_page，范围限制）
- 好处：避免全表查询一次性返回；保证健壮性与可控的响应大小。
- 标准化响应：JsonResponse；CSV 使用 StreamingHttpResponse 流式输出
- 好处：正确的 Content-Type；流式 CSV 避免一次性构建大型字符串造成内存峰值。
- 类型标注与现代语法（类型注解、列表推导、f-strings、with 上下文管理器）
- 好处：提高可读性与静态检查效果；减少低级错误（资源泄露、拼接错误）。
- 统一文件/路径处理（pathlib 构建导出文件名）
- 好处：避免硬编码字符串；跨平台一致的路径/文件名处理方式。
- 结构化日志与异常处理（logger.info/exception，统一 400/404/500 返回）
- 好处：便于排查问题与监控；避免裸露 print 与不规范异常，提升可观测性。
- 安全与性能改进
- 好处：使用 ORM 防注入；分页与 values 减少 IO 与内存；列表推导提升速度；CSV 用 writer 确保正确转义。
- 可扩展的缓存（可选）
- 好处：热点页可以用 Django cache 缓存 JSON 输出（例如短期 60s），进一步降低数据库压力。
- 计算函数 compute 重写为 sum + 类型标注
测试建议（示例用例）
- 分页与参数校验
- 请求 /users/?page=1&per_page=20 返回 200，数据条数不超过 20，包含 count/page/total_pages。
- 请求 /users/?page=abc 返回 400。
- 请求 /users/?page=9999 返回 404，包含 total_pages。
- 格式切换与 CSV 导出
- 请求 /users/?format=csv 返回 200，Content-Type 为 text/csv；响应头包含 Content-Disposition，文件名含 page。
- CSV 内容首行为 id,name,email；字段正确转义（包含逗号、引号的姓名能正确输出）。
- 性能与选择字段
- 断言视图使用 values 只拉取 id/email/first_name/last_name/username，不访问多余字段。
- 错误处理与日志
- 模拟数据库异常，视图返回 500 且 logger 捕获 exception。
路由示例（urls.py）
from django.urls import path
from .views import UserListView
urlpatterns = [
path('users/', UserListView.as_view(), name='user-list'),
]
备注
- 如果项目中已有自定义 User 模型并包含 name 字段，可将 _serialize_user 中的 name 组装逻辑改为直接读取 u['name']。
- 如需 DRF 风格，可将视图改为继承 rest_framework.views.APIView，并使用 Response/renderer；思路与实现基本一致。
