1616from shared .models .cve import AffectedProduct , Metric , Reference , Version
1717from shared .models .linkage import (
1818 CVEDerivationClusterProposal ,
19- MaintainersEdit ,
20- PackageEdit ,
19+ MaintainerOverlay ,
20+ PackageOverlay ,
2121 ReferenceOverlay ,
2222)
2323from shared .models .nix_evaluation import get_major_channel
@@ -112,8 +112,8 @@ class Reference(BaseModel):
112112 categorized_references : CategorizedReferences
113113
114114
115- def apply_package_edits (
116- packages : dict , edits : list [PackageEdit ]
115+ def apply_package_overlays (
116+ packages : dict , edits : list [PackageOverlay ]
117117) -> dict [str , CachedSuggestion .Package ]:
118118 """
119119 Returns the packages dict with user-supplied package edits applied.
@@ -122,7 +122,7 @@ def apply_package_edits(
122122 to_skip = {
123123 edit .package_attribute
124124 for edit in edits
125- if edit .edit_type == PackageEdit . EditType .REMOVE
125+ if edit .edit_type == MaintainerOverlay . Type .REMOVE
126126 }
127127
128128 return {attr : data for attr , data in packages .items () if attr not in to_skip }
@@ -214,11 +214,11 @@ def cache_new_suggestions(suggestion: CVEDerivationClusterProposal) -> None:
214214
215215 prefetched_metrics = Metric .objects .filter (container__cve = suggestion .cve )
216216 original_packages = channel_structure (all_versions , derivations )
217- maintainers_edits = list (
218- suggestion .maintainers_edits .select_related ("maintainer" ).all ()
217+ maintainer_overlays = list (
218+ suggestion .maintainer_overlays .select_related ("maintainer" ).all ()
219219 )
220- package_edits = list (suggestion .package_edits .all ())
221- packages = apply_package_edits (original_packages , package_edits )
220+ package_overlays = list (suggestion .package_overlays .all ())
221+ packages = apply_package_overlays (original_packages , package_overlays )
222222
223223 only_relevant_data = CachedSuggestion (
224224 pk = suggestion .pk ,
@@ -231,7 +231,7 @@ def cache_new_suggestions(suggestion: CVEDerivationClusterProposal) -> None:
231231 original_packages = packages ,
232232 packages = packages ,
233233 metrics = [to_dict (m ) for m in prefetched_metrics ],
234- categorized_maintainers = categorize_maintainers (packages , maintainers_edits ),
234+ categorized_maintainers = categorize_maintainers (packages , maintainer_overlays ),
235235 categorized_references = categorize_references (
236236 suggestion .references , list (suggestion .reference_overlays .all ())
237237 ),
@@ -397,7 +397,7 @@ def parse_drv_name(name: str) -> tuple[str, str]:
397397
398398
399399def maintainers_list (
400- packages : dict , edits : list [MaintainersEdit ]
400+ packages : dict , edits : list [MaintainerOverlay ]
401401) -> list [CachedSuggestion .Maintainer ]:
402402 """
403403 Returns a deduplicated list (by GitHub ID) of all the maintainers, as dicts,
@@ -411,12 +411,12 @@ def maintainers_list(
411411 to_skip_or_seen : set [int ] = {
412412 m .maintainer .github_id
413413 for m in edits
414- if m .edit_type == MaintainersEdit . EditType .REMOVE
414+ if m .edit_type == MaintainerOverlay . Type .REMOVE
415415 }
416416 to_add : list [CachedSuggestion .Maintainer ] = [
417417 CachedSuggestion .Maintainer .model_validate (to_dict (m .maintainer ))
418418 for m in edits
419- if m .edit_type == MaintainersEdit . EditType .ADD
419+ if m .edit_type == MaintainerOverlay . Type .ADD
420420 ]
421421
422422 maintainers : list [CachedSuggestion .Maintainer ] = list ()
@@ -473,7 +473,7 @@ def categorize_references(
473473
474474def categorize_maintainers (
475475 packages : dict [str , CachedSuggestion .Package ],
476- maintainers_edits : list [MaintainersEdit ],
476+ maintainer_overlays : list [MaintainerOverlay ],
477477) -> CachedSuggestion .CategorizedMaintainers :
478478 """
479479 Categorize maintainers associated to the packages of a suggestion.
@@ -490,10 +490,10 @@ def categorize_maintainers(
490490 removed_github_ids = set ()
491491 added_maintainers = []
492492
493- for edit in maintainers_edits :
494- if edit .edit_type == MaintainersEdit . EditType .REMOVE :
493+ for edit in maintainer_overlays :
494+ if edit .edit_type == MaintainerOverlay . Type .REMOVE :
495495 removed_github_ids .add (edit .maintainer .github_id )
496- elif edit .edit_type == MaintainersEdit . EditType .ADD :
496+ elif edit .edit_type == MaintainerOverlay . Type .ADD :
497497 added_maintainers .append (
498498 CachedSuggestion .Maintainer .model_validate (to_dict (edit .maintainer ))
499499 )
0 commit comments