Merge pull request #1 from whileman133/sqlite3-backup-connector

Implement SqliteBackupConnector

Author: whileman133

dbbackup/db/sqlite.py

@@ -1,7 +1,8 @@
+import sqlite3
 import warnings
 from io import BytesIO
 from shutil import copyfileobj
-from tempfile import SpooledTemporaryFile
+from tempfile import SpooledTemporaryFile, NamedTemporaryFile
 
 from django.db import IntegrityError, OperationalError
 
@@ -108,3 +109,38 @@ def restore_dump(self, dump):
         path = self.connection.settings_dict["NAME"]
         with open(path, "wb") as db_file:
             copyfileobj(dump, db_file)
+
+
+class SqliteBackupConnector(BaseDBConnector):
+    """
+    Create a dump using the SQLite backup command,
+    which is safe to execute when the database is
+    in use (unlike simply copying the database file).
+    Restore by copying the backup file over the
+    database file.
+    """
+    extension = 'sqlite3'
+
+    def _write_dump(self, fileobj):
+        pass
+
+    def create_dump(self):
+        if not self.connection.is_usable():
+            self.connection.connect()
+        # Important: ensure the connection to the DB
+        # has been established.
+        self.connection.ensure_connection()
+        src_db_connection = self.connection.connection
+
+        bkp_db_file = NamedTemporaryFile()
+        bkp_path = bkp_db_file.name
+        with sqlite3.connect(bkp_path) as bkp_db_connection:
+            src_db_connection.backup(bkp_db_connection)
+
+        bkp_db_file.seek(0)
+        return bkp_db_file
+
+    def restore_dump(self, dump):
+        path = self.connection.settings_dict["NAME"]
+        with open(path, "wb") as db_file:
+            copyfileobj(dump, db_file)

dbbackup/tests/test_connectors/test_sqlite.py

@@ -4,7 +4,7 @@
 from django.db import connection
 from django.test import TestCase
 
-from dbbackup.db.sqlite import SqliteConnector, SqliteCPConnector
+from dbbackup.db.sqlite import SqliteConnector, SqliteCPConnector, SqliteBackupConnector
 from dbbackup.tests.testapp.models import CharModel, TextModel
 
 
@@ -65,3 +65,17 @@ def test_restore_dump(self):
         connector = SqliteCPConnector()
         dump = connector.create_dump()
         connector.restore_dump(dump)
+
+
+class SqliteBackupConnectorTest(TestCase):
+    def test_create_dump(self):
+        connector = SqliteBackupConnector()
+        dump = connector.create_dump()
+        dump_content = dump.read()
+        self.assertTrue(dump_content)
+        self.assertTrue(dump_content.startswith(b"SQLite format 3"))
+
+    def test_restore_dump(self):
+        connector = SqliteBackupConnector()
+        dump = connector.create_dump()
+        connector.restore_dump(dump)

docs/changelog.rst

@@ -9,6 +9,7 @@ Unreleased
 * Drop support for end-of-life Python 3.7 and 3.8.
 * Drop support for end-of-life Django 3.2.
 * Drop support for ``DBBACKUP_STORAGE`` AND  ``DBBACKUP_STORAGE_OPTIONS`` settings, use Django's ``STORAGES['dbbackup']`` setting instead.
+* Implement new ``SqliteBackupConnector`` to backup SQLite3 databases using the ``.backup`` command (safe to execute on DBs with active connections).
 
 4.3.0 (2025-05-09)
 ----------

docs/databases.rst

@@ -128,6 +128,11 @@ simple raw copy of your database file, like a snapshot.
 
 In-memory database aren't dumpable with it.
 
+SqliteBackupConnector
+~~~~~~~~~~~~~~~~~
+
+The :class:`dbbackup.db.sqlite.SqliteBackupConnector` makes a copy of the SQLite database file using the ``.backup`` command, which is safe to execute with active connections to the database. It supports dumping in-memory databases by construction.
+
 MySQL
 -----