lessons tomorrow

helpers/mongo.py

@@ -53,18 +53,23 @@ class Mongo:
     def lessons_collection(self):
         return self["lessons"]
 
-    def get_today_lessons(self, user: User):
-        today = now(user)
-        tomorrow = today + datetime.timedelta(days=1)
-        tomorrow = datetime.datetime(year=tomorrow.year, month=tomorrow.month, day=tomorrow.day)
+    def _get_lessons_on_date(self, user: User, date: datetime.datetime):
+        date = datetime.datetime(year=date.year, month=date.month, day=date.day)
+        next_date = date + datetime.timedelta(days=1)
         lessons = []
         for lesson in self.lessons_collection.find({
             "hse_user_id": user.hse_id,
-            "begin": {"$gte": today, "$lte": tomorrow}}
+            "begin": {"$gte": date, "$lte": next_date}}
         ):
             lessons.append(lesson)
         lessons.sort(key=lambda les: les["begin"])
         return lessons
 
+    def get_today_lessons(self, user: User):
+        return self._get_lessons_on_date(user, now(user))
+
+    def get_tomorrow_lessons(self, user: User):
+        return self._get_lessons_on_date(user, now(user) + datetime.timedelta(days=1))
+
 
 mongo = Mongo()