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
            diametr = row["Диаметр, мм <Производство>"]
17
            strength_1 = row["Класс прочности <Производство>"]
18
            steel_mark = row["Марка стали <Производство>"]
19
            strength_steel = consolidator.strength_mapping(strength_1,steel_mark)
20
            thickness = row["Толщина стенки, мм <Производство>"]
21
            seam_ = row["Тип трубы <Производство>"]
22
            seam=None
23
            if pd.notna(seam_) and seam_=="Одношовная": seam="ШВ1"
24
            elif pd.notna(seam_): seam="ШВ2"
25
            
26
            if diametr==508 : 
27
                product_up="ТРУБА Д508 ГН"
28
                print(product_up)
29
            else:
30
                strength_led = consolidator.strength_led(strength_steel[0])
31
                if strength_led=="К0": strength_led="К52"
32
                diametr_led = consolidator.nearest_in_guid("Параметры по Продукту ТБД", f"`Продукт`=='{product}'", "Доп параметр1", diametr, "Доп параметр1")
33
                thickness_led = consolidator.nearest_in_guid("Параметры по Продукту ТБД", 
34
                    f"`Продукт`=='{product}' and `Доп параметр2`=='{seam}' and `Доп параметр1`=={diametr_led}", "Мин", thickness, "Мин")
35
                diametr_led_s = str(diametr_led).replace(".0","")
36
                thickness_led_s = str(thickness_led).replace(".0","")                
37
                thickness_range = consolidator.scalar_in_guid("Параметры по Продукту ТБД",
38
                    f"`Доп параметр3` == '{strength_led}' and `Доп параметр1`=={diametr_led} and `Мин`<={thickness_led_s}<=`Макс` and `Доп параметр2`=='{seam}'","Диапазон")
39
                print(f"strength_steel: {strength_steel}; diametr_led_s: {diametr_led_s}; strength_led: {strength_led}; thickness_range: {thickness_range}; thickness: {thickness}")
40
                # Наименование_УП = Продукт & Диам_тип & " " & Диап_толщин & " " & Шов & " " & Кл_прочности_прив
41
                product_up=f"{product}{diametr_led_s} {thickness_range} {seam} {strength_led}"
42
                if pd.notna(row["НТД внутреннего покрытия <Производство>"]) and pd.notna(row["НТД наружного покрытия <Производство>"]): product_up = product_up + " ИНВ"
43
                elif pd.notna(row["НТД внутреннего покрытия <Производство>"]): product_up = product_up + " ИВ"
44
                elif pd.notna(row["НТД наружного покрытия <Производство>"]): product_up = product_up + " ИН"
45
            row["error"] = pd.NA
46
        except Exception as exp: 
47
            row["error"] = f"{exp}"
48
        finally:
49
            row["Продукт УП"] = product_up
50
            row["Время укрупнения"]=datetime.datetime.now()
51
            return row
52
        
53
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
54
        try:
55
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
56
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
57
        except: raise
58
    
59
    def get_compiled(self):
60
        code:str=f"print('{__file__}')"
61
        try:
62
            with open(__file__,encoding="utf-8",mode="r") as f:
63
                code=f.read()
64
        except Exception as exp: 
65
            code=f"print('error: {exp}')"
66
        finally:
67
            cc = compile(code,"","exec")
68
            return cc
69

70
if __name__=="__pipe_tbd__":
71
    consolidator:MainConsolidator
72
    record:pd.Series
73
    result:pd.Series
74
    result = handler.handle(record,consolidator)
75

76