refactoring

2025-12-21 17:51:41 +03:00
parent d9bf44e669
commit 46b4e95d45
16 changed files with 42 additions and 10 deletions
--- a/server/backend/services/excel.py
+++ b/server/backend/services/excel.py
@@ -0,0 +1,107 @@
+import pandas as pd
+import server.backend.handlers.yandex_handler as yandex_handler
+import server.backend.handlers.wb_handler as wb_handler
+import server.backend.handlers.ozon_handler as ozon_handler
+import server.backend.handlers.ozon_purchases_handler as ozon_purchases_handler
+import server.backend.handlers.wb_purchases_handler as wb_purchases_handler
+import server.backend.handlers.ozon_wb_yandex_com_handler as ozon_wb_yandex_com_handler
+class BaseHandler:
+    def __init__(self, file_path):
+        self.file_path = file_path
+        self.dfs = self.read()
+
+    def read(self):
+        try:
+            return pd.read_excel(self.file_path, sheet_name=None)
+        except Exception as e:
+            raise Exception(f"⚠️ Ошибка при чтении {self.file_path}: {e}")
+
+    def process(self):
+        raise NotImplementedError
+
+class YandexHandler(BaseHandler):
+    def process(self):
+        # читаем Excel внутри объекта
+        #доставать дату
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=[0, 1, 3]) #skip header
+        # проверяем наличие нужных листов
+        if "Получено от потребителей" not in dfs or "Возвращено потребителям" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        # вызываем функцию evaluating
+        validated_data = yandex_handler.evaluating(dfs)
+        print("Реализация Яндекс завершена, валидированных строк:", len(validated_data[0]), "Реализация", len(validated_data[1]), "Возвраты")
+
+class WBHandler(BaseHandler):
+    def process(self):
+        dfs = pd.read_excel(self.file_path, sheet_name=None)
+        #доставать дату по месяцу и просто день ставить последний
+        if "Sheet1" not in dfs :
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        validated_data = wb_handler.evaluating(dfs)
+        print("Реализация WB завершена, валидированных строк:", len(validated_data[0]), "Реализация", len(validated_data[1]), "Возвраты")
+
+class OZONHandler(BaseHandler):
+    def process(self):
+        #Доставать № документа и дату
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=14, skipfooter=17) #skip the header and the footer
+        if "Отчет о реализации" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        validated_data = ozon_handler.evaluating(dfs)
+        print("Реализация OZON завершена, валидированных строк:", len(validated_data[0]), "Реализация", len(validated_data[1]), "Возвраты")
+
+class OZONPurchasesHandler(BaseHandler):
+    def process(self):
+        #доставать дату и номер документа
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=12, skipfooter=1)
+        if "Отчет о выкупленных товарах" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        validated_data = ozon_purchases_handler.evaluating(dfs)
+        print("Выкупы OZON завершены, валидированных строк:", len(validated_data), "Реализация")
+
+class WBPurchasesHandler(BaseHandler):
+    def process(self):
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=9, skipfooter=7)
+        if "Sheet1" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        validated_data = wb_purchases_handler.evaluating(dfs)
+        print("Выкупы WB завершены, валидированных строк:", len(validated_data), "Реализация")
+
+class OZONComHandler(BaseHandler):
+    def process(self):
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skipfooter=1)
+        if "Лист_1" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        df = dfs["Лист_1"]  
+        cont = df.iloc[1, 0]
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=[0,2],skipfooter=1)
+        if cont != "«Интернет решения» ООО":
+            raise Exception(f"В файле {self.file_path.name} неверный контрагент")
+        validated_data = ozon_wb_yandex_com_handler.evaluating(dfs)
+        print("Передача на коммисию OZON завершена, валидированных строк:", len(validated_data), "Реализация")
+
+class WBComHandler(BaseHandler):
+    def process(self):
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skipfooter=1)
+        if "Лист_1" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        df = dfs["Лист_1"]  
+        cont = df.iloc[1, 0]
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=[0,2], skipfooter=1)
+        if cont != '"Вайлдберриз" ООО':
+            raise Exception(f"В файле {self.file_path.name} неверный контрагент")
+        validated_data = ozon_wb_yandex_com_handler.evaluating(dfs)
+        print("Передача на коммисию WB завершена, валидированных строк:", len(validated_data), "Реализация")
+
+class YandexComHandler(BaseHandler):
+    def process(self):
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skipfooter=1)
+        if "Лист_1" not in dfs:
+            raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
+        df = dfs["Лист_1"]  
+        cont = df.iloc[1, 0]
+        dfs = pd.read_excel(self.file_path, sheet_name=None, skiprows=[0,2], skipfooter=1)
+        if cont != "Яндекс Маркет ООО":
+            raise Exception(f"В файле {self.file_path.name} неверный контрагент")
+        validated_data = ozon_wb_yandex_com_handler.evaluating(dfs)
+        print("Передача на коммисию YANDEX завершена, валидированных строк:", len(validated_data), "Реализация")
+