From 840a85c62f5e8f5965357fc74f953b466cbd09de Mon Sep 17 00:00:00 2001
From: Felix Stupp <felix.stupp@outlook.com>
Date: Sun, 7 Nov 2021 01:51:30 +0100
Subject: [PATCH] MediaCollection.__to_watch_episodes: Add pony's Query as
 return type

---
 server/entertainment_decider/models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/server/entertainment_decider/models.py b/server/entertainment_decider/models.py
index 0559b6f..ec0b176 100644
--- a/server/entertainment_decider/models.py
+++ b/server/entertainment_decider/models.py
@@ -8,6 +8,7 @@ import logging
 from typing import Callable, Dict, Iterable, List, Optional, Set, Tuple, TypeVar, Union
 
 from pony import orm
+from pony.orm.core import Query
 
 db = orm.Database()
 
@@ -468,7 +469,7 @@ class MediaCollection(db.Entity, Tagable):
     def was_extracted(self) -> bool:
         return self.last_updated is not None
 
-    def __to_watch_episodes(self) -> Iterable[MediaCollectionLink]:
+    def __to_watch_episodes(self) -> Query | Iterable[MediaCollectionLink]:
         return orm.select(link for link in self.media_links if not link.element.watched and not link.element.ignored)
 
     @property