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

37
            ###### П0130_Наименование_УП_для_труб ###########
38
            if pd.notna(gmsps) and "ГУ" in gmsps:
39
                product_up=f"{product}{diametr_led_s} {thickness_range} {gmsps}"
40
                if en_pr=="EN" and "ПР" in gmsps: product_up=product_up +" EN"
41
            if pd.notna(gmsps) and "ГК" in gmsps:
42
                product_up=f"{product}{diametr_led_s} {gmsps}"
43
                if en_pr=="EN" and "ПР" in gmsps: product_up=product_up +" EN"
44
            if diametr_led==457 and "EN" not in product_up: product_up = product_up+" EN"
45
            #################################################
46
            #             
47
            row["error"] = pd.NA
48
        except Exception as exp: 
49
            row["error"] = f"{exp}"
50
        finally:
51
            row["Продукт УП"]=product_up
52
            row["Время укрупнения"]=datetime.datetime.now()
53
            return row
54
        
55
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
56
        try:
57
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
58
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
59
        except: raise
60
    
61
    def get_compiled(self):
62
        code:str=f"print('{__file__}')"
63
        try:
64
            with open(__file__,encoding="utf-8",mode="r") as f:
65
                code=f.read()
66
        except Exception as exp: 
67
            code=f"print('error: {exp}')"
68
        finally:
69
            cc = compile(code,"","exec")
70
            return cc
71

72
if __name__=="__pipe_thermo_t3__":
73
    consolidator:MainConsolidator
74
    record:pd.Series
75
    result:pd.Series
76
    result = handler.handle(record,consolidator)
77

78