R6W @ 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) PaginatedQuerypageFc C sd | | _ | | _ | | _ t | t rE | | _ | j j | _ n | | _ | j j | _ d S)N) paginate_bypage_var check_bounds
isinstancer queryZ model_classZmodelselect)selfquery_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 argsgetr isdigitmaxint)r Z curr_pager r r get_page s zPaginatedQuery.get_pagec C s) t t j t | j j | j S)N)r mathceilfloatr 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_stringpathr 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)Nnext)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_dbinit_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_dictdb_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)
Nnameenginez: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)
popKeyError RuntimeErrorrsplit
__import__sysmodulesgetattr
issubclassr AssertionError ImportErrorAttributeError)r Z config_dictr? r@ r,
class_namemoduleZdatabase_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_requestclose_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_closedclose)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"