R6W @ s d Z d d l Z d d l Td d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m
Z d d
l m Z
d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z Gd d d e Z Gd d d e Z d d Z Gd d d e Z Gd d d e Z Gd d d e Z Gd! d" d" e Z
Gd# d$ d$ e
Z d S)%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 Z d Z d d d Z d d Z d S)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 | j d S)N)r beginr )r r r r _begin- s ztransaction._beginN)__name__
__module__ __qualname__r r r r r r r
( s r
c s e Z d Z d f d d Z d d Z d d Z d d Z f d
d Z d d
Z d d Z d d Z
d d Z d d Z d d Z
d d Z d d d d Z d d Z d d Z d! d" d# Z d$ d% Z d& d' Z d! d( d) Z d d* d+ Z S), APSWDatabaseNc s/ | | _ i | _ t t | j | | d S)N)timeout_modulessuperr r )r databaser! kwargs) __class__r r r 2 s zAPSWDatabase.__init__c C s | | j | <d S)N)r" )r mod_namemod_instr r r register_module7 s z APSWDatabase.register_modulec C s | j | =d S)N)r" )r r' r r r unregister_module: s z APSWDatabase.unregister_modulec K s^ t j | | } | j d k r1 | j | j y | j | Wn | j Yn X| S)N)apsw
Connectionr! Zsetbusytimeout_add_conn_hooksclose)r r$ r% connr r r _connect= s
zAPSWDatabase._connectc s' t t | j | | j | d S)N)r# r r-
_load_modules)r r/ )r& r r r- H s z APSWDatabase._add_conn_hooksc C s4 x- | j j D] \ } } | j | | q W| S)N)r" itemsZ createmodule)r r/ r' r( r r r r1 L s zAPSWDatabase._load_modulesc sL xE | j j D]4 \ } \ } f d d } | j | | q Wd S)Nc s } | | j | j f S)N)stepfinalize)instance)klassr r make_aggregateS s z5APSWDatabase._load_aggregates.<locals>.make_aggregate)Z _aggregatesr2 Zcreateaggregatefunction)r r/ name
num_paramsr7 r )r6 r _load_aggregatesQ s "z APSWDatabase._load_aggregatesc C s4 x- | j j D] \ } } | j | | q Wd S)N)Z _collationsr2 Zcreatecollation)r r/ r8 fnr r r _load_collationsX s z APSWDatabase._load_collationsc C s= x6 | j j D]% \ } \ } } | j | | | q Wd S)N)Z
_functionsr2 Zcreatescalarfunction)r r/ r8 r; r9 r r r _load_functions\ s "z APSWDatabase._load_functionsc C s2 | j d x | j D] } | j | q Wd S)NT)enableloadextension _extensions
loadextension)r r/ extensionr r r _load_extensions` s
z APSWDatabase._load_extensionsc C sF | j j | | j sB | j } | j d | j | d S)NT)r? addZ is_closedZget_connr> r@ )r rA r/ r r r load_extensione s
zAPSWDatabase.load_extensionc C s | j | | p f | S)N)execute)r cursorsqlparamsr r r _execute_sqll s zAPSWDatabase._execute_sqlTc C sJ t j | | f | j $ | j } | j | | | Wd QRX| S)N)r
debugZexception_wrapper
get_cursorrI )r rG rH Zrequire_commitrF r r r execute_sqlp s
zAPSWDatabase.execute_sqlc C s | j j r | j j Sd S)N)Z_metaZauto_increment
getconnectionZlast_insert_rowid)r rF Zmodelr r r last_insert_idw s zAPSWDatabase.last_insert_idc C s | j j S)N)rM Zchanges)r rF r r r
rows_affected{ s zAPSWDatabase.rows_affectedr c C s | j j d | d S)Nz begin %s;)rK rE )r r r r r r ~ s zAPSWDatabase.beginc C s | j j d d S)Nzcommit;)rK rE )r r r r commit s zAPSWDatabase.commitc C s | j j d d S)Nz rollback;)rK rE )r r r r rollback s zAPSWDatabase.rollbackc C s
t | | S)N)r
)r r r r r r
s zAPSWDatabase.transactionc C s
t | | S)N)r )r Zsidr r r r s zAPSWDatabase.savepoint)r r r r r) r* r0 r- r1 r: r< r= rB rD rI rL rN rO r rP rQ r
r r r )r& r r 1 s( r c C s | d k r t | Sd S)N)str)svr r r nh s rU c s" e Z d Z f d d Z S)r c s8 t t | j | } | d k r4 | r0 d p3 d Sd S)N r )r# r db_value)r rT )r& r r rW s zBooleanField.db_value)r r r rW r r )r& r r s r c @ s e Z d Z e Z d S)r N)r r r rU rW r r r r r s r c @ s e Z d Z e Z d S)r N)r r r rU rW r r r r r s r c @ s e Z d Z e Z d S)r N)r r r rU rW r r r r r s r c @ s e Z d Z e Z d S)r N)r r r rU rW r r r r r s r ) __doc__r+ 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 rU r r r r <module> s6
]