from pepperbot.core.message.chain import MessageChain, Image, Text, At from pepperbot.extensions.command import as_command, CLIArgument from pepperbot.extensions.command.sender import CommandSender from pepperbot.adapters.universal.api import UniversalGroupBot from rapidocr_onnxruntime import RapidOCR import os import asyncio import functools import re from typing import Dict, Optional, Any import math 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, bot: UniversalGroupBot, ): 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 += "EE0000 252 21贸\n" err_msg += "教捐 252（33贸易站）\n" err_msg += "千梵 243烟火 单推狮蝎\n" err_msg += "琉璃 243-153" use_tequila = False power_plants = 2 if not card: await chain.onebot_reply(Text(err_msg)) return elif "153" in card or "243" in card: use_tequila = True power_plants = 3 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_list = [] output = "" lmb_avg = 0 order_avg = 0 gold_avg = 0 exp_avg = 0 external_gold = 5000 external_lmb = 43483 external_exp = 36284 try: exp_list += [["manual", "0"]] * (len(date_list) - len(exp_list)) 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" output_list.append(output.strip()) output = "" 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 + external_lmb) / ( balanced_exp_by_e + external_exp ) 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 + external_lmb) / ( balanced_exp_by_order + external_exp ) total_by_order = lmb_avg + balanced_exp_by_order output += "====计算结果====\n\n" output += f"订单平均：{lmb_avg:.0f}\n" if power_plants == 3: output += f"赤金平均：{gold_avg:.0f}\n" output += f"经验平均：{exp_avg:.0f}\n" if exp_avg != 0: output += f"产出钱书比例：{lmb_avg / exp_avg:.2f}\n" else: output += f"产出钱书比例：正无穷\n" output += f"综合钱书比例：{(lmb_avg + external_lmb) / (exp_avg + external_exp):.2f}\n" output += f"钱书总和：{lmb_avg + exp_avg:.0f}\n" output += "\n-----效率算法-----\n" output += f"贸易效率：{trading_efficiency:.0%}\n" output += f"等效赤金：{gold_by_e:.0f}\n" output += f"经验调平：{balanced_exp_by_e:.0f}\n" if power_plants == 2: 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" if power_plants == 2: output += f"钱书比例：{lmb_exp_ratio_by_order:.2f}\n" output += f"调平总和：{total_by_order:.0f}\n\n" 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 + external_lmb) / (balanced_exp + external_exp) total = (lmb_avg * 0.2 + gold_avg * 0.8 + exp_avg + 4000) / 10000 gongsun = (lmb_avg * 0.8 + gold_avg * 0.8 + exp_avg + 4000) / 10000 kaltsit = math.ceil((lmb_avg + gold_avg + exp_avg) / 10000) stupid = (lmb_avg + exp_avg) / 10000 output += "====计算结果====\n" output += f"订单平均：{lmb_avg:.0f}\n" output += f"赤金平均：{gold_avg:.0f}\n" output += f"经验平均：{exp_avg:.0f}\n" output += f"经验调平：{balanced_exp:.0f}\n" output += f"钱书比例：{lmb_exp_ratio:.2f}\n\n" output += "======产能======\n" output += f"{total:.2f}E = 0.2贸 + 0.8金 + 书 + 4000\n" output += f"{gongsun:.2f}公 = 0.8贸 + 0.8金 + 书 + 4000\n" output += f"{kaltsit:.0f}凯 = 贸 + 金 + 书向上取整到万\n" output += f"{stupid:.1f}友 = 贸 + 书" output_list.append(output.strip()) await bot.arbitrary.onebot.send_group_forward_msg( raw_event["group_id"], [ { "type": "node", "data": { "name": card, "uin": raw_event["user_id"], "content": [{"type": "text", "data": {"text": i}}], }, } for i in output_list ], ) return except Exception as e: 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}\n" err_msg += str(e) if str(e) != "timed out": await chain.onebot_reply(Text(err_msg), At("1040110848")) @as_command( need_prefix=False, aliases=["裁缝计算"], include_class_name=False, require_at=False, ) class TailoringCommand: async def initial(self, sender: CommandSender, arg: Optional[Any] = CLIArgument()): if not arg or not isinstance(arg, str): await sender.reply(Text("请输入效率！")) return if arg.endswith("%"): arg = arg[:-1] try: efficiency = float(arg) except: await sender.reply(Text("输入无效！")) return if efficiency > 30: efficiency /= 100 total_efficiency = efficiency + 1.1 equiv_efficiency = total_efficiency * 1.2410210128 gold_efficiency = total_efficiency * (1.2410210128 - 1.0141677) * 1.025 lmb = equiv_efficiency * 24 * 60 / 70.137931 * 500 gold = gold_efficiency * 24 * 60 / 72 * 500 output = f"整站效率：{total_efficiency:.0%}\n" output += f"等效贸易：{equiv_efficiency - 1:.0%}\n" output += f"等效制造：{gold_efficiency:.0%}\n" output += f"订单收益：{lmb:.0f}\n" output += f"赤金收益：{gold:.0f}" await sender.reply(Text(output))