consolidator
61 строка · 3.0 Кб
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:
13product_up=None
14try:
15product = row["Первичный продукт"]
16carving_type = consolidator.carving_type(row["Тип резьбы <Производство>"])
17diametr_led = consolidator.nearest_in_guid("Параметры по Продукту ТЭСЦ-5",
18f"`Производство`=='Муфта ТЭСЦ-5' and `Продукт`=='{product}'",
19"Доп параметр1",
20row["Диаметр, мм <Производство>"],
21"Доп параметр2")
22diametr_led_s = str(diametr_led).replace(".0","")
23strength_group = consolidator.scalar_in_guid("Группа по классу прочности для муфт",
24f"`Марка стали`=='{row['Класс прочности <Производство>']}'",
25"Группа прочности")
26# if pd.isna(strength_group): strength_group="<NA>"
27print(f"{product} ; carving_type: {carving_type}; diametr_led_s: {diametr_led_s}; strength_group: {strength_group}")
28# Наименование_УП = Продукт & Диам_тип & " " & Гр_прочн & " " & Резьба_тип
29product_up=f"{product}{diametr_led_s} {strength_group} {carving_type}"
30### НКТ-костыль issue http://vsys01775:8282/flea/pyexcelcons3/-/issues/4 ###
31if "НКТ" in str(row["Полное наименование материала"]):
32product_up=f"{product}{diametr_led_s} НКТ {strength_group} {carving_type}"
33except Exception as exp:
34row["error"] = f"{exp}"
35finally:
36row["Продукт УП"]=product_up
37row["Время укрупнения"]=datetime.datetime.now()
38return row
39
40def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
41try:
42if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
43else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
44except: raise
45
46def get_compiled(self):
47code:str=f"print('{__file__}')"
48try:
49with open(__file__,encoding="utf-8",mode="r") as f:
50code=f.read()
51except Exception as exp:
52code=f"print('error: {exp}')"
53finally:
54cc = compile(code,"","exec")
55return cc
56
57if __name__=="__sleeve_cpm__":
58consolidator:MainConsolidator
59record:pd.Series
60result:pd.Series
61result = handler.handle(record,consolidator)
62
63