o Ph-@sBddlZddlZddlmZddlZddlZddlmZddlm Z ddl m Z ddl m Z dd l mZddl m Z dd lmZdd lmZd d Zd;ddZddZddZddZddd Zd?d"d#Zd?d$d%Zd&d'Zd(d)Zd*d+Z d@d-d.Zd/d0Zd1d2Z d3d4Z!d5d6Z"d7e#d8e#fd9d:Z$dS)AN)Path)config) provision)util)command)script)Script)ScriptDirectorycCstjr dtjSdS)Nscratch_scratch)rFOLLOWER_IDENTrrX/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/alembic/testing/env.py_get_staging_directorys rTgenericFcCst}|rIttd}tj|rJd|tj|||d|rIzt |dWn t y2Ynw|dvs;J|t t|dd|vrFdndt j |}|S)Nscriptsz2staging directory %s already exists; poor cleanup?)templateenv.py)pep3147_envonlysimplepep3147_everythingpep3147r)_testing_config _join_pathrospathexistsrinitrload_python_fileAttributeErrormake_sourcelessr r from_config)creater sourcelesscfgrscrrr staging_envs,    r(cCs(ddlm}|jttddS)NrenginesT)sqlalchemy.testingr*testing_reaper close_allshutilrmtreerr)rrrclear_staging_env;s  r0cCsPttd}t|d}t|d}||WddS1s!wYdS)Nrzscript.py.makow)rropenwrite)txtdir_rfrrrscript_file_fixtureBs    "r7cCspttd}d|}t|d}t|}|rt|t|d}||WddS1s1wYdS)Nrz6 from alembic import context config = context.config rr1)rrrpyc_file_from_pathrunlinkr2r3)r4r5rpyc_pathr6rrrenv_file_fixtureIs     "r;foo.dbcKs4ttd}d||f}|r||d<tj|||dS)Nrzsqlite:///%s/%sscope)urlfutureoptions)rr testing_utiltesting_engine)tempnamer?r=r@r5r>rrr_sqlite_file_db]s  rDc Cs\ttd}d|d}|pdtjjjv}td|d|d|r!dnd d |r(d nd d S)Nrz sqlite:///z/foo.dbr? [alembic] script_location =  sqlalchemy.url =  sourceless = truefalse zsqlalchemy.future = truea [loggers] keys = root,sqlalchemy [handlers] keys = console [logger_root] level = WARNING handlers = console qualname = [logger_sqlalchemy] level = DEBUG handlers = qualname = sqlalchemy.engine [handler_console] class = StreamHandler args = (sys.stderr,) level = NOTSET formatter = generic [formatters] keys = generic [formatter_generic] format = %%(levelname)-5.5s [%%(name)s] %%(message)s datefmt = %%H:%%M:%%S rrrdb __class__ __module___write_config_file)r%r?r5r>sqlalchemy_futurerrr_sqlite_testing_configes    rRrKc CsZttd}dtjjjv}d|}td|d|d|rdndd |r$dndd |d S) Nrr?zsqlite:///%s/foo.dbrErFz sqlalchemy.future = rHrIrGz_ path_separator = space version_locations = %(here)s/model1/ %(here)s/model2/ %(here)s/model3/ aR [loggers] keys = root [handlers] keys = console [logger_root] level = WARNING handlers = console qualname = [handler_console] class = StreamHandler args = (sys.stderr,) level = NOTSET formatter = generic [formatters] keys = generic [formatter_generic] format = %%(levelname)-5.5s [%%(name)s] %%(message)s datefmt = %%H:%%M:%%S rL)r%extra_version_locationr5rQr>rrr_multi_dir_testing_configs   rT postgresqlcCs2ttd}td|dt|dd|dS)Luse a postgresql url with no host so that connections guaranteed to failrz" [tool.alembic] script_location ="z" z z [alembic] sqlalchemy.url = aR:// [loggers] keys = root [handlers] keys = console [logger_root] level = WARNING handlers = console qualname = [handler_console] class = StreamHandler args = (sys.stderr,) level = NOTSET formatter = generic [formatters] keys = generic [formatter_generic] format = %%(levelname)-5.5s [%%(name)s] %%(message)s datefmt = %%H:%%M:%%S )rr_write_toml_configtextwrapdedentdialect directivesr5rrr_no_sql_pyproject_configs r]cCs(ttd}td|d|d|dS)rVrz [alembic] script_location =rFz:// aO [loggers] keys = root [handlers] keys = console [logger_root] level = WARNING handlers = console qualname = [handler_console] class = StreamHandler args = (sys.stderr,) level = NOTSET formatter = generic [formatters] keys = generic [formatter_generic] format = %%(levelname)-5.5s [%%(name)s] %%(message)s datefmt = %%H:%%M:%%S )rrrPrZrrr_no_sql_testing_configs r^cCsDt|}t|jd}||Wd|S1swY|SNr1)rPr2toml_file_namer3)tomltextinitextr&r6rrrrWs  rWcCsBt}t|jd}||Wd|S1swY|Sr_)rr2config_file_namer3)textr&r6rrrrPs  rPcCsBddlm}tttjstt|ttdttdS)NrConfigztest_alembic.inizpyproject.toml)alembic.configrfraccessrF_OKmkdirrrerrrr s    rasciic Cs|j|}|j}t|}|r||}t|d }||Wdn1s*wYt |}|r;t |t ||} |j| j}|j| jkrRtd|jj| dd|rit||dkrddnddSdS)Nwbz2Can't change down_revision on a refresh operation.T)_replacerrr) revision_map get_revisionrrXrYencoder2r3rr8rr9r _from_pathrevision down_revision Exception add_revisionr") scriptdirrev_idcontentencodingr%oldrfpr:r rrr write_script+s*         r|cCsddl}|||dkr-t|}tjjd}tj |\}}||}t |||}n |dvs3Jt|}t |tj sAJt|dS)Nrr)rr) py_compilecompilerr8 importlib machineryBYTECODE_SUFFIXESrrsplitextr.moverhrir9)rstyler}r:suffixfilepathextsimple_pyc_pathrrrr"Es      r"cCst}t}t}t|}|j|ddddt||d|d|j|dd|dt||d|d |d d d |j|d d|dt||d|d |d|||fS)Nz revision aTbase)refreshheadz"Rev A" revision = 'z' down_revision = None from alembic import op def upgrade(): op.execute("CREATE STEP 1") def downgrade(): op.execute("DROP STEP 1") z revision bu/# coding: utf-8 "Rev B, méil, %3" revision = '' down_revision = 'z}' from alembic import op def upgrade(): op.execute("CREATE STEP 2") def downgrade(): op.execute("DROP STEP 2") zutf-8)ryz revision cz"Rev C" revision = 'z}' from alembic import op def upgrade(): op.execute("CREATE STEP 3") def downgrade(): op.execute("DROP STEP 3") rrwr r#generate_revisionr|)r&abcr rrrthree_rev_fixtureZsD  rcCst}t}t}t|}|j|d|dddt||d|d|d|j|d|dddt||d|d|d |j|d |dddt||d |d|d |||fS) z:Create a multiple head fixture from the three-revs fixturezrevision d from bT)rsplicerz"Rev D" revision = 'rz}' from alembic import op def upgrade(): op.execute("CREATE STEP 4") def downgrade(): op.execute("DROP STEP 4") zrevision e from dz"Rev E" revision = 'z}' from alembic import op def upgrade(): op.execute("CREATE STEP 5") def downgrade(): op.execute("DROP STEP 5") zrevision f from bz"Rev F" revision = 'z}' from alembic import op def upgrade(): op.execute("CREATE STEP 6") def downgrade(): op.execute("DROP STEP 6") r)r&rrrder6r rrrmulti_heads_fixturesR     rc Csjttd}dtjjjv}d|}ddd|D}t d|d|r)d nd d |d|d S) z?alembic.ini fixture to work exactly with the 'multidb' templaterr?z, z css&|]\}}d|d|jVqdS)[z] sqlalchemy.url = N)r>).0keyvaluerrr s  z*_multidb_testing_config..rEz( sourceless = false sqlalchemy.future = rHrIz databases = aQ [loggers] keys = root [handlers] keys = console [logger_root] level = WARNING handlers = console qualname = [handler_console] class = StreamHandler args = (sys.stderr,) level = NOTSET formatter = generic [formatters] keys = generic [formatter_generic] format = %%(levelname)-5.5s [%%(name)s] %%(message)s datefmt = %%H:%%M:%%S ) rrrrMrNrOjoinkeysitemsrP)r*r5rQ databasesrrr_multidb_testing_configs"   rrmorecGstt|j|S)N)strrjoinpathas_posix)rrrrrr,sr)TrF)r<FN)FF)FrK)rUrK)rkF)%importlib.machineryrrpathlibrr.rXr+rrrKrrArr r r rr(r0r7r;rDrRrTr]r^rWrPrr|r"rrrrrrrrrsB          "   . , ,' KV1