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 'Труба изолированная ТЭСЦ-3'
11
    @staticmethod
12
    def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:
13
        product_up = pd.NA
14
        try:
15
            product = row["Первичный продукт"]
16
            diametr = row["Диаметр, мм <Производство>"]
17
            thickness = row["Толщина стенки, мм <Производство>"]
18
            gmsps = row["ГМСПС"]
19
            en_pr=None
20
            if "ТУ" in str(row["НТД качества (трубы стальные) <Производство>"]) or "ГОСТ" in str(row["НТД качества (трубы стальные) <Производство>"]):
21
                en_pr="ПР"
22
            else:
23
                if pd.notna(row["НТД качества (трубы стальные) <Производство>"]):
24
                    en_pr="EN"
25
            diametr_led = consolidator.nearest_in_guid("Параметры по Продукту ТЭСЦ-3",
26
                f"`Производство`=='Труба изолированная ТЭСЦ-3' and `Продукт`=='{product}'",
27
                "Доп параметр1",
28
                diametr,
29
                "Доп параметр1"
30
                )
31
            diametr_led_s=str(diametr_led).replace(".0","")
32
            thickness_range = consolidator.scalar_in_guid("Параметры по Продукту ТЭСЦ-3",
33
                f"`Производство`=='Труба изолированная ТЭСЦ-3' and `Продукт`=='{product}' and {thickness}>=`Мин` and {thickness}<=`Макс` and `Доп параметр2`.str.contains('{gmsps}')",
34
                "Диапазон")
35
            print(f"product: {product}; diametr: {diametr}; thickness: {thickness}; gmsps: {gmsps}; en_pr: {en_pr}; diametr_led: {diametr_led}; thickness_range: {thickness_range}")
36
            product_up = f"{product}{diametr_led_s} {gmsps}"
37

38
            ###### П0130_Наименование_УП_для_труб ###########
39
            if pd.notna(gmsps) and "ГУ" in gmsps:
40
                product_up=f"{product}{diametr_led_s} {thickness_range} {gmsps}"
41
                if en_pr=="EN" and "ПР" in gmsps: product_up=product_up +" EN"
42
            if pd.notna(gmsps) and "ГК" in gmsps:
43
                product_up=f"{product}{diametr_led_s} {gmsps}"
44
                if en_pr=="EN" and "ПР" in gmsps: product_up=product_up +" EN"
45
            if diametr_led==457 and "EN" not in product_up: product_up = product_up+" EN"
46
            #################################################
47

48
            if pd.notna(row["НТД наружного покрытия <Производство>"]) and pd.isna(row["НТД внутреннего покрытия <Производство>"]): product_up=product_up + " ИН"
49
            if pd.isna(row["НТД наружного покрытия <Производство>"]) and pd.notna(row["НТД внутреннего покрытия <Производство>"]): product_up=product_up + " ИВ"
50
            if pd.notna(row["НТД наружного покрытия <Производство>"]) and pd.notna(row["НТД внутреннего покрытия <Производство>"]): product_up=product_up + " ИНВ"
51
            print(product_up)
52
            row["error"] = pd.NA
53
        except Exception as exp: 
54
            row["error"] = f"{exp}"
55
        finally:
56
            row["Продукт УП"]=product_up
57
            row["Время укрупнения"]=datetime.datetime.now()
58
            return row
59
        
60
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
61
        try:
62
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
63
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
64
        except: raise
65
    
66
    def get_compiled(self):
67
        code:str=f"print('{__file__}')"
68
        try:
69
            with open(__file__,encoding="utf-8",mode="r") as f:
70
                code=f.read()
71
        except Exception as exp: 
72
            code=f"print('error: {exp}')"
73
        finally:
74
            cc = compile(code,"","exec")
75
            return cc
76

77
if __name__=="__pipe_insulated_t3__":
78
    consolidator:MainConsolidator
79
    record:pd.Series
80
    result:pd.Series
81
    result = handler.handle(record,consolidator)
82

83