o Jho(@slddlmZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZejejejejd ZeeZeeZdZGdddeZ d"dej ej!ej"ej#ffddZ$ddZ%ddZ&ddZ'ddZ(ddZ)d"ddZ*e+d kre,e*ej-d!ddSdS)#)getpassN) connectionsdefault_migration_table)logger)utils)CONFIG_FILENAME) find_config) read_config) save_config)config_changed)update_argparser_defaults)rz .yoyo-migratec@s eZdZdS)InvalidArgumentN)__name__ __module__ __qualname__rrS/var/www/html/pro-man-master/venv/lib/python3.10/site-packages/yoyo/scripts/main.pyr/srreturnc Csdddddd}t\}}}||\}}|jpt}t|jr!|nd}i} |D]\} } z t|| d| | | <Wq*tj yDYq*wd| vrQ| d | d<t || t || |j D]} t | | q`||} | j||dj||| fS) z Parse the config file and command line args. :return: tuple of ``(parsed config file, argument parser, parsed arguments)`` getbooleangetgetint) batch_modesourcesdatabase verbositymigration_tableNDEFAULTrr)make_argparserparse_known_argsconfigr r use_config_fileitemsgetattr configparser NoOptionErrorsplitr choicesvalues parse_args__dict__update)argv config_args globalparser argparser subparsers global_args_ configfiler#defaultsargnamegettersubpargsrrrr,3s4        r,cCstjdd}|jddddd|jdd d td d |jd dddtj dd |jddddddd tjd|gd}|jdd}ddlm }ddlm }ddlm }| ||| ||| |||||fS) zY Return a top-level ArgumentParser parser object, plus a list of sub_parsers F)add_helpz--configz-cNzPath to config file)defaulthelpz-vrcountzAVerbose output. Use multiple times to increase level of verbosity)destactionr=r>z-bz--batchr store_truez-Run in batch mode. Turns off all user promptsz--no-config-filez --no-cacher$ store_falseTz%Don't look for a yoyo.ini config fileyoyo)progparentsz Commands help)r>r)migrate) newmigration)init) argparseArgumentParser add_argument min_verbositysysstdoutisattyadd_subparsersrGrHrIinstall_argparsers) global_parserr2r3rGrHrIrrrr!jsH           r!cCstjt|ddS)zI Configure the python logging module with the requested loglevel levelN)logging basicConfigverbosity_levelsrUrrrconfigure_loggingsrZcCs"td|rt||dSdS)NzSave migration configuration to {}? This is saved in plain text and contains your database password. Answering 'y' means you do not have to specify the migration source or database connection for future runs)rconfirmformatr )r#pathrrrprompt_save_configs r^c s^t|D]}tj|t}tj|sqt|dfdd }|dd|dddd|jp1t}|j smt d ||rjt |zt d |rUt|Wd SWd Styitd |Yd SwdStd |z |jp|dd|_Wn tjyYnwzt|dsdd|_WdSWdStjyYdSwdS)Nr c snz ||WdStjy6z||}Wn tjy%YYdSw|r,||}|||YdSwN)rr'r(set)oldnamenewname transformsectionvaluer# legacy_configrrtransfer_settingsz/upgrade_legacy_config..transfer_settingdburirrcSs|dkrtS|S)NNoner)vrrrsz'upgrade_legacy_config..z*Move legacy configuration in {!r} to {!r}?z%Delete legacy configuration file {!r}zGCould not remove %r. Manually remove this file to avoid future warningsTzaFound legacy configuration in %r. Run yoyo in interactive mode to update your configuration filesF)Nr )reversedosr]joinLEGACY_CONFIG_FILENAMEisfiler r#rrrr[r\r unlinkOSErrorrwarnrrr'r(varsr)r;r#rdirr]rh config_pathrrfrupgrade_legacy_configsp        rxc Csz|j}Wnty|dd}Ynwz|j}Wnty8z|dd}Wn tjy5t}YnwYnw|durAtdz|jrWt d|}t |}|j |dj }Wn tyaYnwt ||S)Nr rrzPlease specify a database urizPassword for %s: )password)rAttributeErrorrrr'r(rrprompt_passwordrr parse_uri_replaceuri get_backend)r;r#rirryparsedrrrrs6        rc Cst|\}}}t|dddur|tj|d|gko&|dgk}t|dd}|j}t t t t |}t |t|drN|ddd|jt|dra|dd|jddt|d rp|dd |j|dd t|d r}d nd |dd tt|d |rt|||rt|Szt|dr|||}Wnty}z||jdWYd}~nd}~ww|r|jr|js|jpt } t!|| rt"|| |S)Nfuncrr r r%z%%rronoffrr)#r,r& print_usagerNstderrexitsectionsr%rmin max_verbositymaxrMrZrurr`rorrreplacerstrrxmainrrerrorr;r$rr#rr r^) r/r#r2r;config_is_emptyrrexitcodee config_filerrrrsJ        r__main__rr_).rrJr'rWrnrNtypingtrDrrrr yoyo.configrr r r r r ERRORWARNINFODEBUGrYrrMrrrp ExceptionrTuple ConfigParserrK Namespacer,r!rZr^rxrrrrr/rrrrsN            70E /