U
S��W � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d d lm Z d d
l m
Z G d d � d e�Zd
d� Zddd�Zdd� Zd dd�ZG dd� de�ZdS ) � N)�abort)�render_template)�request)�Database)� DoesNotExist)�Model)�Proxy)� SelectQuery)�connectc @ s. e Zd Zd dd�Zdd� Zdd� Zd d
� Zd S )
�PaginatedQuery�pageFc C sD || _ || _|| _t|t�r.|| _| jj| _n|| _| j�� | _d S �N) � paginate_by�page_var� check_bounds�
isinstancer �queryZ model_classZmodel�select)�self�query_or_modelr r r � r �/flask_utils.py�__init__ s
zPaginatedQuery.__init__c C s, t j�| j�}|r(|�� r(tdt|��S dS )N� )r �args�getr �isdigit�max�int)r Z curr_pager r r �get_page s zPaginatedQuery.get_pagec C s t t�t| j�� �| j ��S r
)r �mathZceil�floatr �countr �r r r r �get_page_count# s z PaginatedQuery.get_page_countc C s2 | j r | �� | �� kr td� | j�| �� | j�S �Ni� )r r r$ r r Zpaginater r# r r r �get_object_list& s z PaginatedQuery.get_object_listN)r F)�__name__�
__module__� __qualname__r r r$ r& r r r r r s �
r c G sD t | t�s| �� } z| j|� �� W S tk
r> td� Y nX d S r% )r r r �wherer r r )r r r r r �get_object_or_404, s
r+ � object_list� r Tc K s4 t ||||�}|�� ||< t| f||�� d�|��S )N)Z
paginationr )r r& r r )Z
template_namer Zcontext_variabler r r �kwargsZpaginated_queryr r r r, 4 s � ���c C s t js t jS dt jt jf S )Nz%s?%s)r Z query_string�pathr r r r �get_current_urlB s r0 �/c C s0 t j�d�rt jd S t j�d�r,t jd S | S )N�next)r r r Zform)�defaultr r r � get_next_urlG s
r4 c @ sZ e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Ze dd� �Z
dd� Z dd� Z dS )�FlaskDBNc C s( d | _ || _|| _|d k r$| �|� d S r
)�database�_app�_db�init_app)r �appr6 r r r r O s
zFlaskDB.__init__c C sf || _ | jd krFd|jkr&|jd }qLd|jkr<|jd }qLtd��n| j}| �||� | �|� d S )NZDATABASEZ DATABASE_URLzKMissing required configuration data for database: DATABASE or DATABASE_URL.)r7 r8 Zconfig�
ValueError�_load_database�_register_handlers)r r: Z
initial_dbr r r r9 V s
zFlaskDB.init_appc C sV t |t�r|}n"t |t�r*| �t|��}nt|�}t | jt�rL| j�|� n|| _d S r
)r r �dict�_load_from_config_dict�db_url_connectr6 r Z
initialize)r r: Z config_valuer6 r r r r< g s
zFlaskDB._load_databasec C s� z|� d�}|� d�}W n tk
r4 td��Y nX d|krP|�dd�\}}n
d| }}z.t|� tj| }t||�}t|t �s�t
�W n\ t k
r� td| ��Y n> t k
r� td| ��Y n t
k
r� td | ��Y nX ||f|�S )
N�name�enginez:DATABASE configuration must specify a `name` and `engine`.�.r �peeweezUnable to import %sz Database engine not found %sz5Database engine not a subclass of peewee.Database: %s)
�pop�KeyError� RuntimeError�rsplit�
__import__�sys�modules�getattr�
issubclassr �AssertionError� ImportError�AttributeError)r Z config_dictrA rB r/ �
class_name�moduleZdatabase_classr r r r? u s,
� z FlaskDB._load_from_config_dictc C s |� | j� |�| j� d S r
)Zbefore_request�
connect_dbZteardown_request�close_db)r r: r r r r= � s zFlaskDB._register_handlersc s* � j d krtd��G � fdd�dt�}|S )Nz Database must be initialized.c s e Zd ZG � fdd�d�ZdS )z*FlaskDB.get_model_class.<locals>.BaseModelc s e Zd Z� jZdS )z/FlaskDB.get_model_class.<locals>.BaseModel.MetaN)r' r( r) r6 r r# r r �Meta� s rU N)r' r( r) rU r r# r r � BaseModel� s rV )r6 rG r )r rV r r# r �get_model_class� s
zFlaskDB.get_model_classc C s@ | j d kr&t| dd �}|d kr&t� | _t| d�s:| �� | _| jS )Nr6 � _model_class)r7 rL r r6 �hasattrrW rX )r r6 r r r r � s
z
FlaskDB.Modelc C s | j �� d S r
)r6 r
r# r r r rS � s zFlaskDB.connect_dbc C s | j �� s| j �� d S r
)r6 Z is_closed�close)r �excr r r rT � s
zFlaskDB.close_db)NN)
r' r( r) r r9 r< r? r= rW �propertyr rS rT r r r r r5 N s
r5 )r, r- r T)r1 )r rJ Zflaskr r r rD r r r r r Zplayhouse.db_urlr
r@ �objectr r+ r, r0 r4 r5 r r r r �<module> s( �