22

✏️ Editing: sqlite_limits.py

"""
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import sqlite3

# Smallest SQLITE_MAX_VARIABLE_NUMBER any supported build ships with; safe
# floor when the real per-connection limit cannot be queried.
# https://www.sqlite.org/limits.html#max_variable_number
CONSERVATIVE_VARIABLE_LIMIT = 999

def sqlite_max_variable_number(database) -> int:
    # The effective cap is the linked build's compile/runtime
    # SQLITE_MAX_VARIABLE_NUMBER, not the library version — some builds run a
    # recent SQLite yet still cap at 999, so probe the live connection.
    # Best-effort: any probe failure must degrade to the safe floor, never
    # propagate, otherwise the probe itself would break the caller's write.
    try:
        limit = database.connection().getlimit(
            sqlite3.SQLITE_LIMIT_VARIABLE_NUMBER
        )
    except Exception:
        return CONSERVATIVE_VARIABLE_LIMIT
    return limit if limit > 0 else CONSERVATIVE_VARIABLE_LIMIT

MR SKB