Xפ 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 e Zd
Z ddZd ZddZ G d de ZdS ) N)abort)render_template)request)Database) DoesNotExist)Model)Proxy) SelectQuery)connectc 4 e Zd Z d
dZd Zd Zd Zd d ZdS ) PaginatedQuerypageNFc || _ || _ |pd | _ || _ t |t
r|| _ | j j | _ d S || _ | j | _ d S N) paginate_bypage_varr check_bounds
isinstancer
querymodelselect)selfquery_or_modelr r r r s u/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/playhouse/flask_utils.py__init__zPaginatedQuery.__init__ sm &
LD ( nk 2 2 -'DJ )DJJJ'DJ **,,DJJJ c | j | j S t j | j }|r1| r t
dt | S dS )N )r r argsgetr isdigitmaxint)r curr_pages r get_pagezPaginatedQuery.get_page s_ 9 9 L$$T]33 ***,, *q#i..)) )qr c t | d sRt t j t | j | j z | _ | j S )N _page_count) hasattrr# mathceilfloatr countr r' r s r get_page_countz PaginatedQuery.get_page_count' s^ t]++ ? "49dj&&(())D,<<$> $> ? ?D r c | j r9| | k rt d | j | | j S Ni )r r% r. r r paginater r- s r get_object_listz PaginatedQuery.get_object_list- sY 43F3F3H3H!H!H #JJJz""4==??D4DEEEr c t ||dz z
d }t ||z | dz }t t |||z
d }t t || d | S )N r )r" minlistrange)r r totalshowstartstops r get_page_rangez PaginatedQuery.get_page_range2 sr TTQY'!,,54<''!+Ctd{++Q//E%&&uu-...r )r NF)r3 )__name__
__module__ __qualname__r r% r. r2 r= r r r
r
sr JN # - - - - F F F
/ / / / / /r r
c t | t s| } | j | S # t
$ r t
d Y d S w xY wr0 )r r
r wherer r r )r r s r get_object_or_404rD < sr nk 2 2 1'..00#~#U+//111
c
s A A$#A$ object_list r Tc t ||||| }| ||< t | f|| d|S )N)r r r r )
paginationr )r
r2 r r% )
template_namer context_variabler r r r kwargspaginated_querys r rE rE D s{ %
! # # #O / > > @ @F "
%
%
'
' r c h t j s t j S t j dt j S )N?)r query_stringpathrA r r get_current_urlrQ T s, | lllG$8$8 99r /c t j d rt j d S t j d rt j d S | S )Nnext)r r r form)defaults r get_next_urlrW Y sQ | $|F## & ! ! $|F## Nr c b e Zd ZdZddedfdZd Zd Zd Zd Z d Z
e d Zd
Z d Z
dS ) FlaskDBa
Convenience wrapper for configuring a Peewee database for use with a Flask
application. Provides a base `Model` class and registers handlers to manage
the database connection during the request/response cycle.
Usage::
from flask import Flask
from peewee import *
from playhouse.flask_utils import FlaskDB
# The database can be specified using a database URL, or you can pass a
# Peewee database instance directly:
DATABASE = 'postgresql:///my_app'
DATABASE = PostgresqlDatabase('my_app')
# If we do not want connection-management on any views, we can specify
# the view names using FLASKDB_EXCLUDED_ROUTES. The db connection will
# not be opened/closed automatically when these views are requested:
FLASKDB_EXCLUDED_ROUTES = ('logout',)
app = Flask(__name__)
app.config.from_object(__name__)
# Now we can configure our FlaskDB:
flask_db = FlaskDB(app)
# Or use the "deferred initialization" pattern:
flask_db = FlaskDB()
flask_db.init_app(app)
# The `flask_db` provides a base Model-class for easily binding models
# to the configured database:
class User(flask_db.Model):
email = CharField()
Nc d | _ || _ || _ || _ |pd| _ || | d S d S )NrA )databasebase_model_class_app_db_excluded_routesinit_app)r appr[ model_classexcluded_routess r r zFlaskDB.__init__ sP
+ / 52 ? MM# ?r c 6 || _ | j =d|j v r|j d }n-d|j v r|j d }nt d | j }d|j v r|j d | _ | || | | d S )NDATABASE DATABASE_URLzKMissing required configuration data for database: DATABASE or DATABASE_URL.FLASKDB_EXCLUDED_ROUTES)r] r^ config
ValueErrorr_ _load_database_register_handlers)r ra
initial_dbs r r` zFlaskDB.init_app s 8 SZ'' Z
3
3:-- Z 7
"G H H H J $
2 2$'J/H$ID ! C,,, $$$$$r c > t |t r|}nGt |t r#| t | }nt | }t | j t r | j | d S || _ d S r )r r dict_load_from_config_dictdb_url_connectr[ r
initialize)r ra config_valuer[ s r rj zFlaskDB._load_database s lH - - 4#HH
d
+
+ 4224 3E3EFFHH &l33H dmU + + % M $ $X . . . . . $DMMMr c * | d }| d }n # t $ r t d w xY wd|v r| dd \ }}nd|}} t | t
j | }t || }t |t sJ nV# t $ r t d|z t $ r t d|z t $ r t d |z w xY w ||fi |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 ImportErrorAttributeErrorAssertionError)r config_dictrt ru rP
class_namemoduledatabase_classs r ro z FlaskDB._load_from_config_dict su 7 ??6**D __X..FF 7 7 7 6 7 7
7 7 &== %}}S! 4 4 D** '*D ? t [&F $VZ 8 8N nh77 7 77 7 ? ? ? 4v =>> > H H H = FGG G ? ? ? 57= > ? ?
? ? ~d22k222s *- A-AB6 6AD c n | | j | | j d S r )before_request
connect_dbteardown_requestclose_db)r ra s r rk zFlaskDB._register_handlers s4 4?+++ T]+++++r c ^ j t d G fdd j }|S )Nz Database must be initialized.c . e Zd Z G fdd ZdS )*FlaskDB.get_model_class..BaseModelc e Zd Z j ZdS )/FlaskDB.get_model_class..BaseModel.MetaN)r> r? r@ r[ r- s r Metar s =r r N)r> r? r@ r r- s r BaseModelr sG
)
)
)
)
)
)
)
)
)
)
)
)r r )r[ rz r\ )r r s ` r get_model_classzFlaskDB.get_model_class sY = >?? ? ) ) ) ) ) ) )- ) ) ) r c | j &t | dd }|t | _ t | d s| | _ | j S )Nr[ _model_class)r] r r r[ r( r r )r r[ s r r z
FlaskDB.Model s[ 9 tZ66H %
t^,, 7 $ 4 4 6 6D r c p | j rt j | j v rd S | j d S r )r_ r endpointr[ r r- s r r zFlaskDB.connect_db s< W%59N%N%N F
r c | j rt j | j v rd S | j s| j d S d S r )r_ r r r[ is_closedclose)r excs r r zFlaskDB.close_db s\ W%59N%N%N F}&&(( " M ! ! ! ! ! " "r )r> r? r@ __doc__r r r` rj ro rk r propertyr r rA r r rY rY ` s % %L $E!% % % %( % % % 3 3 38, , , ! ! X!
" " " " "r rY )rE rF r NT)rR )r) r} flaskr r r rw r r r r r
playhouse.db_urlr rp objectr
rD rE rQ rW rY rA r r r sq
! ! ! ! ! ! 6 6 6 6 6 6*/ */ */ */ */V */ */ */Z 8EIM : : :
Q" Q" Q" Q" Q"f Q" Q" Q" Q" Q"r