From 952fd088c989eaa7b55e0996baba8359f3e2bd43 Mon Sep 17 00:00:00 2001 From: Felix Stupp Date: Sun, 19 Mar 2023 10:41:37 +0100 Subject: [PATCH] Add tag based on extractor used --- server/entertainment_decider/extractors/generic.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/server/entertainment_decider/extractors/generic.py b/server/entertainment_decider/extractors/generic.py index bb83a76..9a14c46 100644 --- a/server/entertainment_decider/extractors/generic.py +++ b/server/entertainment_decider/extractors/generic.py @@ -11,6 +11,8 @@ from typing import Generic, Optional, TypeVar from ..models import ( MediaCollection, MediaElement, + Tag, + TagKey, ) @@ -215,6 +217,7 @@ class GeneralExtractor(Generic[E, T]): def _update_object(self, object: E, data: ExtractedDataOnline[T]) -> ChangedReport: object.uri = data.object_uri + object.tag_list.add(self._get_extractor_tag()) self._update_object_raw(object, data.data) self._update_hook(object, data) object.last_updated = datetime.now() @@ -263,3 +266,10 @@ class GeneralExtractor(Generic[E, T]): if object is not None: return object return self.store_object(self._extract_offline(uri)) + + def _get_extractor_tag(self) -> Tag: + return TagKey.get_or_create_tag( + tag_key=self.key, + title=f"[Extractor] {self.long_name}", + use_for_preferences=True, + )