R6�W � @ s d d l Z d d l 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 d
d � Z d d d d d d � Z d d � Z d d d � Z Gd d � d e � Z d S)� N)�abort)�render_template)�request)�Database)� DoesNotExist)�Model)�Proxy)� SelectQuery)�connectc @ sF e Z d Z d d d d � Z d d � Z d d � Z d d
� Z d S) �PaginatedQuery�pageFc C sd | | _ | | _ | | _ t | t � rE | | _ | j j | _ n | | _ | j 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 | j � } | r: | j � r: t d t | � � Sd S)N� )r �args�getr �isdigit�max�int)r Z curr_pager r r �get_page s zPaginatedQuery.get_pagec C s) t t j t | j j � � | j � � S)N)r �math�ceil�floatr �countr
)r r r r �get_page_count# s z PaginatedQuery.get_page_countc C sG | j r+ | j � | j � k r+ t d � | j j | j � | j � S)Ni� )r r r# r r Zpaginater
)r r r r �get_object_list& s !
z PaginatedQuery.get_object_listN)�__name__�
__module__� __qualname__r r r# r$ r r r r r s
r c G sU t | t � s | j � } y | j | � j � SWn t k
rP t d � Yn Xd S)Ni� )r r r Zwherer r r )r r r r r �get_object_or_404, s
r( � object_list� r Tc K sD t | | | | � } | j � | | <t | d | d | j � | �S)NZ
paginationr )r r$ r r )Z
template_namer Zcontext_variabler
r r �kwargsZpaginated_queryr r r r) 4 s c C s$ t j s t j Sd t j t j f S)Nz%s?%s)r Z query_string�pathr r r r �get_current_urlB s r- �/c C s> t j j d � r t j d St j j d � r: t j d S| S)N�next)r r r Zform)�defaultr r r � get_next_urlG s
r1 c @ s� e Z d Z d d d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z e d d � � Z
d d � Z d d � Z d S)�FlaskDBNc C s8 d | _ | | _ | | _ | d k r4 | j | � d S)N)�database�_app�_db�init_app)r �appr3 r r r r O s
zFlaskDB.__init__c C s� | | _ | j d k re d | j k r7 | j d } qn d | j k rV | j d } qn t d � � n | j } | j | | � | j | � d S)NZDATABASEZ DATABASE_URLzKMissing required configuration data for database: DATABASE or DATABASE_URL.)r4 r5 �config�
ValueError�_load_database�_register_handlers)r r7 Z
initial_dbr r r r6 V s zFlaskDB.init_appc C s} t | t � r | } n3 t | t � r? | j t | � � } n t | � } t | j t � rp | j j | � n | | _ d S)N)r r �dict�_load_from_config_dict�db_url_connectr3 r Z
initialize)r r7 Z config_valuer3 r r r r: g s zFlaskDB._load_databasec C s* y" | j d � } | j d � } Wn t k
rB t d � � Yn Xd | k rj | j d d � \ } } n
d | } } y? t | � t j | } t | | � } t | t � s� t
� Wnd t k
r� t d | � � YnC t k
r� t d | � � Yn" t
k
r t d | � � Yn X| | | � 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_dictr? r@ r, �
class_name�moduleZdatabase_classr r r r= u s*
z FlaskDB._load_from_config_dictc C s$ | j | j � | j | j � d S)N)Zbefore_request�
connect_dbZteardown_request�close_db)r r7 r r r r; � s zFlaskDB._register_handlersc s; � j d k r t d � � G� f d d � d t � } | S)Nz Database must be initialized.c s) e Z d Z G� f d d � d � Z d S)z*FlaskDB.get_model_class.<locals>.BaseModelc s e Z d Z � j Z d S)z/FlaskDB.get_model_class.<locals>.BaseModel.MetaN)r% r& r' r3 r )r r r �Meta� s rS N)r% r& r' rS r )r r r � BaseModel� s rT )r3 rE r )r rT r )r r �get_model_class� s zFlaskDB.get_model_classc C s^ | j d k r9 t | d d � } | d k r9 t � | _ t | d � sW | j � | _ | j S)Nr3 � _model_class)r4 rJ r r3 �hasattrrU rV )r r3 r r r r � s z
FlaskDB.Modelc C s | j j � d S)N)r3 r
)r r r r rQ � s zFlaskDB.connect_dbc C s | j j � s | j j � d S)N)r3 Z is_closed�close)r �excr r r rR � s zFlaskDB.close_db)
r% r& r' r r6 r: r= r; rU �propertyr rQ rR r r r r r2 N s
r2 )r rH Zflaskr r r rB r r r r r Zplayhouse.db_urlr
r> �objectr r( r) r- r1 r2 r r r r �<module> s"