consolidator
59 строк · 2.7 Кб
1import os2import sys3import pandas as pd4import datetime5from MainConsolidator import MainConsolidator6
7class handler(object):8@staticmethod9def module_name():10return 'Муфта ТЭСЦ-5'11@staticmethod12def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:13product_up=None14try: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"Группа прочности")26if pd.isna(strength_group): strength_group="ПD"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}"30row["error"] = pd.NA31except Exception as exp:32row["error"] = f"{exp}"33finally:34row["Продукт УП"]=product_up35row["Время укрупнения"]=datetime.datetime.now()36return row37
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: raise43
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 cc54
55if __name__=="__sleeve_t5__":56consolidator:MainConsolidator57record:pd.Series58result:pd.Series59result = handler.handle(record,consolidator)60
61