'j dddlmZddlmZmZmZmZddlmZm Z er ddl m Z ddl m Z  dddZdS)) annotations)IterableHashable TYPE_CHECKINGOptional) DXFValueErrorDXFAttributeError) DXFEntity)KeyFuncNentitiesIterable[DXFEntity] dxfattribstrkeyOptional[KeyFunc]returndict[Hashable, list[DXFEntity]]cJt|frtddkrfd}|tdt}|D]T}|js ||}n#t$rY#wxYw|+||g}||U|S)ay Groups a sequence of DXF entities by a DXF attribute like ``'layer'``, returns a dict with `dxfattrib` values as key and a list of entities matching this `dxfattrib`. A `key` function can be used to combine some DXF attributes (e.g. layer and color) and should return a hashable data type like a tuple of strings, integers or floats, `key` function example:: def group_key(entity: DXFEntity): return entity.dxf.layer, entity.dxf.color For not suitable DXF entities return ``None`` to exclude this entity, in this case it's not required, because :func:`groupby` catches :class:`DXFAttributeError` exceptions to exclude entities, which do not provide layer and/or color attributes, automatically. Result dict for `dxfattrib` = ``'layer'`` may look like this:: { '0': [ ... list of entities ], 'ExampleLayer1': [ ... ], 'ExampleLayer2': [ ... ], ... } Result dict for `key` = `group_key`, which returns a ``(layer, color)`` tuple, may look like this:: { ('0', 1): [ ... list of entities ], ('0', 3): [ ... ], ('0', 7): [ ... ], ('ExampleLayer1', 1): [ ... ], ('ExampleLayer1', 2): [ ... ], ('ExampleLayer1', 5): [ ... ], ('ExampleLayer2', 7): [ ... ], ... } All entity containers (modelspace, paperspace layouts and blocks) and the :class:`~ezdxf.query.EntityQuery` object have a dedicated :meth:`groupby` method. Args: entities: sequence of DXF entities to group by a DXF attribute or a `key` function dxfattrib: grouping DXF attribute like ``'layer'`` key: key function, which accepts a :class:`DXFEntity` as argument and returns a hashable grouping key or ``None`` to ignore this entity z4Specify a dxfattrib or a key function, but not both.r c8|jS)N)dxf get_default)entityrs G/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/groupby.pyzgroupby..KsVZ33I>>NzMno valid argument found, specify a dxfattrib or a key function, but not both.)allrdictis_aliver setdefaultappend)r rrresult dxf_entity group_keygroups ` rgroupbyr&sp Is   B   B>>>> {    /3ffF % % "   JII     H   %%i44E LL $ $ $ Ms A%% A21A2)r N)r rrrrrrr) __future__rtypingrrrrezdxf.lldxf.constrr ezdxf.entitiesr ezdxf.eztypesr r&rrr-s#""""">>>>>>>>>>>>>>>>>>>>&((((((%%%%%% !PPPPPPPr