From fc35a62871350030c67ddc5b86197f52adcae472 Mon Sep 17 00:00:00 2001
From: Felix Stupp <felix.stupp@banananet.work>
Date: Sat, 5 Nov 2022 23:36:20 +0100
Subject: [PATCH] Extract set_watch_in_order_auto to MediaCollection

---
 server/entertainment_decider/extractors/collection/youtube.py | 3 +--
 server/entertainment_decider/models.py                        | 4 ++++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/server/entertainment_decider/extractors/collection/youtube.py b/server/entertainment_decider/extractors/collection/youtube.py
index fddb307..153cb7e 100644
--- a/server/entertainment_decider/extractors/collection/youtube.py
+++ b/server/entertainment_decider/extractors/collection/youtube.py
@@ -97,8 +97,7 @@ class YouTubeCollectionExtractor(CollectionExtractor[Dict]):
         )
         object.add_single_uri(info["link"])
         video_list = data["videos"]
-        if object.watch_in_order_auto:
-            object.watch_in_order = not is_channel
+        object.set_watch_in_order_auto(not is_channel)
         if is_channel:
             video_list = reversed(video_list)
         for index, video in enumerate(video_list):
diff --git a/server/entertainment_decider/models.py b/server/entertainment_decider/models.py
index cf2be28..669ba03 100644
--- a/server/entertainment_decider/models.py
+++ b/server/entertainment_decider/models.py
@@ -1065,6 +1065,10 @@ class MediaCollection(db.Entity, UriHolder, Tagable):
 
     ### methods
 
+    def set_watch_in_order_auto(self, watch_in_order: bool) -> None:
+        if self.watch_in_order_auto:
+            self.watch_in_order = watch_in_order
+
     def add_episode(
         self,
         media: MediaElement,