R6�W� � @ 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�_modules�superr r )r �databaser! �kwargs)� __class__r r r 2 s zAPSWDatabase.__init__c C s | | j | <d S)N)r" )r �mod_name�mod_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_hooks�close)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)�step�finalize)�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 �cursor�sql�paramsr 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)�s�vr 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
]