o 8cl#@sddlZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddlmZddlmZedd e d dd DZeeZGd d d eZGddde Z GdddeZddZeddZed4ddZe   d5ddZe  d6ddZed7dd Ze  d8d"d#Z ed9d$d%Z!ed:d'd(Z"ed;d)d*Z#e  dd/d0Z&ed>d1dZ'ed?d2d3Z(dS)@Nwraps) current_app) __version__)Config)command) CommandErrorcCsg|]}t|qS)int).0vr r L/var/www/html/gps/gps/lib/python3.10/site-packages/flask_migrate/__init__.py sr.c@s eZdZddZeddZdS)_MigrateConfigcKs||_||_|j|_||_dSN)migratedb directoryconfigure_args)selfrrkwargsr r r __init__s z_MigrateConfig.__init__cCs|jjS)z Backwards compatibility, in old releases app.extensions['migrate'] was set to db, and env.py accessed app.extensions['migrate'].metadata )rmetadata)rr r r rsz_MigrateConfig.metadataN)__name__ __module__ __qualname__rpropertyrr r r r rsrcs$eZdZfddZddZZS)rcs$|dd|_tj|i|dS)Ntemplate_directory)poprsuperr)rargsr __class__r r r!szConfig.__init__cCs.|jr|jStjtjt}tj|dS)N templates)rospathabspathdirname__file__join)r package_dirr r r get_template_directory%szConfig.get_template_directory)rrrrr- __classcell__r r r#r r s rc@s:eZdZd ddZdddZddZd d Zdd d ZdS)MigrateN migrationscKsFg|_||_t||_||_|dur|dur!||||dSdSdSr)configure_callbacksrstrralembic_ctx_kwargsinit_apprapprrrr r r r-s zMigrate.__init__cKsX|p|j|_t|p |j|_|j|t|dsi|_t||jfi|j|jd<dS)N extensionsr)rr2rr3updatehasattrr7rr5r r r r45s   zMigrate.init_appcCs|j||Sr)r1append)rfr r r configure>s zMigrate.configurecCs|jD]}||}q|Sr)r1)rconfigr;r r r call_configure_callbacksBs  z Migrate.call_configure_callbackscCs|dur|j}t|}ttj|d}|d||jdur$t |_|p'gD] }t |j|dq(t |jdsi|durbt |jdgt |t sMt |trZ|D] }|jj|qOn|jj|nt |jdd||S)N alembic.iniscript_locationTx)rr2rr&r'r+set_main_optioncmd_optsargparse Namespacesetattrr9 isinstancelisttuplerAr:r>)rrx_argoptsr=optrAr r r get_configGs&      zMigrate.get_config)NNr0)NN)NNN)rrrrr4r<r>rMr r r r r/,s    r/cstfdd}|S)Nc s\z |i|WdSttfy-}ztdt|tdWYd}~dSd}~ww)NzError: )r RuntimeErrorlogerrorr2sysexit)r"rexcr;r r wrapped_szcatch_errors..wrappedr)r;rVr rUr catch_errors^srWc Cs~t}|dtt|D]+}ttj||d}t | }Wdn1s0wY|d||qdS)zList available templates.zAvailable templates: READMENz%s - %s) r print_stdoutsortedr&listdirr-openr'r+nextstrip)r=tempnamereadmesynopsisr r r list_templatesis rbFcCs|dur tjdj}d}|dur d|vsd|vr tj|\}}t|d}|d|tj|d|_ tjdj |}|rE|durEd}n|durKd }t j ||||d dS) z"Creates a new migration repositoryNr/\)rr@r?z flask-multidbflask)templatepackage)rr7rr&r'splitrrBr+config_file_namerr>rinit)rmultidbrfrgrr=r r r rjvs     rjheadc Cs2tjdj|} tj| ||||||||d dS)zCreate a new revision file.r autogeneratesqlrlsplice branch_label version_pathrev_idNrr7rrMrrevision) rmessagernrorlrprqrrrsr=r r r rus   ruc Cs:tjdjj|dg|d} tj| |d||||||d dS)z#Alias for 'revision --autogenerate'rrn)rKrJTrmNrt) rrvrorlrprqrrrsrJr=r r r rs   rcurrentcCs2tdkrtjdj|}t||dStd)zEdit current revision.)rrrz$Alembic 0.8.0 or greater is requiredN)alembic_versionrr7rrMreditrOrrur=r r r rzs  rzcCs*tjdj|}tj|||||ddS)z;Merge two revisions together. Creates a new migration filer)rvrqrsN)rr7rrMrmerge)r revisionsrvrqrsr=r r r r}s  r}cCs,tjdjj||d}tj||||ddS)zUpgrade to a later versionrrJrotagN)rr7rrMrupgraderrurorrJr=r r r rsr-1cCs<tjdjj||d}|r|dkrd}tj||||ddS)zRevert to a previous versionrrrzhead:-1rN)rr7rrMr downgraderr r r rs  rcCs"tjdj|}t||dS)z.Show the revision denoted by the given symbol.rN)rr7rrMrshowr{r r r rsrcCsDtjdj|}tdkrtj||||ddStj|||ddS)z.List changeset scripts in chronological order.r)r r)verboseindicate_currentrN)rr7rrMryrhistory)r rev_rangerrr=r r r rs   rcCs&tjdj|}tj|||ddS)z4Show current available heads in the script directoryr)rresolve_dependenciesN)rr7rrMrheads)rrrr=r r r rs rcC$tjdj|}tj||ddS)zShow current branch pointsrrN)rr7rrMrbranchesrrr=r r r rrcCr)z/Display the current revision for each database.rrN)rr7rrMrrwrr r r rwrcCs(tjdj|}tj||||ddS)zP'stamp' the revision table with the given revision; don't run any migrationsrrN)rr7rrMrstamp)rrurorr=r r r rsr)NFNF) NNFFrlFNNN) NNFrlFNNNN)Nrw)Nr|NNN)NrlFNN)NrFNN)Nrl)NNFF)NFF)NF)NrlFN))rD functoolsrloggingr&rRreralembicr__alembic_version__alembic.configr AlembicConfigr alembic.utilrrIrhry getLoggerrrPobjectrr/rWrbrjrurrzr}rrrrrrrwrr r r r sj         2