consolidator
59 строк · 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["Первичный продукт"]
16product = row["Первичный продукт"]
17diametr = row["Диаметр, мм <Производство>"]
18vm_code = row['Вид материала в SAP ERP'][:4]
19steel_mark = row["Марка стали <Производство>"]
20carving = row["Тип резьбы <Производство>"]
21strength = consolidator.strength_by_steel(steel_mark) if pd.isna(row["Класс прочности <Производство>"]) else row["Класс прочности <Производство>"]
22carving_led= "БР" if pd.isna(carving) else consolidator.scalar_in_guid("Резьба",f"`Исх резьба`=='{carving}'","Результ Резьба")
23diametr_led = consolidator.nearest_in_guid("Параметры по Продукту ТПЦ",f"`Производство`=='Труба обсадная бесшовная ТПЦ'","Диаметр",diametr,"Диаметр")
24diametr_led_s = str(diametr_led).replace(".0","")
25strength_group = consolidator.scalar_in_guid("Группа по классу прочности для труб ТПЦ",f"`Марка стали`=='{strength}'","Группа прочности")
26print(f"{strength}: {carving_led} : {strength_group}")
27
28product_up=f"{product} Д{diametr_led_s} ГО {strength_group} {carving_led}"
29
30row["error"] = pd.NA
31except Exception as exp:
32row["error"] = f"{exp}"
33finally:
34row["Продукт УП"]=product_up
35row["Время укрупнения"]=datetime.datetime.now()
36return row
37
38def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
39try:
40if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
41else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
42except: raise
43
44def get_compiled(self):
45code:str=f"print('{__file__}')"
46try:
47with open(__file__,encoding="utf-8",mode="r") as f:
48code=f.read()
49except Exception as exp:
50code=f"print('error: {exp}')"
51finally:
52cc = compile(code,"","exec")
53return cc
54
55if __name__=="__pipe_casing_tpc__":
56consolidator:MainConsolidator
57record:pd.Series
58result:pd.Series
59result = handler.handle(record,consolidator)
60
61