From f282ad7ff1dad6e8a85233e9184e1922400feb2b Mon Sep 17 00:00:00 2001
From: Felix Stupp <felix.stupp@banananet.work>
Date: Sun, 30 Oct 2022 22:54:43 +0100
Subject: [PATCH] Add property MediaElement.average_release_per_week{,_now}

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

diff --git a/server/entertainment_decider/models.py b/server/entertainment_decider/models.py
index a164a72..eb7e10f 100644
--- a/server/entertainment_decider/models.py
+++ b/server/entertainment_decider/models.py
@@ -511,6 +511,28 @@ class MediaElement(db.Entity, Tagable):
     def started(self) -> bool:
         return not self.skip_over and self.progress != 0
 
+    @property
+    def average_release_per_week(self) -> float:
+        return (
+            orm.avg(
+                link.collection.average_release_per_week
+                for link in self.collection_links
+                if link.collection.watch_in_order
+            )
+            or 0
+        )
+
+    @property
+    def average_release_per_week_now(self) -> float:
+        return (
+            orm.avg(
+                link.collection.average_release_per_week_now
+                for link in self.collection_links
+                if link.collection.watch_in_order
+            )
+            or 0
+        )
+
     @property
     def ignored_recursive(self) -> bool:
         return (