#6dZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z dd l mZdd l mZdd l mZdd lmZdd lmZej(dZdZdZdZdZdZdZdZGddej:ZdZdZ dZ!ddZ"ddZ#ddZ$dZ%y) zJModule containing the command utils for resource-config/terraform surface.)absolute_import)division)unicode_literalsN)core_exceptions)log) console_io)files) platforms)times)runtime)templatez^#zterraform importzterraform_import_{ts}.{suffix}z%Y%m%d-%H-%M-%Sz; module "{module_name}" {{ source = "{module_source}" }} z= #!/bin/sh # Terraform Import Script generated by gcloud cli zInvalid export Path {}. Path should point to an export Terraform(.tf) file or directotry of Terraform files. Please see `gcloud alpha resource-config bulk-export` for more details.zgcloud-export-modules.tfceZdZdZy)TerraformGenerationErrorzEBase Exception for errors that occur during import script generation.N)__name__ __module__ __qualname____doc__Blib/googlecloudsdk/command_lib/util/declarative/terraform_utils.pyrr:sMrrc(tjj|r|jdr|g}nHtjj |rt j |dd}nttg}g}|D]}tjj|}d|vs|djrtj|Ot j|t} t|j}t j tjj#||j%ddj'f}|j)||st/d j1| |r3t3j4dj1dj7||S#t*$r|j)|Y[t j,$r!} t/d j1|| d } ~ wwxYw)zDRead files from export path and parse out import command statements..tfFc$|jdS)Nr)endswith)xs rz"ParseExportFiles..Gs E!2r) include_dirsfile_predicatedefaultr#z0Could not parse Terrorm data from {path}:: {err})patherrNz-No Terraform importable data found in {path}.)r"z=Error generating imports for the following resource files: {} )osr"isfilerisdirr GetDirectoryTreeListing ValueErrorINVALID_EXPORT_PATH_MSGbasenameisdigitremoveFilteredFileReader IMPORT_REGEXlistpopExpandHomeAndVarsdirname partitionstripappend IndexErrorErrorrformatrwarningjoin) export_path input_files import_data error_filesin_filein_file_base_name import_readercommand import_linees rParseExportFilesrF>sWW^^K [%9%9%%@-K ww}}[!//24K , --++g((1%%):1)=)E)E)Gii..w Em *}%))+..rwww/GH((-a0668: ;'&  "7>> ?  KKGNN IIk " $% $7# [[*& > E E a F )* **s-A>GHH0H  Hctjjrdnd}tj t j t jt|S)z.Generate a default filename for import script.cmdsh)tssuffix) r OperatingSystem IsWindowsIMPORT_SCRIPT_DEFAULT_NAMEr9r FormatDateTimeNowIMPORT_DATE_FORMAT)rKs rGenerateDefaultScriptFileNamerRjsM--7795t& # * *   eiik+= >v + OOrctjjdtjj||}djtjj |j |dj tjjdj}|djrdj|}||fS)N.)start-rzgcp-{}) r%r"r;relpathnormpathreplacesplitseplstriprstripr,r9) import_pathdest_dir module_source module_names rConstructModuleParametersrcqs'',,s!ww{(KM-gg{**8R89?? &&6#;vvx^//+.K  ##rc |r|jnd}|r-tjj|jnd}d}d}|rtjj |r+dj |}t j|ddddtjj|}tjj|xstj}tjj|r/tj|stdj ||rRtjj|r/tj|stdj |d}|}||fS) z9Helper function for generating output file and directory.Nz{} already exists.zDo you want to overwrite?TzAborted script generation.) prompt_stringr cancel_string cancel_on_noz-Error writing output file: {} is not writablez>Cannot write output to directory {}. Please check permissions.)r5r%r"abspathr&r9rPromptContinuer+r3r GetCWDr'HasWriteAccessInDirrr)) output_file output_dir dest_filer`overwrite_prompts rProcessOutputParametersrp|sA'2 !!#+7A 0 0 23t*) ( ww~~k".55kB 34    -Iww{+=u||~H ww}}Xu'@'@'J $ 9 @ @ J LL  j!  % %j 1 3396*3E GGIH H rc ^tjj||}dgi}|D]f\}}t||\}}|j t dd} |dj dj| d|| djhtjj|d|d<d} tjjrdnd} | dk(r td } n| dk(rt|d <td } tj |d 5} t#j$| fi|} | j'| dddtj(|d|t/|fS#1swY,xYw#tj*$r}t-dj|d}~wwxYw)aGenerate Terraform import shell script from template. Args: import_data: string, Import data for each resource. dest_file: string, Filename path to write the generated script to. If dest_file is None, then a default filename will be generated. dest_dir: string, Directory path to write the generated script to. If dest_dir is None, then script will be written to CWD. Returns: tuple(string, int, [string])), the path to the generated script, number of import statements generated and list of files that could not be processed. Raises: TerraformGenerationError: If and error occurs writing to disk/stdout. dataNz${cmd} module.{module_name}.{cmd_sfx}r)rHrbcmd_sfxWINDOWSBASHzwindows_shell_template.tpl bash_commentszbash_shell_template.tplT create_pathError writing import script::{})r%r"r;rcr4_IMPORT_CMD_PREFIXr6r9r5linesepr rLrM_BuildTemplate_BASH_COMMENTSr FileWriterr Contextrender_contextchmodr8rlen)r>rnr`output_file_namecontextr_import_statement_rbimport_cmd_dataoutput_template template_keyfctxrEs rGenerateImportScriptrs$WW\\(I6 RL''2#k#.{HENA{&001CDQRHO FOAHH A "((*I,-(3JJOOGFO4'&//'77AAC,Y$%ABOv-GO$%>?O5   d ,/0 OOA ) )c$$S) ,HH u% 3{+ ++ , , 5 ")003 555s0E:(E.E:.E73E::F, F''F,ctjj||}td}t }|D]'\}}t ||\} } |j | | f)g} |D].} | jtj| d| d0d|i} tjj| | d< tj|d5}tj|fi| }|j|d d d tj |d |t'|fS#1swY,xYw#tj"$r}t%d j|d }~wwxYw) z.Generate terraform modules file from template.zterraform_module_template.tplrrs)rbraprojectmodulesTrxNrzr{)r%r"r;r~setrcaddr6MODULE_TEMPLATEr9r}r rr rrrr8rr)r>rrnr`rrmodule_contentsr_rrarbmodule_declarationsmodulerrrrEs rGenerateModuleFilersXWW\\(I6"#BC/E/#nk1!:X"M;m45$f6!9F1INP  'zz':;')5   * = OOA ) )c$$S) >HH u% 3/ // > = 5 ")003 555s0>D4(D(=D4(D1-D44E&E!!E&ctjjt}tjj |d|}t j |}|S)Nterraform_templates)filename)r%r"r3__file__r;r Template)template_file_namedir_name template_path file_templates rr~r~sD WW__X &('',,x)>13-##];- r)NN)&r __future__rrrr%re"googlecloudsdk.calliope.exceptionsrgooglecloudsdk.corergooglecloudsdk.core.consolergooglecloudsdk.core.utilr r r makor r compiler/r|rNrQrrr*TF_MODULES_FILENAMEr8rrFrRrcrprrr~rrrrsQ&' >#2*.*rzz% '=&  L 1N44N)XO$D/,d08r