fix: Add text if empty teacher row

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rtu_schedule_parser"
-version = "2.3.1"
+version = "2.3.2"
 description = "Easy extraction of the MIREA - Russian Technological University schedule from Excel documents."
 authors = ["Sergey Dmitriev <[email protected]>"]
 

rtu_schedule_parser/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.3.1"
+__version__ = "2.3.2"
 __author__ = "Sergey Dmitriev"
 
 from .excel_parser import ExcelScheduleParser

rtu_schedule_parser/excel_formatter.py

@@ -406,10 +406,22 @@ def try_get_room(room) -> Room:
 
         return result
 
+    def __replace_empty_teachers_to_text(self, text: str) -> str:
+        """Иногда стоят подгруппы бе преподавателей, заменяем их на текст. Такое бывает, если расписание не доделано."""
+        return re.sub(
+            r"^(\d п/г)$",
+            r"Нет,\g<1>",
+            text.strip(),
+            flags=re.IGNORECASE | re.MULTILINE,
+        )
+
+
     def get_teachers(self, names_cell_value: str) -> list[str] | list[tuple[str, int]]:
         if not re.search(r"[а-яА-Я]", names_cell_value):
             return []
 
+        names_cell_value = self.__replace_empty_teachers_to_text(names_cell_value)
+
         teachers_names = names_cell_value.strip()
 
         re_typos = r"[а-яё]{1}(,) {0,2}[а-яё]{1}[. ]"

rtu_schedule_parser/excel_parser.py

@@ -161,9 +161,12 @@ def __parse_lessons(
                         lesson_teachers[0], tuple
                     ):
                         if len(lesson_teachers) < lessons_len:
-                            if lesson_teachers[0][0] == lesson_teachers[1][0]:
-                                lesson_teachers.insert(0, ("", None))
-                            else:
+                            try:
+                                if lesson_teachers[0][0] == lesson_teachers[1][0]:
+                                    lesson_teachers.insert(0, ("", None))
+                                else:
+                                    lesson_teachers.append(("", None))
+                            except IndexError:
                                 lesson_teachers.append(("", None))
 
                         subgroup = lesson_teachers[i][1]

tests/excel_formatter/test_get_teachers.py

@@ -160,4 +160,16 @@ def test_get_teacher_21(excel_formatter):
 def test_get_teacher_22(excel_formatter):
     result = excel_formatter.get_teachers("1 п/г\n2 п/г")
 
-    assert result == []
+    assert result == [
+        ('Нет', 1),
+        ('Нет', 2),
+    ]
+
+
+def test_get_teacher_23(excel_formatter):
+    result = excel_formatter.get_teachers("1 п/г\n\n2 п/г")
+
+    assert result == [
+        ('Нет', 1),
+        ('Нет', 2),
+    ]

tests/test_rtu_schedule_parser.py

@@ -2,4 +2,4 @@
 
 
 def test_version():
-    assert __version__ == "2.3.1"
+    assert __version__ == "2.3.2"