sqlite3_file_control

Author: hashbrowncipher

Lib/test/test_sqlite3/test_dbapi.py

@@ -25,6 +25,7 @@
 import sqlite3 as sqlite
 import subprocess
 import sys
+import tempfile
 import threading
 import unittest
 import urllib.parse
@@ -735,6 +736,24 @@ def test_database_keyword(self):
         with contextlib.closing(sqlite.connect(database=":memory:")) as cx:
             self.assertEqual(type(cx), sqlite.Connection)
 
+    def test_wal_preservation(self):
+        with tempfile.TemporaryDirectory() as dirname:
+            path = os.path.join(dirname, "db.sqlite")
+            with contextlib.closing(sqlite.connect(path)) as cx:
+                cx.set_file_control(sqlite.SQLITE_FCNTL_PERSIST_WAL, 1)
+                cu = cx.cursor()
+                cu.execute("PRAGMA journal_mode = WAL")
+                cu.execute("CREATE TABLE foo (id int)")
+                cu.execute("INSERT INTO foo (id) VALUES (1)")
+                self.assertTrue(os.path.exists(path + "-wal"))
+            self.assertTrue(os.path.exists(path + "-wal"))
+
+            with contextlib.closing(sqlite.connect(path)) as cx:
+                cu = cx.cursor()
+                self.assertTrue(os.path.exists(path + "-wal"))
+                cu.execute("INSERT INTO foo (id) VALUES (2)")
+            self.assertFalse(os.path.exists(path + "-wal"))
+
 
 class CursorTests(unittest.TestCase):
     def setUp(self):
@@ -1863,6 +1882,8 @@ def test_on_conflict_replace(self):
         self.assertEqual(self.cu.fetchall(), [('Very different data!', 'foo')])
 
 
+
+
 @requires_subprocess()
 class MultiprocessTests(unittest.TestCase):
     CONNECTION_TIMEOUT = 0  # Disable the busy timeout.

Modules/_sqlite/clinic/connection.c.h

@@ -2173,6 +2173,50 @@ pysqlite_connection_create_collation_impl(pysqlite_Connection *self,
     Py_RETURN_NONE;
 }
 
