pepperbot/apps/riic_report_analysis.py

from pepperbot.core.message.chain import MessageChain, Image, Text, At
from rapidocr_onnxruntime import RapidOCR
import os
import asyncio
import functools
import re
from typing import Dict


rapid_ocr = None
if not rapid_ocr:
    rapid_ocr = RapidOCR("rapid_ocr.yml")


class RIICReportAnalysis:
    async def group_message(self, chain: MessageChain, raw_event: Dict):
        if len(chain) != 1 or len(chain.images) != 1:
            return
        img_seg: Image = chain[0]
        img_path = await img_seg.download()
        loop = asyncio.get_running_loop()
        result, elapse = await loop.run_in_executor(
            None, functools.partial(rapid_ocr, img_path)
        )
        os.remove(img_path)
        if not result or not "副手简报" in [i[1] for i in result]:
            return

        card = raw_event["sender"]["card"]

        err_msg = "请修改群昵称（群名片），标注基建布局。若使用2电站布局，请额外标注贸易站等级。\n"
        err_msg += "如：\n"
        err_msg += "EE0000 252 11贸\n"
        err_msg += "教捐 252（33贸易站）\n"
        err_msg += "千梵 243烟火 单推狮蝎\n"
        err_msg += "琉璃 243-153"

        use_tequila = False

        if not card:
            await chain.onebot_reply(Text(err_msg))
            return
        elif "153" in card or "243" in card:
            use_tequila = True
        elif "252" in card:
            if m := re.search(r"([1-3][1-3])贸", card):
                if "3" in m.group(1):
                    if m.group(1) != "33":
                        await chain.onebot_reply(Text("暂不支持此类基建布局"))
                        return
                    else:
                        use_tequila = True
            else:
                await chain.onebot_reply(Text(err_msg))
                return
        elif "342" in card:
            if m := re.search(r"([1-3][1-3][1-3])贸", card):
                if "3" in m.group(1):
                    if m.group(1) != "333":
                        await chain.onebot_reply(Text("暂不支持此类基建布局"))
                        return
                    else:
                        use_tequila = True
            else:
                await chain.onebot_reply(Text(err_msg))
                return
        else:
            await chain.onebot_reply(Text(err_msg))
            return

        lmb_height = 0
        lmb_list = []
        date_list = []
        exp_list = []
        gold_list = []
        for i in result:
            if i[1] == "龙门币":
                lmb_height = i[0][0][1]
                break
        for i in result:
            if m := re.search(r"([0-9]+\.[0-9]+)", i[1]):
                if float(i[2]) < 0.6:
                    continue
                date_list.append([int(i[0][0][0]), m.group(1)])
                continue
            if m := re.search(r"EXP([0-9]+)的作战记录", i[1]):
                exp_list.append([int(i[0][0][0]), m.group(1)])
                continue
            if m := re.search(r"([0-9]+)的贵金属", i[1]):
                gold_list.append([int(i[0][0][0]), m.group(1)])
                continue
            if lmb_height - 20 < i[0][0][1] < lmb_height + 20:
                if m := re.search(r"([0-9]+)", i[1]):
                    lmb_list.append([int(i[0][0][0]), m.group(1)])

        lmb_list.sort(key=lambda x: x[0])
        gold_list.sort(key=lambda x: x[0])
        exp_list.sort(key=lambda x: x[0])

        output = ""
        lmb_avg = 0
        order_avg = 0
        gold_avg = 0
        exp_avg = 0
        external_gold = 5000

        try:
            for i in range(len(date_list)):
                lmb_avg += int(lmb_list[2 * i][1])
                order_avg += int(lmb_list[2 * i + 1][1])
                gold_avg += int(gold_list[i][1])
                exp_avg += int(exp_list[i][1])

                # output += f"【{date_list[i][1]}】\n"
                # output += f"💵 {lmb_list[2 * i][1]}订单({lmb_list[2 * i + 1][1]})\n"
                # output += f"🧈 {gold_list[i][1]}赤金\n"
                # output += f"📼 {exp_list[i][1]}经验\n\n"

            lmb_avg /= len(date_list)
            order_avg /= len(date_list)
            gold_avg /= len(date_list)
            exp_avg /= len(date_list)

            if use_tequila:
                trading_efficiency = lmb_avg / 1.62 / 500 * 70.138 / 24 / 60
                gold_by_e = trading_efficiency * 0.07 * 24 * 60 / 72 * 500
                balanced_gold_by_e = lmb_avg - external_gold - gold_by_e
                overflow_gold_by_e = gold_avg - balanced_gold_by_e
                more_exp_by_e = overflow_gold_by_e * 0.8
                balanced_exp_by_e = exp_avg + more_exp_by_e
                lmb_exp_ratio_by_e = (lmb_avg + 43483) / (balanced_exp_by_e + 36284)
                total_by_e = lmb_avg + balanced_exp_by_e

                gold_by_order = order_avg * 0.2 * 500
                balanced_gold_by_order = lmb_avg - external_gold - gold_by_order
                overflow_gold_by_order = gold_avg - balanced_gold_by_order
                more_exp_by_order = overflow_gold_by_order * 0.8
                balanced_exp_by_order = exp_avg + more_exp_by_order
                lmb_exp_ratio_by_order = (lmb_avg + 43483) / (
                    balanced_exp_by_order + 36284
                )
                total_by_order = lmb_avg + balanced_exp_by_order

                output += "====计算结果====\n\n"
                output += f"订单平均：{lmb_avg:.0f}\n\n"
                output += "-----效率算法-----\n"
                output += f"贸易效率：{trading_efficiency * 100:.0f}%\n"
                output += f"等效赤金：{gold_by_e:.0f}\n"
                output += f"经验调平：{balanced_exp_by_e:.0f}\n"
                output += f"钱书比例：{lmb_exp_ratio_by_e:.2f}\n"
                output += f"调平总和：{total_by_e:.0f}\n\n"
                output += "-----数量算法-----\n"
                output += f"等效赤金：{gold_by_order:.0f}\n"
                output += f"经验调平：{balanced_exp_by_order:.0f}\n"
                output += f"钱书比例：{lmb_exp_ratio_by_order:.2f}\n"
                output += f"调平总和：{total_by_order:.0f}"
            else:
                balanced_gold = lmb_avg - external_gold
                overflow_gold = gold_avg - balanced_gold
                more_exp = overflow_gold * 0.8
                balanced_exp = exp_avg + more_exp
                lmb_exp_ratio = (lmb_avg + 43483) / (balanced_exp + 36284)
                total = lmb_avg + balanced_exp

                output += "====计算结果====\n"
                output += f"订单平均：{lmb_avg:.0f}\n"
                output += f"经验调平：{balanced_exp:.0f}\n"
                output += f"钱书比例：{lmb_exp_ratio:.2f}\n"
                output += f"调平总和：{total:.0f}"

            await chain.onebot_reply(Text(output.strip()))
        except:
            err_msg = "出错啦！\n"
            err_msg += f"date_list = {date_list}\n"
            err_msg += f"lmb_list = {lmb_list}\n"
            err_msg += f"gold_list = {gold_list}\n"
            err_msg += f"exp_list = {exp_list}"
            await chain.onebot_reply(Text(err_msg), At("1040110848"))