o hH%@s`ddlmZmZmZGdddeZddZddZdd Zd d Z d d Z ddZ ddZ dS)) ListBlock StreamBlock StructBlockcs4eZdZdZdddfdd ZfddZZS)InvalidBlockDefErrorz'Exception for invalid block definitionsN)instancerevisioncs"||_||_tj|i|dS)N)rrsuper__init__)selfrrargskwargs __class___/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/blocks/migrations/utils.pyr szInvalidBlockDefError.__init__csjd}|jdur,|d|jjj|jj7}|jdur%|d|jj|jj7}|jr,|d7}|t 7}|S)Nz#Invalid block def in {} object ({})z# for revision id ({}) created at {} ) rformatr__name__idr created_atr r__str__)r messager rrr s  zInvalidBlockDefError.__str__)r __module__ __qualname____doc__r r __classcell__rrr rrsrcCs ||dkS)Nrr) block_name block_pathrrrshould_alter_blocks rcKst|dkr ||St|trt|f|||d|St|tr-t|f|||d|St|tr>t|f|||d|St d|)ae Maps the value of a block. Args: block_value: The value of the block. This would be a list or dict of children for structural blocks. block_def: The definition of the block. block_path: A '.' separated list of names of the blocks from the current block (not included) to the nested block of which the value will be passed to the operation. operation: An Operation class instance (extends `BaseBlockOperation`), which has an `apply` method for mapping values. Returns: mapped_value: r) operation block_defrzUnexpected Structural Block: ) lenapply isinstancermap_stream_block_valuermap_list_block_valuermap_struct_block_value ValueError) block_valuer!rr r rrrmap_block_value%sB      r*c Ksg}|D]E}t|d|s||qz |j|d}Wnty,td|dwt|df||ddd|}|i|d|iq|S)a Maps each child block in a StreamBlock value. Args: stream_block_value: The value of the StreamBlock, a list of child blocks block_def: The definition of the StreamBlock block_path: A '.' separated list of names of the blocks from the current block (not included) to the nested block of which the value will be passed to the operation. Returns mapped_value: The value of the StreamBlock after mapping all the children. typezNo current block def named {}valueNr!r)rappend child_blocksKeyErrorrrr*)stream_block_valuer!rr mapped_value child_blockchild_block_defmapped_child_valuerrrr%bs*    r%c Ksi}|D]7\}}t||s|||<qz|j|}Wnty)td|wt|f||ddd|}|||<q|S)a Maps each child block in a StructBlock value. Args: stream_block_value: The value of the StructBlock, a dict of child blocks block_def: The definition of the StructBlock block_path: A '.' separated list of names of the blocks from the current block (not included) to the nested block of which the value will be passed to the operation. Returns mapped_value: The value of the StructBlock after mapping all the children. zNo current block def named r-Nr.)itemsrr0r1rr*) struct_block_valuer!rr r3key child_valuer5altered_child_valuerrrr's&     r'cKsPg}t|D]}t|df|j|ddd|}|i|d|iq|S)a  Maps each child block in a ListBlock value. Args: stream_block_value: The value of the ListBlock, a list of child blocks block_def: The definition of the ListBlock block_path: A '.' separated list of names of the blocks from the current block (not included) to the nested block of which the value will be passed to the operation. Returns mapped_value: The value of the ListBlock after mapping all the children. r,r-Nr.)formatted_list_child_generatorr*r4r/)list_block_valuer!rr r3r4r6rrrr&s  r&ccs`d}t|dts d}nd|dvs|dddkrd}|D]}|s'|Vqd|dVqdS)NFrTr+item)r+r,)r$dict)r= is_old_formatchildrrrr<sr<cKs<|dkrg}n|d}|jj}t|f|||d|}|S)a/ Applies changes to raw stream data Args: raw_data: The current stream data (a list of top level blocks) block_path_str: A '.' separated list of names of the blocks from the top level block to the nested block of which the value will be passed to the operation. eg:- 'simplestream.struct1' would point to, [..., { type: simplestream, value: [..., { type: struct1, value: {...} }] }] NOTE: If we're directly applying changes on the top level stream block, then this will be "". NOTE: When the path contains a ListBlock child, 'item' must be added to the block as the name of said child. eg:- 'list1.item.stream1' where the list child is a StructBlock would point to, [ ..., { type: list1, value: [ { type: item, value: { ..., stream1: [...] } }, ... ] } ] operation: A subclass of `operations.BaseBlockOperation`. It will have the `apply` method for applying changes to the matching block values. streamfield: The streamfield for which data is being migrated. This is used to get the definitions of the blocks. Returns: altered_raw_data: r.)r!rr )splitfield stream_blockr*)raw_datablock_path_strr streamfieldr rr!altered_raw_datarrrapply_changes_to_raw_datas/ rJN) wagtail.blocksrrr Exceptionrrr*r%r'r&r<rJrrrrs=*("