S6dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l mZd ZGd d e j ZGddeZGddeZGddeZGddeZGddeZdZdZddZGddeZy)z+Helper methods for record-set transactions.)absolute_import)division)unicode_literalsN)apis) exceptions)yaml)resource_printer)filesztransaction.yamlceZdZdZy)Errorz*Base exception for all transaction errors.N__name__ __module__ __qualname____doc__2lib/googlecloudsdk/api_lib/dns/transaction_util.pyr r "s2rr ceZdZdZy)TransactionFileAlreadyExistsz Transaction file already exists.Nr rrrrr&s(rrceZdZdZy)UnableToAccessTransactionFilez"Unable to access transaction file.Nr rrrrr*s*rrceZdZdZy)TransactionFileNotFoundzTransaction file not found.Nr rrrrr.s#rrceZdZfdZxZS)CorruptedTransactionFileErrorc,tt| dy)NzFCorrupted transaction file. Please abort and start a new transaction.)superr__init__)self __class__s rrz&CorruptedTransactionFileError.__init__4s '7 45r)rrrr __classcell__)r!s@rrr2s 55rrceZdZdZy)RecordDoesNotExistz$Specified record-set does not exist.Nr rrrr$r$:s,rr$c6tj|gd|y)zWrites the given change in yaml format to the given file. Args: yaml_file: file, File into which the change should be written. change: Change, Change to be written out. r) print_formatoutN)r Print) yaml_filechanges rWriteToYamlFiler+>s&IFrcg}|D]\}|j}|j|_|d|_|d|_|d|_|d|_|j |^|S)aFConverts list of record-set dictionaries into list of ResourceRecordSets. Args: messages: Messages object for the API with Record Sets to be created. record_set_dictionaries: [{str:str}], list of record-sets as dictionaries. Returns: list of ResourceRecordSets equivalent to given list of yaml record-sets namettltyperrdatas)ResourceRecordSetkindr-r.r/r0append)messagesrecord_set_dictionaries record_setsrecord_set_dict record_sets r_RecordSetsFromDictionariesr9Hss+0o++-J ooJO%f-JO$U+JN%f-JO(3Jz"1 rcptjd|} tj|xsi}|j d|j d t |j}t||d|_ t||d|_ |S#tj$r t wxYw)a[Returns the change contained in the given yaml file. Args: yaml_file: file, A yaml file with change. api_version: [str], the api version to use for creating the change object. Returns: Change, the change contained in the given yaml file. Raises: CorruptedTransactionFileError: if the record_set_dictionaries are invalid dns additions deletions) rGetMessagesModulerloadYAMLParseErrorrgetChanger9r<r=)r) api_versionr4 change_dictr*s rChangeFromYamlFilerE_s # #E; 7(*))I&,"Kook"*ook"* ' )) ?? &0 K(*&0 K(*& -   * ' ))*s BB5c$eZdZdZddZdZdZy)TransactionFilez7Context for reading/writing from/to a transaction file.ctjj|stdj |||_ |dk(rt j||_y|dk(rt j||_ytdj |#t$r$}d}|j ||}t|d}~wwxYw)NzTransaction not found at [{0}]rwzUnrecognized mode [{}]z.Unable to open transaction [{0}] because [{1}]) ospathisfilerformat!_TransactionFile__trans_file_pathr FileReader_TransactionFile__trans_file FileWriter ValueErrorIOErrorr)r trans_file_pathmodeexpmsgs rrzTransactionFile.__init__s 77>>/ * # * 1 1/ B DD-D / !,,_= 3;!,,_=188>?? / ris 2&' ,*$9*" 3J  3)5)+E+$e$5E5--G.:/f/r