OglUbddlmZddlmZmZmZmZmZmZm Z ddl Z ddl Z ddl m Z ddlmZddlmZddlmZmZddlmZGd d ZGd d e jZdd Z ddZGddZdZGddZ ddZ ddZddZ dd dZ!dd!dZ" d" d#dZ#y)$) annotations)IterableIteratorCallableHashableSequenceUnionOptionalN)abc) DXFEntity)groupby)Vec3Vec2)EntityQueryParserc(eZdZddZddZdZdZy)_AttributeDescriptorc||_yN)name)selfrs P/var/www/html/public_html/myphp/venv/lib/python3.12/site-packages/ezdxf/query.py__init__z_AttributeDescriptor.__init__s  Nc8|j|jSr) __getitem__r)robjobjtypes r__get__z_AttributeDescriptor.__get__styy))rc<|j|j|yr) __setitem__r)rrvalues r__set__z_AttributeDescriptor.__set__s  5)rc:|j|jyr) __delitem__r)rrs r __delete__z_AttributeDescriptor.__delete__"s  "r)rstrr)__name__ __module__ __qualname__rrr"r%rrrrs**#rrceZdZdZedZedZedZedZedZ edZ edZ ed Z d- d.d Z d/d Zd0d ZdZdZdZd1dZ d2dZdZdZdZdZdZdZdZdZdZdZdZdZ d3d1dZ!d4d Z"d4d!Z#e$d"Z%e$d#Z& d5 d6d$Z'd5d7d%Z(d5d7d&Z) d8 d9d'Z*d:d(Z+d;d)Z,d;d*Z-d;d+Z.d;d,Z/y )< EntityQuerya EntityQuery is a result container, which is filled with dxf entities matching the query string. It is possible to add entities to the container (extend), remove entities from the container and to filter the container. Query String ============ QueryString := EntityQuery ("[" AttribQuery "]")* The query string is the combination of two queries, first the required entity query and second the optional attribute query, enclosed in square brackets. Entity Query ------------ The entity query is a whitespace separated list of DXF entity names or the special name ``*``. Where ``*`` means all DXF entities, exclude some entity types by appending their names with a preceding ``!`` (e.g. all entities except LINE = ``* !LINE``). All DXF names have to be uppercase. Attribute Query --------------- The attribute query is used to select DXF entities by its DXF attributes. The attribute query is an addition to the entity query and matches only if the entity already match the entity query. The attribute query is a boolean expression, supported operators are: - not: !term is true, if term is false - and: term & term is true, if both terms are true - or: term | term is true, if one term is true - and arbitrary nested round brackets Attribute selection is a term: "name comparator value", where name is a DXF entity attribute in lowercase, value is a integer, float or double quoted string, valid comparators are: - "==" equal "value" - "!=" not equal "value" - "<" lower than "value" - "<=" lower or equal than "value" - ">" greater than "value" - ">=" greater or equal than "value" - "?" match regular expression "value" - "!?" does not match regular expression "value" Query Result ------------ The EntityQuery() class based on the abstract Sequence() class, contains all DXF entities of the source collection, which matches one name of the entity query AND the whole attribute query. If a DXF entity does not have or support a required attribute, the corresponding attribute search term is false. Examples: - 'LINE[text ? ".*"]' is always empty, because the LINE entity has no text attribute. - 'LINE CIRCLE[layer=="construction"]' => all LINE and CIRCLE entities on layer "construction" - '*[!(layer=="construction" & color<7)]' => all entities except those on layer == "construction" and color < 7 layercolorlinetype lineweightltscale invisible true_color transparencyNcd|_d|_||g|_y|dk(rt||_yt |}|Dcgc] }||s |c}|_ycc}w)z Setup container with entities matching the initial query. Args: entities: sequence of wrapped DXF entities (at least GraphicEntity class) query: query string, see class documentation TN*)selected_dxf_attribute ignore_caseentitieslistentity_matcher)rr:querymatchentitys rrzEntityQuery.__init__ss[,.#  DM c\ NDM"5)E2:LeFmVLDMLs AAc,t|jS)zReturns count of DXF entities.lenr:rs r__len__zEntityQuery.__len__s4==!!rc,t|jS)z&Returns iterable of DXFEntity objects.)iterr:rCs r__iter__zEntityQuery.__iter__sDMM""rczt|tr|j|S|jj |S)zReturns DXFEntity at index `item`, supports negative indices and slicing. Returns all entities which support a specific DXF attribute, if `item` is a DXF attribute name as string. ) isinstancer&&_get_entities_with_supported_attributer:r)ritems rrzEntityQuery.__getitem__s4 dC >>tD D}}((..rc^t|ts td|j||y)zOSet the DXF attribute `key` for all supported DXF entities to `value`. +key has to be a string (DXF attribute name)N)rIr& TypeError_set_dxf_attribute_for_all)rkeyr!s rr zEntityQuery.__setitem__s*#s#IJ J ''U3rc\t|ts td|j|y)z@Discard the DXF attribute `key` from all supported DXF entities.rMN)rIr&rN_discard_dxf_attribute_for_all)rrPs rr$zEntityQuery.__delitem__s&#s#IJ J ++C0rch|jDcgc]}|js|c}|_|Scc}w)zRemove destroyed entities.)r:is_alive)res rpurgezEntityQuery.purges)$(MM@qQZZ@  As//c^|jfd|jD}|_|S)Nc3ZK|]"}|jjs|$ywr)dxf is_supported).0rU attributes r zEEntityQuery._get_entities_with_supported_attribute..s' (:(:9(EA s ++) __class__r:r8)rr\r=s ` rrJz2EntityQuery._get_entities_with_supported_attributes3 }}  (1$ rc~|jD]} |jj||!y#t$rY.wxYwr)r:rYsetAttributeError)rrPr!rUs rrOz&EntityQuery._set_dxf_attribute_for_alls@ A  #u% "  s 0 <<c\|jD]}|jj|yr)r:rYdiscard)rrPrUs rrRz*EntityQuery._discard_dxf_attribute_for_alls% A EEMM#  rcp|js td|j|tjS)zEqual selector (self == other). Returns all entities where the selected DXF attribute is equal to `other`. no DXF attribute selected)r8rN_select_by_operatoroperatoreqrothers r__eq__zEntityQuery.__eq__s0 **78 8''x{{;;rcp|js td|j|tjS)zNot equal selector (self != other). Returns all entities where the selected DXF attribute is not equal to `other`. re)r8rNrfrgneris r__ne__zEntityQuery.__ne__s0**78 8''x{{;;rct|js td|j|tjdS)zLess than selector (self < other). Returns all entities where the selected DXF attribute is less than `other`. Raises: TypeError: for vector based attributes like `center` or `insert` reFvectors)r8rNrfrgltris r__lt__zEntityQuery.__lt__5**78 8''x{{E'JJrct|js td|j|tjdS)zGreater than selector (self > other). Returns all entities where the selected DXF attribute is greater than `other`. Raises: TypeError: for vector based attributes like `center` or `insert` reFrp)r8rNrfrggtris r__gt__zEntityQuery.__gt__rtrct|js td|j|tjdS)zLess equal selector (self <= other). Returns all entities where the selected DXF attribute is less or equal `other`. Raises: TypeError: for vector based attributes like `center` or `insert` reFrp)r8rNrfrgleris r__le__zEntityQuery.__le__rtrct|js td|j|tjdS)zGreater equal selector (self >= other). Returns all entities where the selected DXF attribute is greater or equal `other`. Raises: TypeError: for vector based attributes like `center` or `insert` reFrp)r8rNrfrggeris r__ge__zEntityQuery.__ge__rtrcPt|tr|j|StS)z"Union operator, see :meth:`union`.)rIr,unionNotImplementedris r__or__zEntityQuery.__or__s! e[ )::e$ $rcPt|tr|j|StS)z0Intersection operator, see :meth:`intersection`.)rIr, intersectionrris r__and__zEntityQuery.__and__s# e[ )$$U+ +rcPt|tr|j|StS)z,Difference operator, see :meth:`difference`.)rIr, differencerris r__sub__zEntityQuery.__sub__ s! e[ )??5) )rcPt|tr|j|StS)z@Symmetric difference operator, see :meth:`symmetric_difference`.)rIr,symmetric_differencerris r__xor__zEntityQuery.__xor__s# e[ ),,U3 3rc>|j}|jr t|tr|j }|j }||_|j }|r|j D]} |jj|}|s:t|ttfr$tdt|jd||jr t|tr|j }|||s|j||S#t$rYwxYw)Nzunsupported operation 'z' for DXF attribute )r8r9rIr&lowerr^r:rY get_defaultrarrrNr'append) rr!oprqr\r=r:r? entity_values rrfzEntityQuery._select_by_operators //    5# 6KKME '0$>> -- ,#)::#9#9)#DL:lT4L#I#1#bkk2B1CD%%.K1## <(E#/#5#5#7LlE*OOF+ , &s.D DDc2fd}j||S)zReturns all entities where the selected DXF attribute matches the regular expression `pattern`. Raises: TypeError: for non-string based attributes czt|tr|j|duStdj)Nz2cannot apply regular expression to DXF attribute: )rIr&r>rNr8)r!regexrs rr>z EntityQuery.match..match;sB%%{{5)55D../1 r) _regex_match)rpatternr>s` rr>zEntityQuery.match2s   %00rc|j}d|_|rtjnd}|jds|dz }|j tj |||}||_|S)NFr$flags)r9re IGNORECASEendswithrfcompile)rrfuncr9re_flagsresults rrzEntityQuery._regex_matchEsj&&  $/2==Q$ sNG)) JJwh / ' rcLt|jr|jdSy)zFirst entity or ``None``.rNrArCs rfirstzEntityQuery.firstSs" t}} ==# #rcLt|jr|jdSy)zLast entity or ``None``.NrArCs rlastzEntityQuery.last[s" t}} ==$ $rcf|j|j||j|_|S)zeExtent the :class:`EntityQuery` container by entities matching an additional query. )rr^r:)rr:r=s rextendzEntityQuery.extendcs* 4>>(E#BCLL  rcz|j|j|j|j|_|S)ziRemove all entities from :class:`EntityQuery` container matching this additional query. )rr^r:rr=s rremovezEntityQuery.removeos5  NN4==% 0 (  rc:|j|j|S)zReturns a new :class:`EntityQuery` container with all entities matching this additional query. Raises: pyparsing.ParseException: query string parsing error )r^r:rs rr=zEntityQuery.queryys~~dmmU33rc0t|j||S)aReturns a dict of entity lists, where entities are grouped by a DXF attribute or a key function. Args: dxfattrib: grouping DXF attribute as string like ``'layer'`` key: key function, which accepts a DXFEntity as argument, returns grouping key of this entity or ``None`` for ignore this object. Reason for ignoring: a queried DXF attribute is not supported by this entity )r r:)r dxfattribrPs rr zEntityQuery.groupbys t}}i55rcL|jt||jS)aReturns a new :class:`EntityQuery` with all entities from this container for which the callable `func` returns ``True``. Build your own operator to filter by attributes which are not DXF attributes or to build complex queries:: result = msp.query().filter( lambda e: hasattr(e, "rgb") and e.rbg == (0, 0, 0) ) )r^filterr:)rrs rrzEntityQuery.filters~~fT4==9::rcv|jt|jt|jzS)zReturns a new :class:`EntityQuery` with entities from `self` and `other`. All entities are unique - no duplicates. r^r`r:ris rrzEntityQuery.union*~~c$--03u~~3FFGGrcv|jt|jt|jzS)z_Returns a new :class:`EntityQuery` with entities common to `self` and `other`. rris rrzEntityQuery.intersectionrrcv|jt|jt|jz S)zjReturns a new :class:`EntityQuery` with all entities from `self` that are not in `other`. rris rrzEntityQuery.differencerrcv|jt|jt|jz S)zkReturns a new :class:`EntityQuery` with entities in either `self` or `other` but not both. rris rrz EntityQuery.symmetric_differencerrNr7)r:Optional[Iterable[DXFEntity]]r=r&)returnint)rIterator[DXFEntity])rr,)r\r&rr,)T)rr&rr,r7)r:Iterable[DXFEntity]r=r&rr,)r=r&rr,)r6N)rr&rPz)Optional[Callable[[DXFEntity], Hashable]]rzdict[Hashable, list[DXFEntity]])rCallable[[DXFEntity], bool]rr,)rjr,rr,)0r'r(r)__doc__rr-r.r/r0r1r2r3r4rrDrGrr r$rVrJrOrRrkrnrsrwrzr}rrrrrfr>rpropertyrrrrr=r rrrrrr*rrr,r,&sAF ! )E  )E#J/H%l3J"9-G$[1I%l3J'7LLOM5MEHM2"#/41   << K K K K    21&  %    49=6676 ) 6$ ;H H H Hrr,ctj|d}t|jt |j |j dfd }|S)NT)parseAllc(|xr|Srr*)r?attrib_matcherentity_matcher_s rmatcherzentity_matcher..matchersv&A>&+AArr?r rbool)r parseStringbuild_entity_name_matcherr,build_entity_attributes_matcher AttribQueryAttribQueryOptions)r= query_argsrrrs @@rr<r<sM"..utDJ/ 0F0FGO4 = =NB NrcJdfd }tdj||S)Nc0|jSr)dxftype)rU_matchs rr>z(build_entity_name_matcher..matchsaiik""r )r=)rUr rr) name_matcherjoin)namesr>rs @rrrs!# 0F Lrc eZdZejej ej ejejejdddZ e e jZ ddZddZy) Relationc(|j|duSrr>rUrs rzRelation.sekk!nD8rc(|j|duSrrrs rrzRelation.su{{1~5r)z==z!=z>=?z!?c |\}}}||_tj||_|rtnd|_|rt jnd}d|vr t j|dz||_ y|j ||_ y)Nc|Srr*xs rrz#Relation.__init__..s1rrrrr) dxf_attribr CMP_OPERATORScompareto_lower convert_caserrrr!)rrelationr9rrr!rs rrzRelation.__init__ss"b%--b1 (3H$/2==Q "9 8DJ**51DJrc |j|jj|j}|j ||j S#t $rYyt$rYywxYw)NF)rrYrrrr!ra ValueError)rr?r!s revaluatezRelation.evaluates\ %%fjj&<&t|dr|jS|S)Nr)hasattrr)r!s rrrs#E735;;=>>rcTeZdZejej dZddZdZddZ y)BoolExpression)&|c||_yrtokens)rrs rrzBoolExpression.__init__s  rc,t|jSr)rFrrCs rrGzBoolExpression.__iter__sDKK  rct|jtr|jj|Sg}g}|jD]@}t |dr!|j |j|0|j |B|j |D]_}|dk(r|j }n5tj||j|j}|j |a|jS)Nr!) rIrrrrrreversepopr OPERATORS)rr?values operatorstokenrr!s rrzBoolExpression.evaluates  KK ;;''/ / [[ (Euj) ennV45  '  (  !BSy"JJL(&004VZZ\6::<P MM%  ! zz|rN)rrr) r'r(r)rgand_or_rrrGrr*rrrrs& ]] \\I !rrc dd}t|tr|St|}||r t||St |Dcgc]}t ||c}Scc}w)NcLt|dk(xr|dtjvS)N)rBrrrs r is_relationz$_compile_tokens..is_relation!s%6{aMF1I1M1M$MMr)rrrr)rIr&tuplerr_compile_tokens)rr9rrs rr r s\N&# 6]F6 ,,>D EU_UK 0 E  EsAcdt|sdSd|k(}tt||dfd }|S)Ncy)NTr*rs rrz1build_entity_attributes_matcher..4sric&j|Sr)r)r?exprs rmatch_bool_exprz8build_entity_attributes_matcher..match_bool_expr8s}}V$$rr)rBrr )roptionsr9rrs @rrr0s9 v;.K /&+> ?D% rc#bKt}|D]}||vs|j||yw)zCYield all unique entities, order of all entities will be preserved.N)r`add)r:doner?s runique_entitiesr>s55D   HHV Ls//c0t|fd|DS)aFilters `names` by `query` string. The `query` string of entity names divided by spaces. The special name "*" matches any given name, a preceding "!" means exclude this name. Excluding names is only useful if the match any name is also given (e.g. "LINE !CIRCLE" is equal to just "LINE", where "* !CIRCLE" matches everything except CIRCLE"). Args: names: iterable of names to test query: query string of entity names separated by spaces Returns: yield matching names c34K|]}|s |ywrr*)r[rr>s rr]zname_query..Vs 2TeDkD 2s )r)rr=r>s @r name_queryrGs  E 2U 22rc"dfd }t|jj}dtt|D]A}|dk(rd |jdrj |dd1j |C|S)Ncr|vS|vSrr*)rUexcludeincludetake_alls rr>zname_matcher..matchZs G# #< rFr7Trr)rUr&rr)r`uppersplit startswithr)r=r> match_stringsrrrrs @@@rrrYs  ++-.MHeGeG 3;H __S ! KKQR ! KK   Lrct||S)zStart a new query based on sequence `entities`. The `entities` argument has to be an iterable of :class:`~ezdxf.entities.DXFEntity` or inherited objects and returns an :class:`EntityQuery` object. )r,)r:r=s rnewr#os x ''r)r=r&rr)rz Sequence[str]rr)rzUnion[str, Sequence]r9rrz$Union[str, Relation, BoolExpression])rrrr&rr)r:rrrr)rz Iterable[str]r=r&rz Iterator[str])r=r&rzCallable[[str], bool]r)r:rr=r&rr,)$ __future__rtypingrrrrrr r rrg collectionsr ezdxf.entities.dxfentityr ezdxf.groupbyr ezdxf.mathrrezdxf.queryparserrrr,r<rrrrr rrrrr#r*rrr+s# .!!/ # #RH#,,RHj   ""J?  F  /3 ) $   "   3$.BE(+(;>((r