o JhY@sddlmZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlmZddlmZmZmZddlmZd d lmZdd lmZmZmZmZed Z d Z!edZ"edZ#ddZ$de%fddZ&ddZ'ddZ(ddZ)dS))date)dedent)NamedTemporaryFileN)default_migration_table) CONFIG_NEW_MIGRATION_COMMAND_KEY)read_migrationsheads Migration)utils)InvalidArgument)pathstatunlinkrenamezyoyo.migrations _tmp_yoyonewz """ {message} """ from yoyo import step __depends__ = {{{depends}}} steps = [ step("") ] z0 -- {message} -- depends: {depends} cCsz|jd|gdd}|jtd|jddddd |jd d d d |jdddtdd|jdddd|jddddddS)NnewzCreate a new migration)parentshelp)funcz --messagez-mMessage)rdefaultz--sqlzCreate file in SQL format store_true)ractionz--migration-tablemigration_tablestorez3Name of table to use for storing migration metadata)destrrrsources*z%Source directory of migration scripts)nargsrz-dz --databasezIDatabase, eg 'sqlite:///path/to/sqlite.db' or 'postgresql://user@host/db')rr) add_parser set_defaults new_migration add_argumentr) global_parser subparsers parser_newr([/var/www/html/pro-man-master/venv/lib/python3.10/site-packages/yoyo/scripts/newmigration.pyinstall_argparsers@s0  r*returnc snz|jd}Wn tytdw|j}t|}tt|ddd}|jr4t}d dd|D}n t }d d d|D}|j ||d }|jrLd nd } |j rvt |||| tjddd } | |Wdn1spwYn t|||| durdSz"|dt} fddt| D} tdd | t| Wn tjyYnwtddS)Nrz%Please specify a migrations directorycSs|jSNid)mr(r(r)dsznew_migration..)keyz css|]}|jVqdSr,r-.0r/r(r(r) gsz new_migration..z, css|] }d|jVqdS)z{!r}N)formatr.r2r(r(r)r4js)messagedependsz.sqlz.pywUTF-8)encodingr DEFAULTcsg|]}|qSr()r5r3partpr(r) ysz!new_migration..zRunning command: %s z Created file)r IndexErrorr r6rsortedrsqlmigration_sql_templatejoinmigration_templater5 batch_mode make_filenameioopenwritecreate_with_editorgetrshlexsplitloggerinfo subprocesscall configparser NoOptionErrorprint) argsconfig directoryr6 migrationsr7template depends_strmigration_source extensionfcommandr(r>r)r#\sD    r#cCs>t|}ttdd|}tdd|d}|S)Nz [^-a-z0-9]+-z-{2,})r unidecoderesubcompilelowerstrip)r6sr(r(r)slugifys rjc Cs$dd|dD}dd|D}t|d}|rdt|}nd}td}d}td }z|d d } Wn t j yCd} Ynwt t |d | |D]2} t | t| t|d ddd} z|| krytt| d d}WqQtyYqQwt |d| |||||S)Ncss|]}|VqdSr,)rhr3liner(r(r)r4z make_filename.. css|]}|r|VqdSr,r(rkr(r(r)r4rmrbrz%Y%m%d01r;prefixz{}{}_*r _rz{}{}_{}_{}{}{})rPnextrjrtodaystrftimer get_random_stringrNrUrVglobr rFr5basenamelenstrintzfill ValueError) rYrZr6r_linesslugdatestrnumberrandrqr?nr(r(r)rIs6  , rIc stt|}tdd|t|ddz }||Wdn1s#wYfddt|D}tfdd|DsE|j t j j }t j d | zt|Wn tygtd Yn!wt j j |krtd Wztj WdStyYdSwztdj d}|t|jd d}Wnbtyd}tdttttdd} | dkrYWztj WdStyYdSw| dkrYqR| dkrYn"| dkrtdtdtdtdtdtdYqRYnwqSt j ddt _ t||||} tj | | Wztj WSty'YSwztj Wwty9Yww)Nr8r9F)moder:dirrqsuffixdeletecsg|]}|jqSr()r5namer<tmpfiler(r)r@sz&create_with_editor..c3s|]}j|vVqdSr,)rr<rr(r)r4sz%create_with_editor..rTzError: could not open editor!zabort: no changes made__doc__rzError loading migrationzRetry editing?zYnq?qyr?z+y: reopen the migration file in your editorz/n: save the migration as-is, without re-editingz$q: quit without saving the migrationz?: show this helpr )r get_editorrtempfile_prefixrLrOrPanyappendrrst_mtimesysr insertrSrTOSErrorrWrr loadgetattrmodule Exception traceback format_excpromptrIr) rYrZr^r_editorr`mtime migrationr6rfilenamer(rr)rMs      "     $ rM)*datetimertextwraprtempfilerrUrxloggingrJrdrOrSrryoyor yoyo.configryoyo.migrationsrrr r mainr osr rrr getLoggerrQrrGrEr*r|r#rjrIrMr(r(r(r)s>         ' "