From d3caed4241328258dee6190463bae8b4b5f3800d Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Mon, 28 Sep 2020 14:39:59 +0300
Subject: Use grouped predictions for dates in table

---
 iss/predictions.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'iss/predictions.py')

diff --git a/iss/predictions.py b/iss/predictions.py
index bab9f17..df6326a 100644
--- a/iss/predictions.py
+++ b/iss/predictions.py
@@ -1,3 +1,4 @@
+from collections import defaultdict
 from skyfield.api import Topos, load
 
 from .utils import chunks
@@ -90,3 +91,18 @@ class Predictions(object):
     def get_predictions(self):
         preds = self.get_prediction_events()
         return [self.get_prediction_details(*p) for p in preds]
+
+    def truncate_prediction_dates(self, pred):
+        for t in ["rise", "culminate", "set"]:
+            pred[t]["time"] = pred[t]["time"][11:]
+        return pred
+
+    def get_grouped_predictions(self):
+        preds = self.get_predictions()
+        res = defaultdict(list)
+
+        for pred in preds:
+            date = pred["rise"]["time"][:10]
+            res[date].append(self.truncate_prediction_dates(pred))
+
+        return res
-- 
cgit v1.3.1