consolidator

1
import os
2
import sys
3
import pandas as pd
4
import datetime
5
from MainConsolidator import MainConsolidator
6

7
class handler(object):
8
    @staticmethod
9
    def module_name():
10
        return 'Труба бесшовная ТПЦ'
11
    @staticmethod
12
    def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:
13
        product_up=None
14
        try:
15
            product = row["Первичный продукт"]
16
            thickness = row["Толщина стенки, мм <Производство>"]
17
            diametr = row["Диаметр, мм <Производство>"]
18
            vm_code = row['Вид материала в SAP ERP'][:4]
19
            steel_mark = row["Марка стали <Производство>"]
20
            
21
            pipe_type = consolidator.scalar_in_guid("Виды материалов ТПЦ",f"`Код`=='{vm_code}'","Тип трубы")
22
            diametr_led = consolidator.nearest_in_guid("Параметры по Продукту ТПЦ",f"`Производство`=='Труба бесшовная ТПЦ'","Диаметр",diametr,"Диаметр")
23
            diametr_led_s = str(diametr_led).replace(".0","")
24
            thickness_range = consolidator.scalar_in_guid("Параметры по Продукту ТПЦ",f"`Производство`=='Труба бесшовная ТПЦ' and `Мин`<={thickness}<=`Макс`","Диапазон")
25
            gmsps = consolidator.get_gmsps(steel_mark,product)
26
            # if pd.isna(gmsps): gmsps = f"[{steel_mark}]" # for debug
27
            print(f"{product}: {thickness}: {diametr} -> {diametr_led_s}: {pipe_type}")
28
            product_up = f"{product} Д{diametr_led_s} {thickness_range} {gmsps}"
29
            if "ИВ" in pipe_type: product_up = f"{product_up} ИВ"
30
            row["error"] = pd.NA
31
        except Exception as exp: 
32
            row["error"] = f"{exp}"
33
        finally:
34
            row["Продукт УП"]=product_up
35
            row["Время укрупнения"]=datetime.datetime.now()
36
            return row
37
        
38
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
39
        try:
40
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
41
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
42
        except: raise
43
    
44
    def get_compiled(self):
45
        code:str=f"print('{__file__}')"
46
        try:
47
            with open(__file__,encoding="utf-8",mode="r") as f:
48
                code=f.read()
49
        except Exception as exp: 
50
            code=f"print('error: {exp}')"
51
        finally:
52
            cc = compile(code,"","exec")
53
            return cc
54

55
if __name__=="__pipe_tpc__":
56
    consolidator:MainConsolidator
57
    record:pd.Series
58
    result:pd.Series
59
    result = handler.handle(record,consolidator)
60

61