o gY+@sddlZddlmZddlmZddlmZddlmZ ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlm Z!ddlm"Z#e$ej%dddddej%dddddeddZ&e&'eddZe&'ej%dddddej%d dd!d"ej%d#d$dd%dej%d&dd'd"ed(d)Ze&'ej%dddddej%d*d+dd,dej%d-dd.d"ej%d/dd0d"ej%d1d2d3dej%d4dd5d"ej%d6dd7dej%d8dd9dej%d:dd;dedd?Z e&'ej%dddddej(d=d2d@edAdBZe&'ej%dddddej%d*d+ddCdej%d6dd7dej%d:dd;dej(dDdEdFedGdHZe&'ej%dddddej%d/dd0d"ej%dIddJdej%dddddej(d=d2d@edKdLZe&'ej%dddddej%d/dd0d"ej%dIddJdej%dddddej(d=dMd@edNdOZe&'ej%dddddej(d=d2d@edPdQZe&'ej%dddddej%dRdSddTdej%dUdVddWd"ej%dXdYddZd"ed[d\Ze&'ej%dddddej%dUdVddWd"ej%d]dd^d"ed_d`Ze&'ej%dddddej%dUdVddWd"edadbZe&'ej%dddddej%dUdVddWd"edcddZe&'ej%dddddej%d/dd0d"ej%dIddJdej%deddfd"ej(d=d2d@edgdhZ e&'ej%dddddedidjZ"dS)kN)g)with_appcontext)list_templates)init)revision)migrate)edit)merge)upgrade) downgrade)show)history)heads)branches)current)stamp)checkz-dz --directoryz4Migration script directory (default is "migrations"))defaulthelpz-xz--x-argTz6Additional arguments consumed by custom env.py scripts)multiplercCs|t_|t_dS)zPerform database migrations.N)r directoryx_arg)rrrM/var/www/html/flaskapp/venv/lib/python3.10/site-packages/flask_migrate/cli.pydbs rcCs tdS)zList available templates.N)_list_templatesrrrrr!s rz --multidbzSupport multiple databases)is_flagrz-tz --templatez/Repository template to use (default is "flask")z --packagezFWrite empty __init__.py files to the environment and version locationscCt|ptj|||dS)z#Creates a new migration repository.N)_initrr)rmultidbtemplatepackagerrrr(s rz-mz --messagezRevision messagez--autogeneratezfPopulate revision script with candidate migration operations, based on comparison of database to modelz--sqlz@head to base new revision onz--splicez6Allow a non-head revision as the "head" to splice ontoz--branch-labelz3Specify a branch label to apply to the new revisionz--version-pathz2Specify specific path from config for version filez--rev-idz9Specify a hardcoded revision id instead of generating onec Cs"t|ptj|||||||| dS)zCreate a new revision file.N) _revisionrr) rmessage autogeneratesqlr"splice branch_label version_pathrev_idrrrr8src Cs(t|ptj||||||||ptj dS)zJAutogenerate a new revision file (Alias for 'revision --autogenerate')N)_migraterrr) rr$r&r"r'r(r)r*rrrrrVsr)rcCt|ptj|dS)zEdit a revision fileN)_editrrrrrrrrtrzMerge revision message revisions)nargscCst|ptj||||dS)z:Merge two revisions together, creating a new revision fileN)_mergerr)rr$r(r*r0rrrr ~s r z--tagz;Arbitrary "tag" name - can be used by custom env.py scriptscC t|ptj||||p tjdS)zUpgrade to a later versionN)_upgraderrrrr&tagrrrrrr  r z-1cCr4)zRevert to a previous versionN) _downgraderrrr6rrrr r8r cCr,)z.Show the revision denoted by the given symbol.N)_showrrr.rrrr r/r z-rz --rev-rangez1Specify a revision range; format is [start]:[end]z-vz --verbosezUse more verbose outputz-iz--indicate-currentz?Indicate current version (Alembic 0.9.9 or greater is required)cCr)z.List changeset scripts in chronological order.N)_historyrr)r rev_rangeverboseindicate_currentrrrr s r z--resolve-dependenciesz+Treat dependency versions as down revisionscCst|ptj||dS)z4Show current available heads in the script directoryN)_headsrr)rr=resolve_dependenciesrrrrs rcCr,)zShow current branch pointsN) _branchesrrrr=rrrrr/rcCr,)z/Display the current revision for each database.N)_currentrrrBrrrrr/rz--purgez?Delete the version in the alembic_version table before stampingcCst|ptj||||dS)zP'stamp' the revision table with the given revision; don't run any migrationsN)_stamprr)rr&r7rpurgerrrrsrcCst|ptjdS)z0Check if there are any new operations to migrateN)_checkrr)rrrrrsr))clickflaskr flask.clir flask_migraterrrrrr#rr+rr-r r3r r5r r9r r:r r;rr?rrArrCrrDrrFgroupoptionrcommandargumentrrrrs