From 3c64bc6f664f211b077dde128243dfe31468b7d6 Mon Sep 17 00:00:00 2001 From: Felix Stupp Date: Sat, 5 Nov 2022 23:23:54 +0100 Subject: [PATCH] generate_preference_score: Add auto tag for extractor --- server/entertainment_decider/models.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/server/entertainment_decider/models.py b/server/entertainment_decider/models.py index 6470e4c..d48c9c0 100644 --- a/server/entertainment_decider/models.py +++ b/server/entertainment_decider/models.py @@ -407,6 +407,17 @@ def generate_preference_list( ) ) + # add tags corresponding to extractors + @cache + def get_extractor_tag(extractor_name: str) -> Tag: + return Tag( + title=f"Automatic for extractor {extractor_name}", + use_for_preferences=True, + ) + + for element in element_list: + element.tag_list.add(get_extractor_tag(element.extractor_name)) + # flush after custom tags orm.flush()