validation file 1.1 (OOP)
This commit is contained in:
@@ -1,8 +1,92 @@
|
||||
import pandas as pd
|
||||
def read_excel(base_dir):
|
||||
try:
|
||||
dfs = pd.read_excel(base_dir, sheet_name=None)
|
||||
return dfs
|
||||
except Exception as e:
|
||||
raise f"⚠️ Ошибка при чтении {file.name}: {e}"
|
||||
# def read_excel(base_dir):
|
||||
# try:
|
||||
# dfs = pd.read_excel(base_dir, sheet_name=None)
|
||||
# return dfs
|
||||
# except Exception as e:
|
||||
# raise f"⚠️ Ошибка при чтении {file.name}: {e}"
|
||||
|
||||
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)
|
||||
# проверяем наличие нужных листов
|
||||
if "Получено от потребителей" not in dfs or "Возвращено потребителям" not in dfs:
|
||||
raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
|
||||
# сама обработка
|
||||
print("Реализация Яндекс")
|
||||
|
||||
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} отсутствуют необходимые листы")
|
||||
print("Реализация WB")
|
||||
|
||||
class OZONHandler(BaseHandler):
|
||||
def process(self):
|
||||
dfs = pd.read_excel(self.file_path, sheet_name=None)
|
||||
if "Отчет о реализации" not in dfs :
|
||||
raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
|
||||
print("Реализация OZON")
|
||||
|
||||
class OZONPurchasesHandler(BaseHandler):
|
||||
def process(self):
|
||||
dfs = pd.read_excel(self.file_path, sheet_name=None)
|
||||
if "Отчет о выкупленных товарах" not in dfs:
|
||||
raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
|
||||
print("Выкупы озон")
|
||||
|
||||
class WBPurchasesHandler(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} отсутствуют необходимые листы")
|
||||
print("Выкупы wb")
|
||||
class OZONComHandler(BaseHandler):
|
||||
def process(self):
|
||||
dfs = pd.read_excel(self.file_path, sheet_name=None)
|
||||
if "Лист_1" not in dfs:
|
||||
raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
|
||||
df = dfs["Лист_1"]
|
||||
cont = df.iloc[1, 0]
|
||||
if cont != "«Интернет решения» ООО":
|
||||
raise Exception(f"В файле {self.file_path.name} неверный контрагент")
|
||||
print("Товары, переданные на комиссию озон")
|
||||
|
||||
class WBComHandler(BaseHandler):
|
||||
def process(self):
|
||||
dfs = pd.read_excel(self.file_path, sheet_name=None)
|
||||
if "Лист_1" not in dfs:
|
||||
raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
|
||||
df = dfs["Лист_1"]
|
||||
cont = df.iloc[1, 0]
|
||||
if cont != '"Вайлдберриз" ООО':
|
||||
raise Exception(f"В файле {self.file_path.name} неверный контрагент")
|
||||
print("Товары, переданные на комиссию wb")
|
||||
|
||||
class YandexComHandler(BaseHandler):
|
||||
def process(self):
|
||||
dfs = pd.read_excel(self.file_path, sheet_name=None)
|
||||
if "Лист_1" not in dfs:
|
||||
raise Exception(f"В файле {self.file_path.name} отсутствуют необходимые листы")
|
||||
df = dfs["Лист_1"]
|
||||
cont = df.iloc[1, 0]
|
||||
if cont != "Яндекс Маркет ООО":
|
||||
raise Exception(f"В файле {self.file_path.name} неверный контрагент")
|
||||
print("Товары, переданные на комиссию yandex")
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
from server.backend.excel import read_excel
|
||||
import server.backend.excel as excel
|
||||
|
||||
from dotenv import load_dotenv #Работа с env
|
||||
import os
|
||||
@@ -6,6 +6,17 @@ from pathlib import Path
|
||||
load_dotenv()
|
||||
base_dir = Path(os.getenv("DIR"))
|
||||
|
||||
handlers = { #метки какие файлы есть и должны быть занесены с вызовами функций из файла
|
||||
"period_closure_income": excel.YandexHandler,
|
||||
"0": excel.WBHandler,
|
||||
"отчет о реализации товара": excel.OZONHandler,
|
||||
"realizationreportcis": excel.OZONPurchasesHandler,
|
||||
"уведомление о выкупе": excel.WBPurchasesHandler,
|
||||
"Товары, переданные на комиссию озон": excel.OZONComHandler,
|
||||
"Товары, переданные на комиссию вб": excel.WBComHandler,
|
||||
"Товары, переданные на комиссию яндекс": excel.YandexComHandler
|
||||
}
|
||||
|
||||
#Проход по всем файлам в директории
|
||||
def validating():
|
||||
if not base_dir.exists():
|
||||
@@ -17,52 +28,31 @@ def validating():
|
||||
|
||||
name = file.stem.lower()
|
||||
print(f"\n🔹 Файл: {file.name}")
|
||||
label = None
|
||||
match file:
|
||||
case _ if "period_closure_income" in name:
|
||||
dfs = read_excel(file)
|
||||
if "Получено от потребителей" and "Возвращено потребителям" in dfs :
|
||||
print("Реализация Яндекс ", name)
|
||||
else:
|
||||
raise Exception(f"В файле {name} отсутствуют необходимые листы")
|
||||
label = "period_closure_income"
|
||||
case _ if name == "0":
|
||||
dfs = read_excel(file)
|
||||
if "Sheet1" in dfs :
|
||||
print("Реализация ВБ", name)
|
||||
else:
|
||||
raise Exception(f"В файле {name} отсутствуют необходимые листы")
|
||||
label = "0"
|
||||
case _ if "отчет о реализации товара" in name:
|
||||
dfs = read_excel(file)
|
||||
if "Отчет о реализации" in dfs :
|
||||
print("Реализация ОЗОН", name)
|
||||
else:
|
||||
raise Exception(f"В файле {name} отсутствуют необходимые листы")
|
||||
case _ if "вб" in name.lower() or "озон" in name.lower() or "яндекс" in name:
|
||||
dfs = read_excel(file)
|
||||
if "Лист_1" in dfs :
|
||||
df = dfs["Лист_1"]
|
||||
if df.iloc[1,0] == "«Интернет решения» ООО":
|
||||
print("Товары, переданные на комиссию", name)
|
||||
elif df.iloc[1,0] == "Яндекс Маркет ООО":
|
||||
print("Товары, переданные на комиссию", name)
|
||||
elif df.iloc[1,0] == '"Вайлдберриз" ООО':
|
||||
print("Товары, переданные на комиссию", name)
|
||||
else:
|
||||
raise Exception(f"В файле {name} неизвестный контрагент")
|
||||
else:
|
||||
raise Exception(f"В файле {name} отсутствуют необходимые листы")
|
||||
|
||||
label = "отчет о реализации товара"
|
||||
case _ if "вб" in name:
|
||||
label = "Товары, переданные на комиссию вб"
|
||||
case _ if "озон" in name:
|
||||
label = "Товары, переданные на комиссию озон"
|
||||
case _ if "яндекс" in name:
|
||||
label = "Товары, переданные на комиссию яндекс"
|
||||
case _ if "realizationreportcis" in name:
|
||||
dfs = read_excel(file)
|
||||
if "Отчет о выкупленных товарах" in dfs :
|
||||
print("Выкупы озон", name)
|
||||
else:
|
||||
raise Exception(f"В файле {name} отсутствуют необходимые листы")
|
||||
|
||||
label = "realizationreportcis"
|
||||
case _ if "уведомление о выкупе" in name:
|
||||
dfs = read_excel(file)
|
||||
if "Sheet1" in dfs :
|
||||
print("Выкупы вб", name)
|
||||
else:
|
||||
raise Exception(f"В файле {name} отсутствуют необходимые листы")
|
||||
case _:
|
||||
label = "уведомление о выкупе"
|
||||
case _: #Для неизвестных файлов
|
||||
print("Неизвестный файл")
|
||||
if label:
|
||||
handler_class = handlers.get(label)
|
||||
if handler_class:
|
||||
# создаём объект для текущего файла
|
||||
handler = handler_class(file) # передаём путь к файлу
|
||||
handler.process() # dfs читается внутри объекта
|
||||
else:
|
||||
print(f"⚠️ Нет обработчика для метки '{label}'")
|
||||
|
||||
Reference in New Issue
Block a user