from models.rank.data.config import ImportanceFeaturesMap
import os
import pandas as pd


def filter_data(data, filter_dict):
    """从shap结果中过滤特征相关性数据"""
    product_content = []
    for key, value in filter_dict.items():
        if key != 'product_code':
            product_content.append(f"{ImportanceFeaturesMap.PRODUCT_FEATRUES_MAP[key]}({value})")
    data = data[data['product_feat'].isin(product_content)]
    return data

def feats_relation_process(shap_result, product_content):
    """生成特征相关性分析报告"""
    # 筛选商户相关性排序结果
    report = filter_data(shap_result, product_content).copy()
    cust_feats_sum = report.groupby("cust_feat")["relation"].sum()
    # 筛选出正相关性的cust_feat
    valid_cust_feats = cust_feats_sum[cust_feats_sum > 0].index.to_list()
    report = report[report["cust_feat"].isin(valid_cust_feats)]
    report = report.reset_index(drop=True)
    report = report.rename(
        columns = {
            "product_feat": "卷烟特征",
            "cust_feat": "商户特征",
            "relation": "相关性"
        }
    )
    return report

def calculate_delivery_by_recommend_data(recommend_data, recommend_cust_infos, delivery_count):
    """根据推荐数据计算投放量，并生成推荐商户报告"""
    recommend_data = pd.DataFrame(recommend_data)
    
    cust_ids = recommend_cust_infos.set_index("BB_RETAIL_CUSTOMER_CODE")
    recommend_data = recommend_data.join(cust_ids, on="cust_code", how="inner")
    recommend_data = recommend_data[["cust_code", "BB_RETAIL_CUSTOMER_NAME", "recommend_score"]]
   # 1. 计算每个商户的理论应得数量（带小数）
    recommend_data["delivery_float"] = (
        recommend_data["recommend_score"] / recommend_data["recommend_score"].sum() * delivery_count
    )

    # 2. 向下取整得到基础配额
    recommend_data["delivery_count"] = recommend_data["delivery_float"].astype(int)

    # 3. 计算余数并排序
    recommend_data["remainder"] = recommend_data["delivery_float"] - recommend_data["delivery_count"]
    recommend_data = recommend_data.sort_values("remainder", ascending=False)

    # 4. 将剩余配额按余数从大到小分配
    remaining = delivery_count - recommend_data["delivery_count"].sum()
    recommend_data.iloc[:remaining, recommend_data.columns.get_loc("delivery_count")] += 1
    
    recommend_data = recommend_data.drop(columns=["delivery_float", "remainder"])
    recommend_data = recommend_data.reset_index()
    # 5. 按recommend_score从大到小重新排序
    recommend_data = recommend_data.sort_values("index")
    # recommend_data["sale_qty"] = recommend_data["sale_qty"].round(0).astype(int) # 将月均销量四舍五入取整
    recommend_data = recommend_data.rename(
        columns={
            "index": "推荐序号", 
            "cust_code": "商户编号", 
            "BB_RETAIL_CUSTOMER_NAME": "商户名称", 
            # "sale_qty": "历史月均销量", 
            "recommend_score": "推荐系数", 
            "delivery_count": "建议投放量（条）"
            }
        )
    recommend_data["推荐序号"] = recommend_data["推荐序号"] + 1
    
    return recommend_data

def eval_report_process(eval_order_data, recommend_data):
    # 获取订单数据并处理
    eval_order_data = eval_order_data[["cust_code", "cust_name", "product_code", "product_name", "sale_qty", "sale_amt"]]
    
    # 确保cust_code是字符串类型
    eval_order_data["cust_code"] = eval_order_data["cust_code"].astype(str)
    
    eval_order_data = eval_order_data.groupby(["cust_code", "cust_name", "product_code", "product_name"])[["sale_qty", "sale_amt"]].mean().reset_index()
    eval_order_data["sale_qty"] = eval_order_data["sale_qty"].round(0).astype(int)
    eval_order_data = eval_order_data.sort_values("sale_qty", ascending=False)
    
    # recommend_data = recommend_data.drop(columns=["sale_qty"])
    # 确保recommend_data中的cust_code也是字符串类型
    recommend_data["商户编号"] = recommend_data["商户编号"].astype(str)
    cust_ids = recommend_data.set_index("商户编号")
    
    # 执行合并操作
    merge_data = eval_order_data.join(cust_ids, on="cust_code", how="left")
    merge_data = merge_data[["cust_code", "cust_name", "product_code", "product_name", "sale_qty", "推荐序号", "推荐系数"]]
    merge_data = merge_data.rename(
        columns={
            "cust_code": "商户编号", 
            "cust_name": "商户名称", 
            "product_code": "卷烟编码", 
            "product_name": "卷烟名称", 
            "sale_qty": "月均销量"
        }
    )
    return merge_data

def split_relation_subtable(data, filter_dict, save_dir):
    """拆分卷烟商户特征相关性子表"""
    data = filter_data(data, filter_dict).copy()
    data.to_csv(os.path.join(save_dir, "feats_interaction.csv"), index=False, encoding='utf-8-sig')
    data['group_key'] = data["product_feat"].str.extract(r'^([^(]+)')
    grouped = data.groupby('group_key')
    sub_tables = {
        name: group.drop(columns=['group_key']).sort_values('relation', ascending=False)
        for name, group in grouped
    }
    
    for name, sub_data in sub_tables.items():
        sub_data.to_csv(os.path.join(save_dir, f"{name}.csv"), index=False, encoding='utf-8-sig')