1、封装的数据库连接类和方法
import sqlite3
def db_select_info(sql):connect = sqlite3.connect("./db.sqlite3")cursor = connect.cursor()cursor.execute(sql)results = cursor.fetchall()connect.commit()connect.close()return results
2、封装的分页方法
from typing import List, Tuple, Dictdef paginate(data: List, page_size: int, page_number: int) -> Tuple[List, Dict]:"""对给定的数据进行分页处理。:param data: 要分页的数据列表:param page_size: 每页的大小:param page_number: 当前页码:return: 当前页的数据列表和分页信息字典"""# 计算总页数total_count = len(data)total_pages = (total_count + page_size - 1) // page_size # 向上取整# 检查页码是否有效，如果不是，则修正为有效的页码page_number = max(1, min(page_number, total_pages))# 计算当前页的起始索引和结束索引start_index = (page_number - 1) * page_sizeend_index = min(start_index + page_size, total_count)# 获取当前页的数据page_data = data[start_index:end_index]# 分页信息pagination_info = {'total_count': total_count,'total_pages': total_pages,'current_page': page_number,'has_next': page_number < total_pages,'has_prev': page_number > 1,}return page_data, pagination_info# 使用示例
data_list = [f'item_{i}' for i in range(1, 51)] # 假设有50条数据
page_size = 10 # 每页10条数据
page_number = 3 # 请求第3页page_data, pagination_info = paginate(data_list, page_size, page_number)
print(f"当前页数据: {page_data}")
print(f"分页信息: {pagination_info}")
3、接口业务类
class TeacherListView(viewsets.ModelViewSet):serializer_class = TeacherSerializerauthentication_classes = [MyAuthentication, ]# Args:# request(HttpRequest): 请求对象。# search_field(str): 要搜索的字段名# search_term(str): 搜索词def list(self,request,*args, **kwargs):try:search_term = self.request.query_params.get('search_term')search_term=eval(search_term)# 从查询参数中获取搜索词if search_term:select_sql=(f"select * from Teacher where name like '%{search_term}%' or source like '%{search_term}%' "f"or unit_name like '%{search_term}%' or branch like '%{search_term}%' or post like '%{search_term}'or phone_num like '%{search_term}%'")print(select_sql)query_data = db_select_info(select_sql)# 分页参数page_size_str = request.query_params.get('size', '10')page_size = int(page_size_str) if page_size_str.isdigit() else 10 # 默认为10page_number_str = request.query_params.get('page', '1')page_number = int(page_number_str) if page_number_str.isdigit() else 1 # 默认为1# 分页处理page_data, pagination_info = paginate(query_data, page_size, page_number)# 返回包含数据和分页信息的JSON响应response_data = {'data': page_data,'pagination': pagination_info,}return Response(res200(REQUEST_GET_SUCCESS, response_data))except Exception as e:print(str(e))return Response(res400(str(e)))
