%.i X d dl Z d dlZd dlZd dlmZ d dlmZ d dlT d dlmZ d dlmZ d dl
m Z d dl T dZ
d Zd
Ze g d ed d
geddgeddgiZ ed eD Zd Z d"dZd Zd Zd Zd Ze dk r.ej Z e Z e \ Z!Z" e#e" d k r/ ed e $ ej% d ee! Z&e"d Z'dZ(e!j( r$d e!j( ) d D Z(e!j* Z*e* ej+ , e' rdndZ* ee*e'fi e&Z-e!j. r)d!/ e d d Z0 ee0e- ee-e(e!j1 e!j2 e!j3 e!j4 dS dS )# N)getpass) OptionParser)*)print_) __version__)CockroachDatabasez.from peewee import *%s
database = %s('%s'%s)
zDclass BaseModel(Model):
class Meta:
database = database
zCclass UnknownField(object):
def __init__(self, *_, **__): pass
) cockroach cockroachdbcrdbmysqlmysqldbpostgres
postgresqlsqlitesqlite3c # > K | ]}t | D ]}||fV d S N)DATABASE_ALIASES).0keyvalues q/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/../../../bin/pwiz.py r ' sT 8 8!1#!68 8 CL 8 8 8 8 8 8 8 c D | t vrPt dd t z t j d | dd }t | } ||fi |}t || S )Nz)Unrecognized database, must be one of: %s, schema)r ) DATABASE_MAPerrjoinkeyssysexitpop Introspector
from_database)
database_type
database_namekwargsr
DatabaseClassdbs r make_introspectorr- + s L(( 7 IIl'')) * *
+ , , ,
ZZ$
'
'F /M } / / / /B % %b % 8 88r FTc B
||| }t j |rdt | z ndfz }t | st t t t d
fd
t }t j
D ]} | |vr|r| |v r
| | d S )N) table_names
include_views
snake_casez, **%s c |pg }j | }|D ]W}|j }||v r| |vrt d|z ||vr0||vr,| | || k r |||| gz Xt dj | z j | }st | }j | |D ]\ }}t |v |dk t dk |j j v g } | r>|j
rt dk rd|_
|j t u }
|
r(r&|j d|j pd} t d| z t d | z t d
t d t d | z | } | rft d
t | D ]8\ }
}t dd d |
D d|d 9t d j rt dj z t dk rMt fd|D }d d |D }t d|z nst d t d
| | d S )Nz # Possible reference cycle: %szclass %s(BaseModel):idr Fz - ?z # %sz %sr2 z class Meta:z table_name = '%s'z indexes = (z ((r c 3 K | ] }d |z V
dS z'%s'N )r fields r r z5print_models.._print_table..| s& A Afun A A A A A Ar z), z),z )z schema = '%s'c , g | ]\ }}|v |j S r8 )name)r colr9 primary_keyss r
z6print_models.._print_table.. s4 %( %( %( )sE,&&
&&&r c 3 K | ] }d |z V
dS r7 r8 )r pks r r z5print_models.._print_table.. s& E E E E E E E Er z& primary_key = CompositeKey(%s)z primary_key = False) foreign_keys
dest_tabler add model_namescolumnsitemssortedr= alllen field_class
pk_classes primary_key UnknownFieldr; raw_column_type get_fieldmulti_column_indexesr! r )tableseenaccumrA foreign_keydestrE r; columnskip
is_unknowndisprP fieldsuniquepk_field_namespk_listr= _print_tabledatabaseignore_unknown introspectorpreserve_orders @r r^ z"print_models.._print_tableI s ,U3 '
>
>K )D u}}e!3!37$>??? 4 D$5$55== LtUeW_===%(
II A A& A A AAAAAFF
; D ,|/BB C C C | q # %( %( %( %(-4%( %( %( ) )N ii E En E E EEEG ;gE F F F F 2 0 1 1 1r
r r )
introspectget_database_kwargsHEADERget_additional_importsget_database_class__name__get_database_namereprr
UNKNOWN_FIELD
BASE_MODELsetrG rD r" ) ra tablesrb r0 r` r1 db_kwargsheaderrR rQ r^ r_ s ` ` ` @@r print_modelsrq 6 sj &&65B2< ' > >H 0022I
++--''))2&&((&/74 ??""R 9 9F
6NNN }
:D D D D D D D D D DL 55D,113344 * *
*Uf__ UD)))* *r c b t j }t d t d| z t d| d z t d| z t dt
z t d d S )Nz# Code generated by:z# python -m pwiz %sz
# Date: %sz%B %d, %Y %I:%M%pz# Database: %sz# Peewee version: %sr2 )datetimenowr strftimeri peewee_version)cmd_linera timestamps r print_headerry s !%%''I
!"""
8 +,,,
<),,-@AA ABBB
l < < > > >???
!N 2333
2JJJJJr c t j d| z t j d S )Nz [91m%s[0m
)r# stderrwriteflush)msgs r r r s8 J*S0111Jr c t d } | j } |ddd |ddd d
|d d
d |dddd t t } |ddd|d |ddd |d d d d |d!d"d#dd$% |d&d'd(dd)% |d*d+dd,d-. |d/d0dd1d2. |d3d4dd5d6. | S )7Nz$usage: %prog [options] database_name)usagez-Hz--hosthost)rU z-pz--portportint)rU typez-uz--useruserz-Pz
--passwordpassword
store_true)rU actionz-ez--engineenginezVDatabase type, e.g. sqlite, mysql, postgresql or cockroachdb. Default is "postgresql".)rU choiceshelpz-sz--schemar z-tz--tablesrn zWOnly generate the specified tables. Multiple table names should be separated by commas.)rU r z-vz--viewsviewsz7Generate model classes for VIEWs in addition to tables.)rU r r z-iz--infoinfozIAdd database information and other metadata to top of the generated file.z-oz--preserve-orderrb z6Model definition column ordering matches source table.)r rU r z-Iz--ignore-unknownr` z.Ignore fields whose type cannot be determined.z-Lz--legacy-naming
legacy_namingz-Use legacy table- and column-name generation.)r
add_optionrG r )parseraoenginess r get_option_parserr s
F
G
G
GF BBtXF####BtXF////BtXF####Bt\
<@@@@\""GBtZh(* * * * BtZh''''BtZh') ) ) ) BtYW\ EG G G GBtXF< ! ! ! ! Bt
;K DF F F FBt
;K <> > > >Bt
|/ ;= = = = Mr c r d}t fd|D } j rt |d< |S )N)r r r r c 3 \ K | ]&}t | |t | fV 'd S r )getattr)r ooptionss r r z%get_connect_kwargs.. sA OOq77A;N;NO1ggq))*OOOOOOr r )dictr r )r opsr* s ` r get_connect_kwargsr sH
,C
OOOOCOOO
O
OF ' $YYz Mr __main__r z%Missing required parameter "database"c ^ g | ]*}| | +S r8 )strip)r rQ s r r> r> s< $ $ $E[[]]$%++-- $ $ $r , )NFFFT)5rs osr# r optparser peeweer r rv playhouse.cockroachdbr playhouse.reflectionre rl rk
MySQLDatabasePostgresqlDatabaseSqliteDatabaser r r r- rq ry r r r rh argvraw_argvr
parse_argsr argsrI
print_helpr$ connectr_ rn splitr pathexistsra r r! rw rb r r` r r8 r r r s
! ! ! ! ! ! 0 0 0 0 0 0 3 3 3 3 3 3 " " " "
;;;GY'\2Xy) t 8 8 /8 8 8 8 8 9 9 9