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
            carving = row["Тип резьбы <Производство>"]
16
            strength_class = row["Класс прочности <Производство>"]
17
            steel_mark = row["Марка стали <Производство>"]
18
            diametr = row["Диаметр, мм <Производство>"]
19
            carving_type = consolidator.scalar_in_guid("Резьба",f"`Исх резьба`=='{carving}'","Результ Резьба")
20
            if pd.isna(carving) or pd.isna(carving_type): carving_type="БР"
21
            # strength_group = consolidator.scalar_in_guid("Группа по классу прочности для труб",f"`Марка стали`=='{strength_class}' or `Марка стали`=='{steel_mark}'","Группа прочности")
22
            # if pd.isna(strength_group): strength_group="ПD"
23
            strength_group = consolidator.strength_group_by_steel(strength_class,steel_mark)
24
            diametr_type = consolidator.nearest_in_guid("Параметры по Продукту ТЭСЦ-5",
25
                f"`Производство`=='Труба обсадная ТЭСЦ-5' and `Продукт`=='{product}'",
26
                "Доп параметр1",
27
                diametr,
28
                "Доп параметр1")
29
            diametr_type_s = str(diametr_type).replace(".0","")
30
            print(f"product: {product}; carving: {carving}; strength_group: {strength_group}; diametr_type: {diametr_type_s}")
31
            product_up=f"{product}{diametr_type_s} ГО {strength_group} {carving_type}"
32
            row["error"] = pd.NA
33
        except Exception as exp: 
34
            row["error"] = f"{exp}"
35
        finally:
36
            row["Продукт УП"]=product_up
37
            row["Время укрупнения"]=datetime.datetime.now()
38
            return row
39
        
40
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
41
        try:
42
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
43
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
44
        except: raise
45
    
46
    def get_compiled(self):
47
        code:str=f"print('{__file__}')"
48
        try:
49
            with open(__file__,encoding="utf-8",mode="r") as f:
50
                code=f.read()
51
        except Exception as exp: 
52
            code=f"print('error: {exp}')"
53
        finally:
54
            cc = compile(code,"","exec")
55
            return cc
56

57
if __name__=="__pipe_casing_t3__":
58
    consolidator:MainConsolidator
59
    record:pd.Series
60
    result:pd.Series
61
    result = handler.handle(record,consolidator)
62

63