fix bug with group

a-berezhkov · Nov 17, 2022 · 41c5b94 · 41c5b94
1 parent 58a3065
commit 41c5b94
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "schedule_parser_portal_petrocollege"
-version = "0.0.3"
+version = "0.0.5"
 authors = [
   { name="Andrei V. Berezhkov", email="[email protected]" },
 ]

diff --git a/src/Lesson.py b/src/Lesson.py
@@ -8,7 +8,7 @@ def __init__(self, string_to_parse):
         self.discipline = ''
         self.room = ''
         self.building = '0'
-        self.groups = [string_to_parse[:6].strip()]
+        self.groups = [Lesson.get_course(string_to_parse[:6].strip())]
         self.__current_string = string_to_parse[6:].strip()
         self.is_dop = False
         self.subgroup = 0
@@ -83,7 +83,7 @@ def get_course(group_name):
         year_start = year_end - 10
         for i in range(year_start, year_end, 1):
             year = str(i)
-            if year[-1] == group_name[1]:
+            if (year[-1] == group_name[1]):
                 course = year_end - i + 1
                 group_data = {"group": group_name, "year": year, "course": course}
         return group_data