first commit

handlers/handler.py

@@ -0,0 +1,71 @@
+import pandas as pd
+from schema.pydantic import settings, Translit
+import os
+class BaseHandler:
+    def __init__(self, file_path:str):
+        self.file_name = file_path
+        self.file_path = os.path.join(settings.INPUTDIR, file_path)
+        
+    def struct(self):
+        try:
+            return pd.ExcelFile(self.file_path)
+        except Exception as e:
+            raise Exception(f"⚠️ Ошибка при получении структуры {self.file_path}: {e}")
+            
+    def read(self):
+        try:
+            return pd.read_excel(self.file_path)
+        except Exception as e:
+            raise Exception(f"⚠️ Ошибка при чтении файла {self.file_path}: {e}")
+
+class Handler(BaseHandler):
+    def get_articles_with_sales(self, columns:dict, sheet_name:str):
+        xls = self.struct()
+
+        if sheet_name not in xls.sheet_names:
+            raise Exception('⚠️ Лист {sheet_name} не найден')
+
+        df = pd.read_excel(xls, sheet_name=sheet_name)
+        df = df.iloc[:, list(columns.values())]
+        df.columns = list(columns.keys())
+
+        #Нормализация
+        df['Артикул'] = df['Артикул'].replace(Translit.TRANSLIT, regex=True)
+        
+        #группировка
+        df['Артикул'] = df['Артикул'].astype(str).str.upper().str.extract(f'({settings.PATTERN})')
+        df.dropna(subset=["Артикул"], inplace=True)
+        agg_dict = {col: "sum" for col in df.columns if col != "Артикул" and col != "Наименование"}  # по умолчанию суммируем все кроме Артикул
+        if "Наименование" in df.columns:
+            agg_dict["Наименование"] = lambda x: "\n".join(sorted(set(x)))
+        df= df.groupby("Артикул", as_index=False).agg(agg_dict)
+
+        #Исчисляемые колонки
+        df["Все удержания в %"] = df.apply(
+            lambda row: (row["Все_удержания_магазина"] / row["Выручка"] * 100) if row["Выручка"] != 0 else 0,
+            axis=1
+        )
+        if "Выкупы" in df.columns and "Не_выкупы" in df.columns:
+            df["Всего заказано"] = df.apply(
+                lambda row: row["Выкупы"]+row["Не_выкупы"],
+                axis=1
+            )
+            df["Процент выкупа"] = df.apply(
+                lambda row: ((row["Выкупы"] - row["Возвраты"]) / row["Всего заказано"] * 100)
+                if row["Всего заказано"] != 0 else 0,
+                axis=1
+            )
+        df["Маржинальность"] = df.apply(
+            lambda row: (row["Прибыль"] / row["Выручка"] * 100) if row["Выручка"] != 0 else 0,
+            axis=1
+        )
+        df=df.round(2)
+        df = df.sort_values(ascending=False,by="Прибыль")
+        def multi_style(val):
+            if val < 0:
+                return "background-color: red"
+            elif val > 0:
+                return "background-color: green; color: white"
+            return ""
+        styled = df.style.map(multi_style, subset=["Маржинальность","Прибыль"])
+        styled.to_excel(f"output/{self.file_name}", engine="openpyxl", index=False)