+/*[clinic input]
+_sqlite3.Connection.set_file_control as pysqlite_connection_set_file_control
+
+    op: int
+      a SQLITE_FCNTL_ constant
+    arg: long
+      argument to pass
+    /
+    dbname: str = NULL
+      database name
+
+Invoke a file control method on the database.
+[clinic start generated code]*/
+
+static PyObject *
+pysqlite_connection_set_file_control_impl(pysqlite_Connection *self, int op,
+                                          long arg, const char *dbname)
+/*[clinic end generated code: output=d9d2d311892893b6 input=0253798d9514fea2]*/
+{
+    int rc;
+    long val = arg;
+
+    if (!pysqlite_check_thread(self) || !pysqlite_check_connection(self)) {
+        return NULL;
+    }
+
+    else {
+        PyErr_SetString(PyExc_TypeError, "arg must be an integer");
+        return NULL;
+    }
+
+    Py_BEGIN_ALLOW_THREADS
+    rc = sqlite3_file_control(self->db, dbname, op, &val);
+    Py_END_ALLOW_THREADS
+
+    if (rc != SQLITE_OK) {
+        _pysqlite_seterror(self->state, self->db);
+        return NULL;
+    }
+
+    return PyLong_FromLong(val);
+}
+
+
 #ifdef PY_SQLITE_HAVE_SERIALIZE
 /*[clinic input]
 _sqlite3.Connection.serialize as serialize
@@ -2601,6 +2645,7 @@ static PyMethodDef connection_methods[] = {
     PYSQLITE_CONNECTION_SET_AUTHORIZER_METHODDEF
     PYSQLITE_CONNECTION_SET_PROGRESS_HANDLER_METHODDEF
     PYSQLITE_CONNECTION_SET_TRACE_CALLBACK_METHODDEF
+    PYSQLITE_CONNECTION_SET_FILE_CONTROL_METHODDEF
     SETLIMIT_METHODDEF
     GETLIMIT_METHODDEF
     SERIALIZE_METHODDEF

Modules/_sqlite/connection.c

@@ -514,6 +514,48 @@ add_integer_constants(PyObject *module) {
     ADD_INT(SQLITE_DBCONFIG_LEGACY_FILE_FORMAT);
     ADD_INT(SQLITE_DBCONFIG_TRUSTED_SCHEMA);
 #endif
+    ADD_INT(SQLITE_FCNTL_LOCKSTATE);
+    ADD_INT(SQLITE_FCNTL_GET_LOCKPROXYFILE);
+    ADD_INT(SQLITE_FCNTL_SET_LOCKPROXYFILE);
+    ADD_INT(SQLITE_FCNTL_LAST_ERRNO);
+    ADD_INT(SQLITE_FCNTL_SIZE_HINT);
+    ADD_INT(SQLITE_FCNTL_CHUNK_SIZE);
+    ADD_INT(SQLITE_FCNTL_FILE_POINTER);
+    ADD_INT(SQLITE_FCNTL_SYNC_OMITTED);
+    ADD_INT(SQLITE_FCNTL_WIN32_AV_RETRY);
+    ADD_INT(SQLITE_FCNTL_PERSIST_WAL);
+    ADD_INT(SQLITE_FCNTL_OVERWRITE);
+    ADD_INT(SQLITE_FCNTL_VFSNAME);
+    ADD_INT(SQLITE_FCNTL_POWERSAFE_OVERWRITE);
+    ADD_INT(SQLITE_FCNTL_PRAGMA);
+    ADD_INT(SQLITE_FCNTL_BUSYHANDLER);
+    ADD_INT(SQLITE_FCNTL_TEMPFILENAME);
+    ADD_INT(SQLITE_FCNTL_MMAP_SIZE);
+    ADD_INT(SQLITE_FCNTL_TRACE);
+    ADD_INT(SQLITE_FCNTL_HAS_MOVED);
+    ADD_INT(SQLITE_FCNTL_SYNC);
+    ADD_INT(SQLITE_FCNTL_COMMIT_PHASETWO);
+    ADD_INT(SQLITE_FCNTL_WIN32_SET_HANDLE);
+    ADD_INT(SQLITE_FCNTL_WAL_BLOCK);
+    ADD_INT(SQLITE_FCNTL_ZIPVFS);
+    ADD_INT(SQLITE_FCNTL_RBU);
+    ADD_INT(SQLITE_FCNTL_VFS_POINTER);
+    ADD_INT(SQLITE_FCNTL_JOURNAL_POINTER);
+    ADD_INT(SQLITE_FCNTL_WIN32_GET_HANDLE);
+    ADD_INT(SQLITE_FCNTL_PDB);
+    ADD_INT(SQLITE_FCNTL_BEGIN_ATOMIC_WRITE);
+    ADD_INT(SQLITE_FCNTL_COMMIT_ATOMIC_WRITE);
+    ADD_INT(SQLITE_FCNTL_ROLLBACK_ATOMIC_WRITE);
+    ADD_INT(SQLITE_FCNTL_LOCK_TIMEOUT);
+    ADD_INT(SQLITE_FCNTL_DATA_VERSION);
+    ADD_INT(SQLITE_FCNTL_SIZE_LIMIT);
+    ADD_INT(SQLITE_FCNTL_CKPT_DONE);
+    ADD_INT(SQLITE_FCNTL_RESERVE_BYTES);
+    ADD_INT(SQLITE_FCNTL_CKPT_START);
+    ADD_INT(SQLITE_FCNTL_EXTERNAL_READER);
+    ADD_INT(SQLITE_FCNTL_CKSM_FILE);
+    ADD_INT(SQLITE_FCNTL_RESET_CACHE);
+
 #undef ADD_INT
     return 0;
 }