consolidator
54 строки · 3.1 Кб
1import os
2import sys
3import pandas as pd
4import datetime
5from MainConsolidator import MainConsolidator
6
7class handler(object):
8@staticmethod
9def module_name():
10return 'КПК Колесо после термообработки'
11@staticmethod
12def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:
13try:
14diameter = consolidator.scalar_in_guid("Диаметры колес",f"`Производство`=='КПК Колесо после термообработки' and `Диаметры SAP`=={row['Диаметр, мм <Производство>']}","Диаметры УП")
15weigth = consolidator.nearest_in_guid("Параметры по Продукту КПК",
16f"`Производство`=='КПК Колесо после термообработки' and `Доп параметр1`=={diameter}",
17"Доп параметр2",
18row["Теоретический вес колеса после ТО, кг <Производство>"],
19"Доп параметр2")
20weigth_s=str(weigth).replace(".",",").replace(",0","")
21standart = consolidator.scalar_in_guid("Стандарты колесной заготовки",f"`Вес`=={weigth}","Стандарт для колесной заготовки")
22print(f"{row['Диаметр, мм <Производство>']} -> {diameter}: {row['Теоретический вес колеса после ТО, кг <Производство>']} -> {weigth_s} : {standart}")
23if pd.isna(diameter) or pd.isna(standart): raise Exception("Не определен вес или стандарт")
24row["error"] = pd.NA
25except Exception as exp:
26row["error"] = f"{exp}"
27finally:
28row["Продукт УП"]=f"ЧЕРН КОЛЕСО Д{diameter} В{weigth_s} {standart}"
29if diameter==473: row["Продукт УП"]="ИНСТРУМЕНТ ДЕФОРМАЦИИ"
30row["Время укрупнения"]=datetime.datetime.now()
31return row
32
33def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
34try:
35if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
36else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
37except: raise
38
39def get_compiled(self):
40code:str=f"print('{__file__}')"
41try:
42with open(__file__,encoding="utf-8",mode="r") as f:
43code=f.read()
44except Exception as exp:
45code=f"print('error: {exp}')"
46finally:
47cc = compile(code,"","exec")
48return cc
49
50if __name__=="__wheel_thermo__":
51consolidator:MainConsolidator
52record:pd.Series
53result:pd.Series
54result = handler.handle(record,consolidator)
55
56