o

�N�g0�@s�dZd
dl
Z
d
dlZd
dlZzd
dlZdZWney!


dZYn
wdZe�	e
�
Zddi
Zdd	�Z
ddd�
Zd
d�Zddd�
Zddd�
ZdS)a�
SQLite sdb Module

:maintainer:    SaltStack
:maturity:      New
:platform:      all

This module allows access to sqlite3 using an ``sdb://`` URI

Like all sdb modules, the sqlite3 module requires a configuration profile to
be configured in either the minion or master configuration file. This profile
requires very little. For example:

.. code-block:: yaml

    mysqlite:
      driver: sqlite3
      database: /tmp/sdb.sqlite
      table: sdb
      create_table: True

The ``driver`` refers to the sqlite3 module, ``database`` refers to the sqlite3
database file. ``table`` is the table within the db that will hold keys and
values (defaults to ``sdb``). The database and table will be created if they
do not exist.

Advanced Usage:
===============

Instead of a table name, it is possible to provide custom SQL statements to
create the table(s) and get and set values.

.. code-block:: yaml

    myadvanced
      driver: sqlite3
      database: /tmp/sdb-advanced.sqlite
    create_statements:
      - "CREATE TABLE advanced (a text, b text, c blob, d blob)"
      - "CREATE INDEX myidx ON advanced (a)"
    get_query: "SELECT d FROM advanced WHERE a=:key"
    set_query: "INSERT OR REPLACE INTO advanced (a, d) VALUES (:key, :value)"
�NTFZsdb�set_�setc
Cstsd
SdS)z,
    Only load if sqlite3 is available.
    FT)
�HAS_SQLITE3�rr�D/opt/saltstack/salt/lib/python3.10/site-packages/salt/sdb/sqlite3.py�__virtual__As

r�strictcCsn|�d
|
��
d
�
}|�d�
}|dkr-td|||dd�}t�|
�
}||�
\}}|�d|�}d|�dd�dS)	Nzutf-8�
r
zNUL-terminated utf-8�
zNUL not allowed�
"z"")�encode�decode�find�UnicodeEncodeError�codecs�lookup_error�replace)�
s�errorsZ	encodableZ	nul_index�errorZ
error_handlerZreplacement�
_rrr�_quoteJs







�


rc
Cs�|d
}
d}t�
|
�
}|��}|�d�
}|�dt�}t|d�
}t|�
}z*|r2|D]}|�|�

q)n|�dd�rM|�d|�d��

|�d	|�d
|�d��

Wn
tjyX


Yn
w|||fS)NZdatabaseZcreate_statements�tableZ_idxZcreate_tableTz
CREATE TABLE z (key text, value blob)zCREATE UNIQUE INDEX z ON z (key))�sqlite3�connectZcursor�get�
DEFAULT_TABLEr�executeZOperationalError)�profileZdbr�conn�curZstmts�idxZsqlrrr�_connect^s*












�

�

�
r"cCsZ|sd
St|�
\}}}t
tjj�|
�
�
}
|�dd|�d��}|�|||
d��
|��
dS)z)
    Set a key/value pair in sqlite3
    FZ	set_queryzINSERT OR REPLACE INTO z VALUES (:key, :value))�key�valueT)	r"�
memoryview�salt�utils�msgpackZpackbrrZcommit)r#r$rrr r�
qrrrrus






�

cCs`|
sd
St|
�
\}}}|
�
dd|�d��}|�|d|i
�}|��}|s%d
Stjjj|ddd�S)	z"
    Get a value from sqlite3
    NZ	get_queryzSELECT value FROM z WHERE key=:keyr#r
F)
�raw)r"rrZfetchoner&r'r(Zunpackb)r#rrr rr)�resrrrr�s







r)
r)
N)�__doc__r�loggingZsalt.utils.msgpackr&rr�ImportErrorr�	getLogger�__name__�logZ__func_alias__rrr"rrrrrr�<module>
s$,



