U
S��W� � @ 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� de�ZG dd� de�Zdd� Z G dd � d e�ZG d d � d e �ZG d d � d e�ZG 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 Zddd�Zdd� 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�_modules�superr$ r )r �databaser% �kwargs�� __class__r r r 2 s zAPSWDatabase.__init__c C s || j |< d S r �r& )r �mod_name�mod_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_hooks�close)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)�instance��klassr 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 �cursor�sql�paramsr 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)�s�vr r r �nh� s r^ c s e Zd Z� fdd�Z� 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 N�r! 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 ]