moved map_subsets_by_family under BuildWorkfile class

This commit is contained in:
iLLiCiTiT 2020-11-03 11:55:25 +01:00
parent b407291401
commit 83c4696591
2 changed files with 15 additions and 16 deletions

View file

@ -36,7 +36,6 @@ from .lib_old import (
get_last_version_from_path,
get_subsets,
get_linked_assets,
map_subsets_by_family,
BuildWorkfile,
ffprobe_streams,
source_hash,

View file

@ -608,20 +608,6 @@ def get_linked_assets(asset_entity):
return inputs
def map_subsets_by_family(subsets):
subsets_by_family = collections.defaultdict(list)
for subset in subsets:
family = subset["data"].get("family")
if not family:
families = subset["data"].get("families")
if not families:
continue
family = families[0]
subsets_by_family[family].append(subset)
return subsets_by_family
class BuildWorkfile:
"""Wrapper for build workfile process.
@ -629,6 +615,20 @@ class BuildWorkfile:
are host related, since each host has it's loaders.
"""
@staticmethod
def map_subsets_by_family(subsets):
subsets_by_family = collections.defaultdict(list)
for subset in subsets:
family = subset["data"].get("family")
if not family:
families = subset["data"].get("families")
if not families:
continue
family = families[0]
subsets_by_family[family].append(subset)
return subsets_by_family
def process(self):
"""Main method of this wrapper.
@ -901,7 +901,7 @@ class BuildWorkfile:
:rtype: dict
"""
# Prepare subsets
subsets_by_family = map_subsets_by_family(subsets)
subsets_by_family = self.map_subsets_by_family(subsets)
profiles_per_subset_id = {}
for family, subsets in subsets_by_family.items():