consolidator

1
import os
2
import requests
3
import json
4
import pandas as pd
5
from MyConfig import MyConfig
6
from NeuroConsolidator import NeuroConsolidator
7
from OraWorker import OraWorker
8

9
cur_path = os.path.dirname(os.path.realpath(__file__))
10
conf =  MyConfig(f"{cur_path}\\config.json")
11
api_url_method= conf.get("NeuroApiUrl",None)
12
neuro = NeuroConsolidator(api_url_method)
13
ora = OraWorker("manage_dw","ufyjwhb","172.17.80.116/scpora",".\\ora")
14

15
def handler(rec:pd.Series)->pd.Series:
16
    try:        
17
        result = neuro.send_single_request(rec,"Вид материала в SAP ERP","Полное наименование материала",n_predictions=2)
18
        product= result[0][0]
19
        probability = result[0][1]  
20
        product2= result[1][0]
21
        probability2 = result[1][1]        
22
        print(f"{rec['Полное наименование материала']} -> {product} : {probability}%")
23
        # print(result)
24
        rec["Продукт УП 1"]=product
25
        rec["Вероятность 1"]=probability
26
        rec["Продукт УП 2"]=product2
27
        rec["Вероятность 2"]=probability2
28
        check = ora.product_check(product,str(rec["ID в SAP ERP VMZ"]))
29
        rec["Есть в УП"] = check[1]
30
        rec["Уже укрупнено в"] = check[0]
31
        rec["Совпадение"] = check[0]==product               
32
        return rec
33
    except Exception as exp:
34
        rec["Продукт УП"]=f"error: {exp}"
35
        rec["Вероятность"]=100
36
        return  rec
37

38

39
if __name__=="__main__":
40
    try:
41
        input_file = f"{cur_path}\\output\\input_data.xlsx"
42
        input_file = "D:\\work\\Укрупнение\\current\\debug_data\\neuro_dump.xlsx"
43
        print(f"work with file: {input_file}")
44
        df = pd.read_excel(input_file)
45
        df = df[["Вид материала в SAP ERP","Полное наименование материала","ID в SAP ERP VMZ"]]
46
        df = df.apply(handler,axis=1)
47
        df.to_excel(f"{cur_path}\\output\\neuro_data.xlsx",index=False)
48
        ora.close()
49
        
50
    except Exception as exp:
51
        print(f"general exception: {exp}")