From bb83897f0b3422456af8c6413963312acbb16e80 Mon Sep 17 00:00:00 2001
From: Marioneq <mq4958@gmail.com>
Date: Fri, 7 Feb 2025 23:18:54 +0100
Subject: [PATCH] SDK: Add notes (only prometheus)

---
 sdk/src/apis/hebe/client.py                |  8 ++++++++
 sdk/src/apis/hebe/constants.py             |  1 +
 sdk/src/interfaces/prometheus/interface.py |  7 ++++++-
 sdk/src/models/note.py                     | 21 +++++++++++++++++++++
 4 files changed, 36 insertions(+), 1 deletion(-)
 create mode 100644 sdk/src/models/note.py

diff --git a/sdk/src/apis/hebe/client.py b/sdk/src/apis/hebe/client.py
index 6453300..ef8aa63 100644
--- a/sdk/src/apis/hebe/client.py
+++ b/sdk/src/apis/hebe/client.py
@@ -15,6 +15,7 @@ from sdk.src.apis.hebe.constants import (
     ENDPOINT_EXAM_BYPUPIL,
     ENDPOINT_GRADE_BYPUPIL,
     ENDPOINT_HEARTBEAT,
+    ENDPOINT_NOTE_BYPUPIL,
     ENDPOINT_REGISTER_HEBE,
     ENDPOINT_REGISTER_JWT,
     ENDPOINT_REGISTER_TOKEN,
@@ -37,6 +38,7 @@ from sdk.src.apis.hebe.student import HebeStudent
 from sdk.src.apis.hebe.signer import get_signature_values
 from sdk.src.models.exam import Exam
 from sdk.src.models.grade import Grade
+from sdk.src.models.note import Note
 
 
 class HebeClient:
@@ -199,3 +201,9 @@ class HebeClient:
             },
         )
         return list(map(Exam.from_hebe_dict, envelope))
+
+    def get_notes(self, student_id: int):
+        envelope = self._send_request(
+            "GET", ENDPOINT_NOTE_BYPUPIL, params={"pupilId": student_id}
+        )
+        return list(map(Note.from_hebe_dict, envelope))
diff --git a/sdk/src/apis/hebe/constants.py b/sdk/src/apis/hebe/constants.py
index 89ec08f..76ea171 100644
--- a/sdk/src/apis/hebe/constants.py
+++ b/sdk/src/apis/hebe/constants.py
@@ -19,3 +19,4 @@ ENDPOINT_SCHOOL_LUCKY = "mobile/school/lucky"
 ENDPOINT_GRADE_BYPUPIL = "mobile/grade/byPupil"
 ENDPOINT_SCHEDULE_WITHCHANGES_BYPUPIL = "mobile/schedule/withchanges/byPupil"
 ENDPOINT_EXAM_BYPUPIL = "mobile/exam/byPupil"
+ENDPOINT_NOTE_BYPUPIL = "mobile/note/byPupil"
diff --git a/sdk/src/interfaces/prometheus/interface.py b/sdk/src/interfaces/prometheus/interface.py
index aa21ccc..18cb4c2 100644
--- a/sdk/src/interfaces/prometheus/interface.py
+++ b/sdk/src/interfaces/prometheus/interface.py
@@ -147,7 +147,7 @@ class PrometheusInterface(CoreInterface):
             self._hebe_client.set_rest_url(get_hebe_url(symbol))
             self._hebe_client.register_jwt(symbols[symbol])
 
-        self._auth_context.symbols = symbols.keys()
+        self._auth_context.symbols = list(symbols.keys())
 
     def _login_efeb(self):
         if not self._auth_context.efeb_web_cookies:
@@ -254,3 +254,8 @@ class PrometheusInterface(CoreInterface):
         self._check_is_auth_context_full()
         self._check_is_student_selected()
         return self._hebe_client.get_exams(self._student_context.student_id, from_, to)
+
+    def get_notes(self):
+        self._check_is_auth_context_full()
+        self._check_is_student_selected()
+        return self._hebe_client.get_notes(self._student_context.student_id)
diff --git a/sdk/src/models/note.py b/sdk/src/models/note.py
new file mode 100644
index 0000000..0e69a6f
--- /dev/null
+++ b/sdk/src/models/note.py
@@ -0,0 +1,21 @@
+from dataclasses import dataclass
+from datetime import datetime
+
+
+@dataclass
+class Note:
+    name: str | None
+    content: str
+    points: str | None
+    creator: str
+    created_at: datetime
+
+    @staticmethod
+    def from_hebe_dict(data: dict):
+        return Note(
+            name=data["Category"]["Name"] if data["Category"] else None,
+            content=data["Content"],
+            points=str(data["Points"]) if data["Points"] else None,
+            creator=data["Creator"]["DisplayName"],
+            created_at=datetime.fromtimestamp(data["DateValid"]["Timestamp"] / 1000),
+        )