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

54
if __name__=="__steel__":
55
    consolidator:MainConsolidator
56
    record:pd.Series
57
    result:pd.Series
58
    result = handler.handle(record,consolidator)
59

60