Django中如何从请求体直接读取SQLite数据库（无需临时文件）

首先咱们先理清你原代码报错的原因：sqlite3.connect()默认接收的是文件路径字符串，但request.body是包含null字节的二进制数据（SQLite文件格式开头就有null字符）。当你直接把二进制数据传给它时，Python会尝试将二进制转成字符串，而null字节会被当作字符串的结束标记，导致路径解析失败，抛出ValueError: embedded null byte。

下面给你两种无需临时文件的解决方案，可根据你的Python版本选择：

方案一：Python 3.11+（最简洁高效） #

从Python 3.11开始，sqlite3模块新增了Connection.deserialize()方法，能直接将二进制格式的SQLite数据库数据加载到内存数据库中，完美适配你的需求：

import sqlite3
from django.http import HttpResponse

def upload_sqlite(request):
    # 创建纯内存的SQLite连接
    conn = sqlite3.connect(":memory:")
    try:
        # 将请求体中的二进制数据库数据加载到内存数据库的main表空间
        conn.deserialize("main", request.body)
        
        # 正常操作数据库
        cursor = conn.cursor()
        cursor.execute("SELECT * FROM table")
        # 处理查询结果，比如获取所有数据
        results = cursor.fetchall()
        # 这里添加你的业务逻辑，比如返回结果或处理数据
        cursor.close()
        
        return HttpResponse(status=200)
    except sqlite3.Error as e:
        # 处理数据库错误，比如请求体不是有效SQLite文件
        return HttpResponse(f"Invalid SQLite database: {str(e)}", status=400)
    finally:
        # 确保连接关闭
        conn.close()

方案二：兼容Python 3.10及以下版本 #

如果你仍在使用低版本Python，无法使用deserialize()方法，可以通过SQLite的备份功能，将请求体数据导入内存数据库，全程在内存中操作，无需物理临时文件：

import io
import sqlite3
from django.http import HttpResponse

def upload_sqlite(request):
    # 将请求体包装成可读取的类文件对象
    db_stream = io.BytesIO(request.body)
    # 创建临时源连接（用于读取二进制数据）
    source_conn = sqlite3.connect(":memory:")
    # 创建目标内存连接（用于后续查询操作）
    target_conn = sqlite3.connect(":memory:")
    
    try:
        # 将类文件对象中的数据写入源连接，再备份到目标连接
        source_conn.execute("PRAGMA journal_mode=OFF")
        db_stream.seek(0)
        source_conn.backup(target_conn, pages=db_stream.read())
        
        # 通过target_conn操作数据库
        cursor = target_conn.cursor()
        cursor.execute("SELECT * FROM table")
        results = cursor.fetchall()
        # 添加你的业务逻辑
        cursor.close()
        
        return HttpResponse(status=200)
    except sqlite3.Error as e:
        return HttpResponse(f"Invalid SQLite database: {str(e)}", status=400)
    finally:
        source_conn.close()
        target_conn.close()

注意事项 #

• 确保请求体是完整的SQLite文件，没有被截断。如果请求过大，可能需要调整Django的DATA_UPLOAD_MAX_MEMORY_SIZE配置。
• 内存数据库的性能取决于数据库大小，如果数据库过大，会占用较多内存，需根据实际情况评估。

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