Python实现多数据模糊查询（代码简洁、易理解）

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, Dict


def 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_size
    end_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 = TeacherSerializer
    authentication_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  # 默认为10

            page_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)))