Чистое резервное копирование Python базы данных SQLite3 в памяти на диск

Question

Без установки дополнительных модулей, как я могу использовать API резервного копирования SQLite для резервного копирования базы данных в памяти в базу данных на диске? Мне удалось успешно выполнить резервное копирование с диска на диск, но передача уже существующего подключения в памяти к sqlite3_backup_init Функция, кажется, проблема.

Мой пример с игрушкой, адаптированный с https://gist.github.com/achimnol/3021995 и сокращенный до минимума, выглядит следующим образом:

import sqlite3
import ctypes

# Create a junk in-memory database
sourceconn = sqlite3.connect(':memory:')
cursor = sourceconn.cursor()
cursor.execute('''CREATE TABLE stocks
             (date text, trans text, symbol text, qty real, price real)''')
cursor.execute("INSERT INTO stocks VALUES ('2006-01-05','BUY','RHAT',100,35.14)")
sourceconn.commit()

target = r'C:\data\sqlite\target.db'
dllpath = u'C:\\Python27\DLLs\\sqlite3.dll'

# Constants from the SQLite 3 API defining various return codes of state.
SQLITE_OK = 0
SQLITE_ERROR = 1
SQLITE_BUSY = 5
SQLITE_LOCKED = 6
SQLITE_OPEN_READONLY = 1
SQLITE_OPEN_READWRITE = 2
SQLITE_OPEN_CREATE = 4

# Tweakable variables
pagestocopy = 20
millisecondstosleep = 100

# dllpath = ctypes.util.find_library('sqlite3') # I had trouble with this on Windows
sqlitedll = ctypes.CDLL(dllpath)
sqlitedll.sqlite3_backup_init.restype = ctypes.c_void_p

# Setup some ctypes
p_src_db = ctypes.c_void_p(None)
p_dst_db = ctypes.c_void_p(None)
null_ptr = ctypes.c_void_p(None)

# Check to see if the first argument (source database) can be opened for reading.
# ret = sqlitedll.sqlite3_open_v2(sourceconn, ctypes.byref(p_src_db), SQLITE_OPEN_READONLY, null_ptr)
#assert ret == SQLITE_OK
#assert p_src_db.value is not None

# Check to see if the second argument (target database) can be opened for writing.
ret = sqlitedll.sqlite3_open_v2(target, ctypes.byref(p_dst_db), SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, null_ptr)
assert ret == SQLITE_OK
assert p_dst_db.value is not None

# Start a backup.
print 'Starting backup to SQLite database "%s" to SQLite database "%s" ...' % (sourceconn, target)
p_backup = sqlitedll.sqlite3_backup_init(p_dst_db, 'main', sourceconn, 'main')
print '    Backup handler: {0:#08x}'.format(p_backup)
assert p_backup is not None

# Step through a backup.
while True:
    ret = sqlitedll.sqlite3_backup_step(p_backup, pagestocopy)
    remaining = sqlitedll.sqlite3_backup_remaining(p_backup)
    pagecount = sqlitedll.sqlite3_backup_pagecount(p_backup)
    print '    Backup in progress: {0:.2f}%'.format((pagecount - remaining) / float(pagecount) * 100)
    if remaining == 0:
        break
    if ret in (SQLITE_OK, SQLITE_BUSY, SQLITE_LOCKED):
        sqlitedll.sqlite3_sleep(millisecondstosleep)

# Finish the bakcup
sqlitedll.sqlite3_backup_finish(p_backup)

# Close database connections
sqlitedll.sqlite3_close(p_dst_db)
sqlitedll.sqlite3_close(p_src_db)

Я получаю ошибку ctypes.ArgumentError: argument 3: <type 'exceptions.TypeError'>: Don't know how to convert parameter 3 по строке 49 (p_backup = sqlitedll.sqlite3_backup_init(p_dst_db, 'main', sourceconn, 'main')). Каким-то образом мне нужно передать ссылку на базу данных в памяти этой функции sqlite3_backup_init.

Я не знаю достаточно C, чтобы понять специфику самого API.

Установка: Windows 7, ActiveState Python 2.7

Answer 1

Похоже, что в Python 3.7 эта функциональность доступна в стандартной библиотеке. Вот несколько примеров, скопированных прямо из официальных документов:

Пример 1, скопируйте существующую базу данных в другую:

import sqlite3

def progress(status, remaining, total):
    print(f'Copied {total-remaining} of {total} pages...')

con = sqlite3.connect('existing_db.db')
bck = sqlite3.connect('backup.db')
with bck:
    con.backup(bck, pages=1, progress=progress)
bck.close()
con.close()

Пример 2, скопируйте существующую базу данных во временную копию:

import sqlite3

source = sqlite3.connect('existing_db.db')
dest = sqlite3.connect(':memory:')
source.backup(dest)

Чтобы ответить на ваш конкретный вопрос о резервном копировании базы данных в памяти на диск, похоже, что это работает. Вот быстрый сценарий с использованием стандартной библиотекиbackup метод:

import sqlite3


source = sqlite3.connect(':memory:')
dest = sqlite3.connect('backup.db')

c = source.cursor()
c.execute("CREATE TABLE test(id INTEGER PRIMARY KEY, msg TEXT);")
c.execute("INSERT INTO test VALUES (?, ?);", (1, "Hello World!"))
source.commit()

source.backup(dest)

dest.close()
source.close()

И backup.db база данных может быть загружена в sqlite3 и осмотрели:

$ sqlite3 backup.db
SQLite version 3.24.0 2018-06-04 14:10:15
Enter ".help" for usage hints.
sqlite> .schema
CREATE TABLE test(id INTEGER PRIMARY KEY, msg TEXT);
sqlite> SELECT * FROM test;
1|Hello World!

Answer 2

Доступ к базе данных в памяти возможен только через созданную библиотеку SQLite (в данном случае встроенную в Python SQLite).

Питона sqlite3 Модуль не предоставляет доступ к API резервного копирования, поэтому невозможно скопировать базу данных в памяти.

Вам нужно будет установить дополнительный модуль или использовать базу данных на диске.

Answer 3

@cl не так. Я решил эту проблему, используя ctypes извлечь основной указатель C объекта подключения Python и создать небольшой пакет: https://pypi.org/project/sqlite3-backup/

Исходный код находится по адресу https://sissource.ethz.ch/schmittu/sqlite3_backup/blob/master/sqlite3_backup/backup.py

Answer 4

Хотя это не является строго решением ваших вопросов (поскольку оно не использует API резервного копирования), оно служит подходом с минимальными усилиями и хорошо работает для небольших баз данных в памяти.

import os
import sqlite3

database = sqlite3.connect(':memory:')

# fill the in memory db with your data here

dbfile = 'dbcopy.db'
if os.path.exists(dbfile):
    os.remove(dbfile) # remove last db dump

new_db = sqlite3.connect(dbfile)
c = new_db.cursor() 
c.executescript("\r\n".join(database.iterdump()))
new_db.close()