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
            steel_mark = row["Марка стали <Производство>"]
18
            strength_1 = steel_mark if pd.isna(row["Класс прочности <Производство>"]) and pd.notna(steel_mark) else row["Класс прочности <Производство>"]
19
            strength_class = consolidator.strength_by_steel(steel_mark)
20
            strength_2 = consolidator.strength_mapping(strength_class,steel_mark)
21
            strength_led = consolidator.strength_led(strength_2[0])
22
            strength_led_int = int(strength_led[1:])
23
            gmsps = consolidator.get_gmsps(steel_mark,product)
24
            
25
            product_up=f"{product} {strength_led}"
26
            if "ГУ" in gmsps: product_up=f"{product} ГУ"
27
            elif strength_led_int >=60:
28
                thickness_led = consolidator.nearest_in_guid("Параметры по Продукту Прокат",
29
                f"`Производство`=='Сляб стан 5000' and `Продукт`=='{product}' and `Доп параметр2`=='{strength_led}'",
30
                "Доп параметр1",thickness,"Доп параметр1")
31
                thickness_led_s = str(thickness_led).replace(".0","")
32
                product_up=f"{product} Т{thickness_led_s} {strength_led}"
33
                print(f"{product} - strength_led: {strength_led}; gmsps: {gmsps}; strength_led_int: {strength_led_int} {thickness}->{thickness_led}")
34
            row["error"] = pd.NA
35
        except Exception as exp: 
36
            row["error"] = f"{exp}"
37
        finally:
38
            row["Продукт УП"]=product_up
39
            row["Время укрупнения"]=datetime.datetime.now()
40
            return row
41
        
42
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
43
        try:
44
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
45
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
46
        except: raise
47
    
48
    def get_compiled(self):
49
        code:str=f"print('{__file__}')"
50
        try:
51
            with open(__file__,encoding="utf-8",mode="r") as f:
52
                code=f.read()
53
        except Exception as exp: 
54
            code=f"print('error: {exp}')"
55
        finally:
56
            cc = compile(code,"","exec")
57
            return cc
58

59
if __name__=="__half_roll__":
60
    consolidator:MainConsolidator
61
    record:pd.Series
62
    result:pd.Series
63
    result = handler.handle(record,consolidator)
64

65