U
SW @ s\ d Z ddlZddlT ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm
Z dd
lm Z
dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ G dd deZG dd deZdd Z G dd d eZG d d d e ZG d d d eZG d!d" d"e Z
G d#d$ d$e
Z dS )%an
Peewee integration with APSW, "another python sqlite wrapper".
Project page: https://rogerbinns.github.io/apsw/
APSW is a really neat library that provides a thin wrapper on top of SQLite's
C interface.
Here are just a few reasons to use APSW, taken from the documentation:
* APSW gives all functionality of SQLite, including virtual tables, virtual
file system, blob i/o, backups and file control.
* Connections can be shared across threads without any additional locking.
* Transactions are managed explicitly by your code.
* APSW can handle nested transactions.
* Unicode is handled correctly.
* APSW is faster.
N)*)_sqlite_date_part)_sqlite_date_trunc)_sqlite_regexp) BooleanField) DateField)
DateTimeField) DecimalField)logger savepoint) TimeField transaction)SqliteExtDatabase)VirtualCharField) VirtualField)VirtualFloatField)VirtualIntegerField) VirtualModelc @ s e Zd ZdddZdd ZdS )r deferredc C s || _ || _d S N)db lock_type)selfr r r /apsw_ext.py__init__) s ztransaction.__init__c C s | j | j d S r )r beginr r r r r _begin- s ztransaction._beginN)r )__name__
__module__ __qualname__r r r r r r r ( s
r c s e Zd Zd, fdd Zdd Zdd Zdd Z fd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dd Zd-dd Zd d Zd d Zd.d"d#Zd$d% Zd&d' Zd/d(d)Zd0d*d+Z ZS )1 APSWDatabaseNc s$ || _ i | _tt| j|f| d S r )timeout_modulessuperr$ r )r databaser% kwargs __class__r r r 2 s zAPSWDatabase.__init__c C s || j |< d S r r& )r mod_namemod_instr r r register_module7 s z APSWDatabase.register_modulec C s | j |= d S r r, )r r- r r r unregister_module: s z APSWDatabase.unregister_modulec K sN t j|f|}| jd k r$|| j z| | W n | Y nX |S r )apswZ
Connectionr% Zsetbusytimeout_add_conn_hooksclose)r r( r) connr r r _connect= s
zAPSWDatabase._connectc s t t| | | | d S r )r' r$ r2
_load_modules)r r4 r* r r r2 H s z APSWDatabase._add_conn_hooksc C s$ | j D ]\}}||| q
|S r )r& itemsZ createmodule)r r4 r- r. r r r r6 L s zAPSWDatabase._load_modulesc s4 | j D ]$\}\ } fdd}||| q
d S )Nc s } | | j | jfS r )stepZfinalize)instanceklassr r make_aggregateS s z5APSWDatabase._load_aggregates.<locals>.make_aggregate)Z _aggregatesr7 Zcreateaggregatefunction)r r4 name
num_paramsr< r r: r _load_aggregatesQ s z APSWDatabase._load_aggregatesc C s$ | j D ]\}}||| q
d S r )Z _collationsr7 Zcreatecollation)r r4 r= fnr r r _load_collationsX s z APSWDatabase._load_collationsc C s* | j D ]\}\}}|||| q
d S r )Z
_functionsr7 Zcreatescalarfunction)r r4 r= r@ r> r r r _load_functions\ s z APSWDatabase._load_functionsc C s$ | d | jD ]}|| qd S NT)enableloadextension _extensions
loadextension)r r4 extensionr r r _load_extensions` s
z APSWDatabase._load_extensionsc C s4 | j | | s0| }|d || d S rC )rE addZ is_closedZget_connrD rF )r rG r4 r r r load_extensione s
zAPSWDatabase.load_extensionc C s | ||p d |S )Nr )execute)r cursorsqlparamsr r r _execute_sqll s zAPSWDatabase._execute_sqlTc C s< t ||f | | }| ||| W 5 Q R X |S r )r
debugZexception_wrapper
get_cursorrO )r rM rN Zrequire_commitrL r r r execute_sqlp s
zAPSWDatabase.execute_sqlc C s |j jr| S d S r )Z_metaZauto_increment
getconnectionZlast_insert_rowid)r rL Zmodelr r r last_insert_idw s zAPSWDatabase.last_insert_idc C s | S r )rS Zchanges)r rL r r r
rows_affected{ s zAPSWDatabase.rows_affectedr c C s | d| d S )Nz begin %s;rQ rK r r r r r r ~ s zAPSWDatabase.beginc C s | d d S )Nzcommit;rV r r r r commit s zAPSWDatabase.commitc C s | d d S )Nz rollback;rV r r r r rollback s zAPSWDatabase.rollbackc C s
t | |S r r rW r r r r s zAPSWDatabase.transactionc C s
t | |S r r )r Zsidr r r r s zAPSWDatabase.savepoint)N)NT)r )r )N)r! r" r# r r/ r0 r5 r2 r6 r? rA rB rH rJ rO rR rT rU r rX rY r r
__classcell__r r r* r r$ 1 s(
r$ c C s |d k rt |S d S r )str)svr r r nh s r^ c s e Zd Z fddZ ZS )r c s( t t| |}|d k r$|r dp"dS d S )N r )r' r db_value)r r] r* r r r` s zBooleanField.db_value)r! r" r# r` rZ r r r* r r s r c @ s e Zd ZeZdS )r Nr! r" r# r^ r` r r r r r s r c @ s e Zd ZeZdS )r
Nra r r r r r
s r
c @ s e Zd ZeZdS )r Nra r r r r r s r c @ s e Zd ZeZdS )r Nra r r r r r s r ) __doc__r1 Zpeeweer r r r Z
_BooleanFieldr Z
_DateFieldr Z_DateTimeFieldr Z
_DecimalFieldr
r r
Z
_TimeFieldr Z _transactionZplayhouse.sqlite_extr r r r r r r$ r^ r r r r <module> s6 ]