ndZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z dd l m Z dd l mZdd l mZdd lmZdd lmZdZdZdZdZGddej0ZdZGddeZdZdZdZddZ GddeZ!dZ"e!eee!eee!eee!e edZ#y) z,Library for safe migrations of config files.)absolute_import)division)unicode_literalsN)datastore_index_xml)cron_xml_parser)dispatch_xml_parser)queue_xml_parser) exceptions)log) console_io)filesz%Translates a cron.xml into cron.yaml.z'Translates a queue.xml into queue.yaml.z-Translates a dispatch.xml into dispatch.yaml.z3Translates a datastore-indexes.xml into index.yaml.c eZdZy)MigrationErrorN)__name__ __module__ __qualname__4lib/googlecloudsdk/command_lib/app/migrate_config.pyrr)srrc |dzS)Nz.bakr) file_paths r_Bakifyr-s V rc4eZdZdZdZdZdZdZdZdZ y) MigrationResultzThe changes that are about to be applied on a declarative form. Args: new_files: {str, str} a mapping from absolute file path to new contents of the file, or None if the file should be deleted. c||_yN new_files)selfrs r__init__zMigrationResult.__init__9s DNrc4|j|jk(Srrrothers r__eq__zMigrationResult.__eq__<s >>U__ ,,rc||k( Srrr"s r__ne__zMigrationResult.__ne__?su} rc|jjD]}t|}tjj |s.tjj |rtdj|tjjdj||tj||y)Nz%Backup file path [{}] already exists.zCopying [{}] to [{}])rkeysrospathisfileexistsrformatr errPrintshutilcopy2)rr*bak_paths r_BackupzMigrationResult._BackupBs##%h WW^^D !  ! 3 : :8 DF F ggmm*11$AB ll4"&rc|jjD]\}}|Dtjj dj |t j|Ltjj dj t jj|rdnd|tj||y)Nz Deleting [{}]z{} [{}] OverwritingWriting) ritemsr r.r/r-r)remover*r,r WriteFileContents)rr* new_contentss r _WriteFileszMigrationResult._WriteFilesMs"nn224l    o,,T23 $  i&&WW^^D1My$H I l35rcD|j|jy)z>Backs up first, then deletes, overwrites and writes new files.N)r3r;)rs rApplyzMigrationResult.ApplyWsLLNrN) rrr__doc__r r$r&r3r;r=rrrrr1s%- #4rrcztj|}tjd|}|d||i}t |S)zMigration script for cron.xml.N)r ReadFileContentsr GetCronYamlrsrcdstxml_str yaml_contentsrs r_MigrateCronXMLrG]s=  " "3 ''!--dG<-D#}-)  ##rcztj|}tjd|}|d||i}t |S)zMigration script for queue.xml.N)r r@r GetQueueYamlrrBs r_MigrateQueueXMLrJes=  " "3 ''"//g>-D#}-)  ##rcztj|}tjd|}|d||i}t |S)z"Migration script for dispatch.xml.N)r r@rGetDispatchYamlrrBs r_MigrateDispatchXMLrMms=  " "3 ''%55dGD-D#}-)  ##rc:tj|}tj|}|di}|rNtj|}tj|}|xj|jz c_d||<|j ||<t |S)z+Migration script for datastore-indexes.xml.N)r r@rIndexesXmlToIndexDefinitionsindexesToYAMLr)rCrDauto_srcrErPr xml_str_2 auto_indexess r_MigrateDatastoreIndexXMLrUus  " "3 ''  < >#)C.  ##rceZdZdZdZy)MigrationScriptzObject representing a migration script and its metadata. Attributes: migrate_fn: a function which accepts a variable number of self-defined kwargs and returns a MigrationResult. description: str, description for help texts and prompts. c ||_||_yr) migrate_fn description)rrYrZs rr zMigrationScript.__init__s DO"DrN)rrrr>r rrrrWrWs #rrWc |jdi|}tjdtj|j ddd|j y)zRun a migration entry, with prompts and warnings. Args: entry: MigrationScript, the entry to run. **kwargs: keyword args for the migration function. z!Please *back up* existing files. Tz,Do you want to run the migration script now?)default prompt_string cancel_on_noNr)rYr warningr PromptContinuerZr=)entrykwargsresults rRunrdsP 5   %f %&++23  B ,,.r)zcron-xml-to-yamlzqueue-xml-to-yamlzdispatch-xml-to-yamlzdatastore-indexes-xml-to-yamlr)$r> __future__rrrr)r0"googlecloudsdk.appengine.datastorergooglecloudsdk.appengine.toolsrrr googlecloudsdk.corer r googlecloudsdk.core.consoler googlecloudsdk.core.utilr _CRON_DESC _QUEUE_DESC_DISPATCH_DESC _INDEX_DESCErrorrrobjectrrGrJrMrUrWrdREGISTRYrrrrrs 3&' B:>;*#2*5 7 @C Z%%)f)X$$$ $ #f # $(D()9;G+,?,:<%45N5@&B  r