exclusion of applications tables

Author: vovanbravin

app/main/checks/report_checks/table_percentage.py

@@ -1,4 +1,3 @@
-
 import fitz
 from docx import Document
 from ..base_check import BaseReportCriterion, answer
@@ -11,16 +10,16 @@
 
 #С данными значениями погрешность 4-5 процентов
 
-
 class TablePercentage(BaseReportCriterion):
     label = "Проверка процентного соотношения таблиц в документе"
     description = "Проверяет, что таблицы занимают не более установленного процента площади документа"
     id = 'table_percentage'
 
 
-    def __init__(self, file_info, max_percentage = 30):
+    def __init__(self, file_info, hasApplication = True,max_percentage = 30):
         super().__init__(file_info)
         self._max_percentage = max_percentage
+        self._hasApplication = hasApplication
 
     def get_font_size(self, run, paragraph) -> float:
         """Функция получения размера шрифта"""
@@ -96,13 +95,45 @@ def getPercentOfTables(self) -> float:
 
         tables_height = 0
 
-        for table in doc_docx.tables:
-            table_height = self.heightTable(table)
-            tables_height += table_height
+        end_index = len(doc_docx.tables)
+
+        if self._hasApplication:
+            end_index = self.find_table_index_after_text('ПРИЛОЖЕНИЕ')
+            end_index = end_index if end_index is not None else len(doc_docx.tables)
+
+        for table_index, table in enumerate(doc_docx.tables):
+            if table_index < end_index:
+                table_height = self.heightTable(table)
+                tables_height += table_height
 
         percentage = (tables_height / total_height) * 100
         return percentage
 
+
+    def find_table_index_after_text(self, target_text: str) -> int | None:
+        """Функция находит первый индекс таблицы после target_text, если не найден, то вернет None"""
+        doc_docx = self.file.file
+        for i, paragraph in enumerate(doc_docx.paragraphs):
+            if target_text in paragraph.text:
+
+                para_elem = paragraph._element
+                following_tables = para_elem.xpath('./following-sibling::w:tbl')
+
+                for table_elem in following_tables:
+
+                    for index, table in enumerate(doc_docx.tables):
+                        if table._element == table_elem:
+                            return index
+
+        return None
+
+    def find_tables_indexs_between_text(self,start_text: str, end_text: str | None = None):
+        """Функция нахождения индексов начала и конца таблиц между двумя текстами"""
+        start_index = self.find_table_index_after_text(start_text)
+        end_index = self.find_table_index_after_text(end_text)
+
+        return (start_index, end_index)
+
     def check(self):
         try:
             if self.file.page_counter() < 4:
@@ -118,9 +149,15 @@ def check(self):
             else:
                 return answer(
                     False,
-                    f"Таблицы занимают {percent_tables:.1f}% документа, "
-                    f"что превышает допустимые {self._max_percentage}%. "
-                    f"Рекомендуется сократить количество или размер таблиц."
+                    f'''
+                    Таблицы занимают {percent_tables:.1f}% документа, что превышает допустимые {self._max_percentage}%.
+                    Рекомендации:
+                    <ul>
+                        <li>Уменьшите количество таблиц в основном тексте документа;</li>
+                        <li>Перенесите вспомогательные таблицы в приложения;</li>
+                        <li>Сократите объем данных в таблицах, оставив только самую важную информацию.</li>
+                    </ul>
+                    '''
                 )
         except Exception as e:
             return answer(False, f"Ошибка при проверке процентного соотношения таблиц: {str(e)}")