python - 如何将内存中的 SQLite 数据库复制到 Python 中的另一个内存中的 SQLite 数据库？

Question

我正在为 Django 编写一个以树状方式运行测试的测试套件。例如，测试用例 A 可能有 2 个结果，测试用例 B 可能有 1 个，测试用例 C 可能有 3 个。树看起来像这样

      X
     /
A-B-C-X
 \   \
  B   X
   \   X
    \ /
     C-X
      \
       X

对于上面树中的每条路径，数据库内容可能不同。因此，在每次 fork 时，我都在考虑创建数据库当前状态的内存副本，然后将该参数提供给下一个测试。

有人知道如何从本质上将内存数据库复制到另一个数据库，然后获取引用以传递该数据库吗？

谢谢!

Answer 1

好吧，在一次有趣的冒险之后，我想出了这个。

from django.db import connections
import sqlite3

# Create a Django database connection for our test database
connections.databases['test'] = {'NAME': ":memory:", 'ENGINE': "django.db.backends.sqlite3"}

# We assume that the database under the source_wrapper hasn't been created
source_wrapper = connections['default']  # put alias of source db here
target_wrapper = connections['test'] 

# Create the tables for the source database
source_wrapper.creation.create_test_db()

# Dump the database into a single text query
query = "".join(line for line in source_wrapper.connection.iterdump())

# Generate an in-memory sqlite connection
target_wrapper.connection = sqlite3.connect(":memory:")
target_wrapper.connection.executescript(query)

现在名为test 的数据库将是default 数据库的副本。使用 target_wrapper.connection 作为对新创建的数据库的引用。