File: //opt/alt/python38/lib64/python3.8/site-packages/playhouse/__pycache__/apsw_ext.cpython-38.pyc
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 �E/opt/alt/python38/lib64/python3.8/site-packages/playhouse/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 )�apsw�
Connectionr% Zsetbusytimeout�_add_conn_hooks�close)r r( r) �connr r r �_connect= s
zAPSWDatabase._connectc s t t| ��|� | �|� d S r )r' r$ r3 �
_load_modules)r r5 r* r r r3 H s z APSWDatabase._add_conn_hooksc C s$ | j �� D ]\}}|�||� q
|S r )r& �itemsZ createmodule)r r5 r- r. r r r r7 L s zAPSWDatabase._load_modulesc s4 | j �� D ]$\}\� }� fdd�}|�||� q
d S )Nc s � � } | | j | jfS r )�step�finalize)�instance��klassr r �make_aggregateS s z5APSWDatabase._load_aggregates.<locals>.make_aggregate)Z _aggregatesr8 Zcreateaggregatefunction)r r5 �name�
num_paramsr> r r< r �_load_aggregatesQ s z APSWDatabase._load_aggregatesc C s$ | j �� D ]\}}|�||� q
d S r )Z _collationsr8 Zcreatecollation)r r5 r? �fnr r r �_load_collationsX s z APSWDatabase._load_collationsc C s* | j �� D ]\}\}}|�|||� q
d S r )Z
_functionsr8 Zcreatescalarfunction)r r5 r? rB r@ r r r �_load_functions\ s z APSWDatabase._load_functionsc C s$ |� d� | jD ]}|�|� qd S �NT)�enableloadextension� _extensions�
loadextension)r r5 � extensionr r r �_load_extensions` s
z APSWDatabase._load_extensionsc C s4 | j �|� | �� s0| �� }|�d� |�|� d S rE )rG �add� is_closed�get_connrF rH )r rI r5 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
�debug�exception_wrapper�
get_cursorrS )r rQ rR �require_commitrP r r r � execute_sqlp s
zAPSWDatabase.execute_sqlc C s |j jr|�� �� S d S r )�_meta�auto_increment�
getconnectionZlast_insert_rowid)r rP �modelr r r �last_insert_idw s zAPSWDatabase.last_insert_idc C s |� � �� S r )r[ Zchanges)r rP r r r �
rows_affected{ s zAPSWDatabase.rows_affectedr c C s | � � �d| � d S )Nz begin %s;�rV rO �r r r r r r ~ s zAPSWDatabase.beginc C s | � � �d� d S )Nzcommit;r_ r r r r �commit� s zAPSWDatabase.commitc C s | � � �d� d S )Nz rollback;r_ r r r r �rollback� s zAPSWDatabase.rollbackc C s
t | |�S r r r` r r r r � s zAPSWDatabase.transactionc C s
t | |�S r r )r �sidr r r r � s zAPSWDatabase.savepoint)N)NT)r )r )N)r! r" r# r r/ r0 r6 r3 r7 rA rC rD rJ rN rS rX r] r^ r ra rb 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 rh 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 rg r* r r rj � s zBooleanField.db_value)r! r" r# rj rd r r r* r r � s r c @ s e Zd ZeZdS )r N�r! r" r# rh rj r r r r r � s r c @ s e Zd ZeZdS )r
Nrk r r r r r
� s r
c @ s e Zd ZeZdS )r Nrk r r r r r � s r c @ s e Zd ZeZdS )r Nrk r r r r r � s r ) �__doc__r1 �peeweer 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$ rh r r r r �<module> s6 ]