h/|ddlmZddlZddlZddlmZmZmZmZmZm Z m Z ddl Z ddl Z ddl mZddlmZddlZddlmZddlmZddlmZmZmZmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*m+Z+er ddl,m-Z-ddl.m/Z/ddgZ0hdZ1ejddZ3 d, d-dZ4d,d.dZ5 d/ d-dZ6d0dZ7GddZ8d1dZ9d2dZ:d3dZ; d4 d5dZ<e jzdZ>e jzd Z?d6d!Z@e jzd"ZAe jzd#ZBd7d$ZCejd8d%ZEd8d&ZFd'ZGd(ZHd9d)ZId:d*ZJejejddf d;d+ZLy)<) annotationsN) TYPE_CHECKINGBinaryIOIterableIteratorCallableUnionOptional) defaultdict)Path)const)repair)has_dxf_unicodedecode_dxf_unicodehas_mif_encodingdecode_mif_to_unicode)DXFTag DXFVertex DXFBinaryTag POINT_CODES BINARY_DATA TYPE_TABLEMAX_GROUP_CODE) group_tagsTags)entity_structure_validator) toencoding)Auditor AuditErrorDrawing) SectionDictreadreadfile>EOFTABLEENDBLKENDSECENDTABSEQENDSECTIONezdxfsurrogateescapect|}t|d5}t||\}}ddd|_|fS#1swYxYw)uRead a DXF document from file system similar to :func:`ezdxf.readfile`, but this function will repair as many flaws as possible, runs the required audit process automatically the DXF document and the :class:`Auditor`. Args: filename: file-system name of the DXF document to load errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: for invalid or corrupted DXF structures UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs rbmodeerrorsN)stropenr#filename)r6r3fpdocauditors -lib/python3.12/site-packages/ezdxf/recover.pyr$r$>sH(8}H hT "bBv. W #CL < # "s =AcFtj||}t|S)u(Read a DXF document from a binary-stream similar to :func:`ezdxf.read`, but this function will detect the text encoding automatically and repair as many flaws as possible, runs the required audit process afterwards and returns the DXF document and the :class:`Auditor`. Args: stream: data stream to load in binary read mode errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: for invalid or corrupted DXF structures UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs r2)Recoverrun_load_and_audit_document)streamr3 recover_tools r:r#r#Ys!&;;vf;5L #L 11ct|}t|d5}tj||t}t |\}}ddd|_|fS#1swYxYw)uURead a DXF document from file system similar to :func:`readfile`, but this function will use a special tag loader, which tries to recover the tag stream if invalid tags occur. This function is intended to load corrupted DXF files and should only be used to explore such files, data loss is very likely. Args: filename: file-system name of the DXF document to load errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: for invalid or corrupted DXF structures UnicodeDecodeError: if `errors` is "strict" and a decoding error occurs r/r0)r3loaderN)r4r5r<r=synced_bytes_loaderr>r6)r6r3r7r@r8r9s r:explorerEps],8}H hT "b{{2f=P{Q / = W #CL < # "s +AA c(ddlm}|}|j|jt |}|j D]\}}|j |||jD]\}}|j|||j||fS)Nrr ) ezdxf.documentr!_load_section_dict section_dictrr3 add_errorfixes fixed_errorr=)r@r!r8r9codemsgs r:r>r>s& )C<445clG!(( c$$)!'' cD#&( KKM <rAceZdZdZd ddZe d ddZddZddZddZ ddZ dd Z dd Z d Z d Zy)r<zLoose coupled recovering tools.Nc|xst|_t|_g|_g|_t j|_yN) bytes_loader tag_loaderdictrIr3rKr DXF12 dxfversion)selfrCs r:__init__zRecover.__init__s9 !0L,06.0 ,.  ++rAct|}|j||}|j|}|j||jj d}|r |j |}||jd<|jdkDr |j|j|j}|jdk} |jD](\} } | dvs t|j| | || <*|S)zExecute the recover process.TABLESAC1009>BLOCKSrZOBJECTSENTITIES) r< load_tagsrebuild_sectionsload_section_dictrIgetrebuild_tablesrVrecover_rootdictfix_broken_layout_linksitemslistcheck_entities) clsr?rCr3r@tagssectionstablesrIis_r12nameentitiess r:r=z Recover.runsv %%ff5006&&x0**..x8 !008F28L % %h /  " "X -  ) ) +  0 0 2#00 ((H4*002ND(BB%) //&A& T"3 rAcHt||j|j|S)Nmessagesr3)safe_tag_loaderrSr3)rWr?r3s r:r_zRecover.load_tagss  DOOdkk&  rAc  fd fd fd} fdg g gd|D] \ dk(r| j  S)aACollect tags between SECTION and ENDSEC or next SECTION tag as list of DXFTag objects, collects tags outside of sections as an extra section. Returns: List of sections as list of DXFTag() objects, the last section contains orphaned tags found outside of sections crjn+jjtjdfgdy)Nz)DXF structure error: missing SECTION tag.F)appendrKrMISSING_SECTION_TAG) collectorinside_sectionrkrWsr: close_sectionz/Recover.rebuild_sections..close_sectionsE * !!"66C I"NrAcr2jjtjdfjdy)N(DXF structure error: missing ENDSEC tag.TrKrvrMISSING_ENDSEC_TAG)rzrxryrWtagsr: open_sectionz.Recover.rebuild_sections..open_sectionsC !!"55B    S !!NrAcdk(rydk(rydk(r6r3jjtjdfyyy)Nr+r(r%r|r})rzcollectryrrWvaluesr:process_structure_tagz7Recover.rebuild_sections..process_structure_tags] !("%!JJ%%&99F "O" rAcrjyjjtjdddfjy)Nz1DXF structure error: found tag outside section: (z, ))rvrKrFOUND_TAG_OUTSIDE_SECTION)rMrxryorphansrWrrsr:rz)Recover.rebuild_sections..collectsY  % !!"<< 6E7!-s#rAFrrv) rWrjrrzrMrrxryrrrkrrs ` @@@@@@@@@@r:r`zRecover.rebuild_sectionssu # " "  " $ $!#')"$ CKD%qy%'   rAc d fd }d fd }dfd }|j}|r|ddk(r|j|g}t |D]D}|d\}}|dk(r |||jjtj dfF j d tdd tdd g} j| |t| _ jtjkr| | y )zMerge sections of same type.cH|vr|j|ddy||<y)N)extend)rnrjrIs r: add_sectionz.Recover.load_section_dict..add_section-s.|#T"))$qr(3%) T"rAc|jD]:\}}|tjvstt |dj |<<yNr)rfr MANAGED_SECTIONSrgrrI)drnsectionrWs r:_build_section_dictz6Recover.load_section_dict.._build_section_dict3sA!" g5111.2:gq3I.JD%%d+"+rAcdD]9}||vs||=jjtjd|df;y)N)CLASSESr]ACDSDATAzRemoved unsupported z section for DXF R12.)rKrvrREMOVED_UNSUPPORTED_SECTION)rrnrWs r:_remove_unsupported_sectionsz?Recover.load_section_dict.._remove_unsupported_sections8sH:19$JJ%%&BB24&8MN;rAr)rr+rz>DXF structure error: missing section name tag, ignore section.HEADERr+N)rnr4returnNone)rrTrr)rrT) poprvrTrKrMISSING_SECTION_NAME_TAG setdefaultrrescue_orphaned_header_vars_detect_dxf_versionrVr rU) rWrkrrrrrrMrnheaderrIs ` @r:razRecover.load_section_dict*s *  K ,,. wqz^3 OOG $G&*f G JD$qyD'* !!";;X  (( q)$q(#   ((9-f5 ??ekk ) ( 6L)rAc d  fd }t ttttj }D]a}|dj j}|dk(r$ |dj j}| |<I||vsN|j|cttddtddggttj }|jtjkr@|jddvr+|j jt"j$d f|D] }|| S#ttf$rYwxYw) zRebuild TABLES section.c ,|vryj|}|rj|n0jttddtd|gj |jttddgy)Nrr&rr))rbrvrrr)rnheadcontentheadsrls r: append_tablez,Recover.rebuild_tables..append_tableksz7"99T?D d# dF1g$6q$#HIJ MM'$- ( MM$q( 345 6rArr&rr+rrZ BLOCK_RECORDz3Removed unsupported BLOCK_RECORD table for DXF R12.)rnr4)rTr rgsetr TABLE_NAMES_ACAD_ORDERrupper IndexErrorAttributeErrorrvrrrVrUremoverKrREMOVED_UNSUPPORTED_TABLE) rWrlr valid_tablesentryrn table_namenamesrrs ` @@r:rczRecover.rebuild_tablesds> 7d#5778 E8>>'')Dw.!&q!5!5!7J).E*%% $$U+q),fQ.ABCDU112 ??ekk ) LL (( !!"<<MD   -#N3s&E  E E c~d}|D]6}|\}}|dk(r|}||j||j|d}8y)N r)rWrrvar_namerrMrs r:rz#Recover.rescue_orphaned_header_varssKCKD%qy% h' c"rAc#Kd}|D]B}|d\}}|tvr|tt|}|r|j||Dyw)N)dr)EXCLUDE_STRUCTURE_CHECKrr remove_tags)rWrormsubclass_markersentity_dxftyperjs r:rhzRecover.check_entitiess\!FJAw11 6v>?$$%56 sA A cl|jjd}|rt|dkryt|dryt |\}}|rN|d||<||d< |j }|jjtjd|dfyy#t j$rd}YHwxYw)Nr]rrrz%Recovered misplaced root DICTIONARY(#z).) rIrblen _is_rootdict_find_rootdict get_handler DXFValueErrorrKrvrMISPLACED_ROOT_DICT)rWobjectsindexrootdicthandles r:rdzRecover.recover_rootdicts##'' 2#g,*   # (1x $QZGEN!GAJ !,,. JJ  22;F82F   &&  sBB32B3cy)zFixes broke links (block_record_handle) between LAYOUT and BLOCK_RECORD entities. See issue #997 for more information. N)rWs r:rezRecover.fix_broken_layout_linkss rArQ)rCOptional[Callable])Nr-)r?rrCrr3r4rr<)r?rr3r4rIterator[DXFTag])rjIterable[DXFTag]rlist[list[DXFTag]])rkrrr)rl list[Tags]rr)rz list[DXFTag]rrrr)rorrmboolrzIterator[Tags])__name__ __module__ __qualname____doc__rX classmethodr=r_r`rarcrrhrdrerrAr:r<r<s)&"&*' #   : Sj8*t3j  "  -=     . rAr<cd}|D]S}|rGt|dj}tjd|r|cStj S|dk(sRd}Utj S)NFrz AC[0-9]{4})rz$ACADVERT)r4stripre fullmatchr rU)rnext_is_dxf_versionrrVs r:rrsj SV**,J||M:6!! ;; / !"&  ;;rAc8|ddk7rytd|DS)Nr)r DICTIONARYFc3&K|] }|dk( yw)) ACAD_GROUPNr).0rs r: z_is_rootdict..s:6Cs''6s)any)rs r:rrs# ay%% :6: ::rAcbt|D]\}}t|s||fcSdtfSr) enumeraterr)rrrs r:rrs5"7+ v  &= , df9rAc|t}tj||d\}}t|}t j |}t j |}t j|}t||||S)uYields :class:``DXFTag`` objects from a bytes `stream` (untrusted external source), skips all comment tags (group code == 999). - Fixes unordered and invalid vertex tags. - Pass :func:`synced_bytes_loader` as argument `loader` to brute force load invalid tag structure. Args: stream: input data stream as bytes loader: low level tag loader, default loader is :func:`bytes_loader` messages: list to store error messages errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data rrq) rR itertoolsteedetect_encodingrtag_reorder_layerfilter_invalid_point_codesfilter_invalid_handlesbyte_tag_compiler)r?rCrrr3rjdetector_streamencodings r:rsrsst0~%MM&.!<D//H  # #D )D  , ,T 2D  ( ( .D T8hv NNrAz[+-]?\d+s[+-]?\d+ctjt|trtnt |}|r|j }t|S)zEmulate the behavior of the C function stoll(), which just stop converting strings to integers at the first invalid char without raising an exception. e.g. "42xyz" is a valid integer 42 )rsearch isinstancer4 INT_PATTERN_S INT_PATTERN_Bgroupintsress r: _search_intrs; ))#As+  C IIK q6MrAz#[+-]?\d+(:?\.\d*)?(:?[eE][+-]?\d+)?s#[+-]?\d+(:?\.\d*)?(:?[eE][+-]?\d+)?ctjt|trtnt |}|r|j }t|S)zEmulate the behavior of the C function stod(), which just stop converting strings to doubles at the first invalid char without raising an exception. e.g. "47.11xyz" is a valid double 47.11 )rrrr4FLOAT_PATTERN_SFLOAT_PATTERN_Brfloatrs r: _search_floatr)s; ))%a-?A C IIK 8OrAc #Kd}d}|j}|s[|}|r t|}ny |}|r6|j d }|d k(r|d k(rd }|dk7rt|||dz }ny |sZy y #t$rZ t|}nJ#t$r>|j dj d}t jd|d|dwxYwYwxYww)aYields :class:``DXFTag`` objects from a bytes `stream` (untrusted external source), skips all comment tags (group code == 999). ``DXFTag.code`` is always an ``int`` and ``DXFTag.value`` is always a raw bytes value without line endings. Works with file system streams and :class:`BytesIO` streams. Raises: DXFStructureError: Found invalid group code. Frignorer2z zInvalid group code "z " at line .N rsEOFTr) readliner ValueErrorrdecoderstripr DXFStructureErrorr)r?eoflinerrMrs r:rRrR7s C DHz  4y   LL)EqyUf_s{T5)) AID 7  &t,D!;;h;7>>vFD11.tfJtfAF sGC A1AC/C1 C; BCACCCCCc#Kd}tdz}|j} d}|r*|}|r t|}d|cxkr|krnnd}ny|r*|}|r#|dk7rt ||j dny\#t$rY1380'  s{T5<<#899 ' "s-&B A7B .B7 BBBBs $DWGCODEPAGEs$ACADVERcttj}dj|Dcgc] }||vs| c}Scc}w)N)rstring whitespacejoin)rwscs r:_strip_whitespacers9 V   B 77q0q!ARKAq0 110s >>cd}d}d}|D]\}}|dk(r |tk(rt}nx|tk(rot}nh|dk(r4|tk(r+t|jtj }d}n/|dk(r*|tk(r!|jtj }d}|s|s|tj k\rdcS|cStj S)aDetect text encoding from header variables $DWGCODEPAGE and $ACADVER out of a stream of DXFTag objects. Assuming a malformed DXF file: The header variables could reside outside of the HEADER section, an ENDSEC tag is not a reliable fact that no $DWGCODEPAGE or $ACADVER header variable will show up in the remaining tag stream. Worst case: DXF file without a $ACADVER var, and a $DWGCODEPAGE unequal to "ANSI_1252" at the end of the file. Nrrrutf8) DWGCODEPAGEACADVERrr r DEFAULT_ENCODINGDXF2007)rjrrVnext_tagrMrs r:rrsHJH e 19 #&'!" QY8{2!%,,u/E/E"FGHH QY8w.e&<&<=JH  '5==86 Fh F  ! !!rAc#Kfd}dfd }dfd }ttsJt|tsJgt|}d}d ||}d}nt|}dz |j} | t vrt|} dz | j| dzk7rt jdd t|} dz  | j| d zk(r@ t|jt| jt| jf} n- t|jt| jf} | }t| | n| tvr9t|tr|} n! tj| |j} | ntj | t}|j}|tur| dk(r(|jj#j%} |j'd }| r-t1|r t3|}nt5|r t7|}t9| |n t9| || #t$r9||j|| j|| jf} YVwxYw#t$r(||j|| jf} YwxYw#t$rt jd d wxYw#t$rt jd d wxYw#t($r;|j'|}j+t,j.dfY]wxYw#t$r|t:urH t9| ||jn#t$rt j||wxYw|turH t9| ||jnC#t$rt j||wxYwt j||YwxYw#t<$rYywxYww)u`Compiles DXF tag values imported by bytes_loader() into Python types. Raises DXFStructureError() for invalid float values and invalid coordinate values. Expects DXF coordinates written in x, y[, z] order, see function :func:`safe_tag_loader` for usage with applied repair filters. Args: tags: DXF tag generator, yielding tag values as bytes like bytes_loader() encoding: text encoding messages: list to store error messages errors: specify decoding error handler - "surrogateescape" to preserve possible binary data (default) - "ignore" to use the replacement char U+FFFD "�" for invalid data - "strict" to raise an :class:`UnicodeDecodeError` exception for invalid data Raises: DXFStructureError: Found invalid DXF tag or unexpected coordinate order. cj|j}|jj}d|d|ddS)Nz Invalid tag (z, "z") near line: r)rMrr )rrMrrr s r: error_msgz$byte_tag_compiler..error_msgs;xx   *tfCwnTF!DDrAct|tr|jdd}t|}d|dd|d}j t j |ftj||S)Nrrrr3z!recovered invalid integer value " " near line  as "") rbytesr rrvrINVALID_INTEGER_VALUEloggerwarningrrrNr rrs r: recover_intz&byte_tag_compiler..recover_intsk a &:AA1!LeE7RST993?@s rAct|tr|jdd}t|}t |}d|dd|d}j t j|ftj||S)Nrrr$z(recovered invalid floating point value "r%r&r') rr(r rrrvrINVALID_FLOATING_POINT_VALUEr*r+r,s r: recover_floatz(byte_tag_compiler..recover_floatsx a &:A a a 8<vUSXRYYZ[@@#FGs rANrr z)Missing required y-coordinate near line: rz)Invalid floating point values near line: zInvalid binary data near line: strictr2z'Fixed unicode decoding error near line rUnion[str, bytes]rrrr5rr)rr4iternextrMrr r rrrrrr from_stringrrbrrr UnicodeDecodeErrorrvrDECODING_ERRORrrrrrr StopIteration)rjrrrr3r"r-r0undo_tagxrMyzpointrtype_rstr_r s `` @r:rrs<E  h $$ $ fc "" " :DH D j #J 66D{"J 66TBY&11CD6KJ vv*  %agg %agg %agg%E%*177^U177^$DE $%  e,,$a.C*66tQWWE  "tS1 wwC<qy !  5 5 7 $||HX|F*40#5d#;D-d3#8#>D t,,H$T5<88w : * -agg 6 -agg 6 -agg 6%E * -agg 6 -agg 6%E "11CD6K&#55=dV1E.$||HV|D  * 9 9"I$ P,& H C<L&,T;qww3G&H H#-L&+&=&=il&K KL"e^L&,T=3I&J J#-L&+&=&=il&K KL#("9"9)A,"GG H   sA QA6P5 K>IK*J#K +P55 K=A P56L# >P5M*Q>J KJ  K#-KKKK#K::P5=#L  P5#AM'#P5&M''P5*P2<NP2%O P2 O+*P2+%PP2.P51P22P55 Q>QQQ)r-)r6zUnion[str, Path]r3r4rtuple[Drawing, Auditor])r?rr3r4rrD)r)rrD)rrgrr4)rrrr)rrrztuple[int, Tags])NNr-) r?rrCrrrOptional[list]r3r4rrr4r6)r?rrr)rr4rr4)rjrrr4)rjrrrrEr3r4rr)M __future__rrtypingrrrrrr r rr collectionsr pathlibr logging ezdxf.lldxfr rezdxf.lldxf.encodingrrrrezdxf.lldxf.typesrrrrrrrezdxf.lldxf.tagsrrezdxf.lldxf.validatorrezdxf.tools.codepager ezdxf.auditrrrGr! ezdxf.eztypesr"__all__r getLoggerr*r$r#rEr>r<rrrrscompilerrrrrr no_type_checkrRrDrrrrrrrrAr:rWs#  # .<++&) :    7 #/@(+620/7(+<  r r j ;"&## !O !O !O!O  !O  !OH ;'  <(  "**CD"**DE **Z%P 2 ""J # ### h hh  h  hhrA