many fixes

helpers/mongo.py

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