sPjdZddgZddlmZddlmZddlmZddlmZm Z dd l m Z er dd l m Z dd lmZGd dZGd dZy)zBom related utilitiesBomRefDiscriminatorBomDependencyGraphFlatMerger)Iterable)chain)random) TYPE_CHECKINGAny Dependency)BomBomRefceZdZdZddeddeddfdZddZd ed ed eddfd Z dd Z ddZ defdZ e ddddeddfdZy)ru Ensure that a collection of BomRef objects has unique, non‑empty :attr:`cyclonedx.model.bom_ref.BomRef.value`. The discriminator inspects each provided BomRef and assigns a newly generated identifier to any instance whose ``value`` is missing or duplicates an earlier one. All original values are preserved and can be restored via :meth:`reset()` or by using this class as a context manager. rbomrefsprefixreturnNc@td|D|_||_y)Nc38K|]}||jfywN)value).0bomrefs `/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/cyclonedx/contrib/bom/utils.py z/BomRefDiscriminator.__init__..4sKvv||4Ks)tuple_bomrefs_prefix)selfrrs r__init__zBomRefDiscriminator.__init__1sK7KK  c$|jyr) discriminaters r __enter__zBomRefDiscriminator.__enter__7s r!exc_typeexc_valexc_tbc$|jyrresetrr&r'r(s r__exit__zBomRefDiscriminator.__exit__:  r!cg}|jD]?\}}|j}|||vr|j}||_|j|Ay)z Enforce uniqueness across all :attr:`cyclonedx.model.bom_ref.BomRef.value`s. Any BomRef whose ``value`` is ``None`` or duplicates a previously encountered value is assigned a newly generated unique identifier. N)rr _make_uniqueappend)r known_valuesr_rs rr#z BomRefDiscriminator.discriminate=sX  'IFALLE} 5))+$    &  'r!c:|jD] \}}||_y)zl Restore all :attr:`cyclonedx.model.bom_ref.BomRef.value`s to their original state. N)rr)rroriginal_values rr+zBomRefDiscriminator.resetMs# '+mm * "FN)FL *r!cv|jttddttddS)N)rstrrr$s rr0z BomRefDiscriminator._make_uniqueUs5,,FH ab 123vx=3D2EFFr!bomr c|td|jDd|jDd|jD|S)aE Create a discriminator for all :class:`cyclonedx.model.bom_ref.BomRefs` contained within a Bom. This includes BomRefs from * :attr:`cyclonedx.model.bom.Bom.components` * :attr:`cyclonedx.model.bom.Bom.services` * :attr:`cyclonedx.model.bom.Bom.vulnerabilities` c34K|]}|jywrbom_ref)rcs rrz/BomRefDiscriminator.from_bom..ds :1QYY :c34K|]}|jywrr<)rss rrz/BomRefDiscriminator.from_bom..es -1QYY -r?c34K|]}|jywrr<)rvs rrz/BomRefDiscriminator.from_bom..fs 41QYY 4r?)r_get_all_componentsservicesvulnerabilities)clsr9rs rfrom_bomzBomRefDiscriminator.from_bomXsH5 : 7 7 9 : - - 4 3 3 4    r!rrN)__name__ __module__ __qualname____doc__rr8r r%r r-r#r+r0 classmethodrHr!rrr%s  2Ct sCD' *GcG5#=Rr!cneZdZdZddZddZdedededdfd Zdd Zdd Z e d e e de e fd Z y)ruh Context‑manager utility that temporarily flattens and merges all :attr:`cyclonedx.model.bom.Bom.dependencies`. When used as a context manager, the :class:`cyclonedx.model.bom.Bom`'s dependency graph is replaced with a flattened, merged representation for the duration of the ``with`` block and automatically restored afterward. rNcH||_|jj|_yr)_bom _dependencies_deps)rr9s rr z%BomDependencyGraphFlatMerger.__init__us YY,, r!c$|jyr) flatten_merger$s rr%z&BomDependencyGraphFlatMerger.__enter__zs r!r&r'r(c$|jyrr*r,s rr-z%BomDependencyGraphFlatMerger.__exit__}r.r!cX|j|j|j_y)um Flatten and merge all :attr:`cyclonedx.model.bom.Bom.dependencies`. This produces a non‑recursive, merged representation of the entire dependency graph and assigns it to the Bom. .. note:: The original dependency graph is not modified. A new, flattened dependency structure is assigned to the Bom. N)_flatten_mergerTrR dependenciesr$s rrVz*BomDependencyGraphFlatMerger.flatten_merges"&!4!4TZZ!@ r!c:|j|j_y)a  Restore the :class:`cyclonedx.model.bom.Bom`'s dependency graph to its original state. .. note:: This does not modify the dependency graph. It simply reassigns the original dependency collection back to the Bom. N)rTrRrSr$s rr+z"BomDependencyGraphFlatMerger.resets#'** r!depscpi}t|}t}|r|j}t|x}|vr"|j ||j |j g}|jx}r)|jd|D|j||rd|jDS)Nc34K|]}|jywr)refrds rrz>BomDependencyGraphFlatMerger._flatten_merge..s3A!%%3r?c3HK|]\}}t|d|Dyw)c32K|]}t|ywrr r`s rrzHBomDependencyGraphFlatMerger._flatten_merge...s6aJqM6sNr )rbrdss rrz>BomDependencyGraphFlatMerger._flatten_merge..s) B r626 7 s ") listsetpopidadd setdefaultr_rZextenditems)r\flattodosseentodotodo_idre todo_depss rrYz+BomDependencyGraphFlatMerger._flatten_merges/1T u99;Dd8#, HHW 2.B ---y- 333 Y' zz|  r!)r9r rNrI)rJrKrLrMr r%r r-rVr+ staticmethodrr rYrOr!rrrjsi- sCD A - Xj1 hz6J  r!N)rM__all__collections.abcr itertoolsrrtypingrr model.dependencyr model.bomr model.bom_refrrrrOr!rr|sL$"  %%* 'BBJC C r!