# 参考[OlivaDiceDocs](https://oliva.dicer.wiki/userdoc)实现的nonebot2骰娘插件 import re import random from .messages import * class Mylist(list): def next(self, index: int): if index < self.__len__()-1: return self[index+1] else: return "" def help_message(): return main_help_message def dhr(t, o): if t == 0 and o == 0: return 100 else: return t*10+o def st(): result = random.randint(1, 20) if result < 4: rstr = "右腿" elif result < 7: rstr = "左腿" elif result < 11: rstr = "腹部" elif result < 16: rstr = "胸部" elif result < 18: rstr = "右臂" elif result < 20: rstr = "左臂" elif result < 21: rstr = "头部" return "D20=%d：命中了%s" % (result, rstr) def en(arg: str) -> str: try: arg = int(arg) except: return en_help_message check = random.randint(1, 100) if check > arg or check > 95: plus = random.randint(1, 10) r = "判定值%d，判定成功，技能成长%d+%d=%d" % (check, arg, plus, arg+plus) return r + "\n温馨提示：如果技能提高到90%或更高，增加2D6理智点数。" else: return "判定值%d，判定失败，技能无成长。" % check class Dices(object): def __init__(self): self.dices = 1 self.faces = 100 self.a = False self.anum = 0 self.h = False self.times = 1 self.bp = 0 self._bp_result = "" self._tens_place = None self._ones_place = None self._head = "" self._mid = "" self._end = "" self.result = 0 self.a_check_mode = self.a_check self._a_check_result = "" self.ex_dice = None self.ex_dice_type = 1 self._ex_result = "" def real_dice(self): if self.faces == 100: self._tens_place = random.randint(0, 9) self._ones_place = random.randint(0, 9) self.result += dhr(self._tens_place, self._ones_place) return dhr(self._tens_place, self._ones_place) else: rint = random.randint(1, self.faces) self.result += rint return rint def bp_dice(self): if not self.bp or self.faces != 100 or self.dices != 1: self._bp_result = "" return self._bp_result self._bp_result = " -> 十位：%d，个位：%d" % ( self._tens_place, self._ones_place) bp = self.bp while bp > 0: bd = random.randint(0, 9) self._bp_result += "，奖励：%d" % bd if dhr(bd, self._ones_place) < dhr(self._tens_place, self._ones_place): self._tens_place = bd self.result = dhr(self._tens_place, self._ones_place) bp -= 1 while bp < 0: bd = random.randint(0, 9) self._bp_result += "，惩罚：%d" % bd if dhr(bd, self._ones_place) > dhr(self._tens_place, self._ones_place): self._tens_place = bd self.set_result() bp += 1 return self._bp_result def a_check(self): if not (self.a and self.anum and self.faces == 100 and self.dices == 1): self._a_check_result = "" return self._a_check_result if self.result == 100: self._a_check_result = " 大失败！" elif self.anum < 50 and self.result > 95: self._a_check_result = "\n检定值%d %d>95 大失败！" % ( self.anum, self.result) elif self.result == 1: self._a_check_result = " 大成功！" elif self.result <= self.anum // 5: self._a_check_result = "\n检定值%d %d≤%d 极难成功" % ( self.anum, self.result, self.anum // 5) elif self.result <= self.anum // 2: self._a_check_result = "\n检定值%d %d≤%d 困难成功" % ( self.anum, self.result, self.anum // 2) elif self.result <= self.anum: self._a_check_result = "\n检定值%d %d≤%d 成功" % ( self.anum, self.result, self.anum) else: self._a_check_result = "\n检定值%d %d>%d 失败" % ( self.anum, self.result, self.anum) return self._a_check_result def xdy(self): self.result = 0 if self.dices == 1: self.real_dice() self.bp_dice() self.a_check() return "" else: dice_results = [] for _ in range(self.dices): dice_result = self.real_dice() dice_results.append(str(dice_result)) return "(%s)" % "+".join(dice_results) def _ex_handle(self): if not self.ex_dice: self._ex_result = "" elif type(self.ex_dice) == int: ex_result = self.ex_dice_type*self.ex_dice self._ex_result = "%s%s%d" % (str( self.result) if self.dices == 1 else "", "+" if self.ex_dice_type == 1 else "-", self.ex_dice) self.result += ex_result elif type(self.ex_dice) == Dices: self.ex_dice.roll ex_result = self.ex_dice_type*self.ex_dice.result self._ex_result = "%s%s%d" % (str( self.result) if self.dices == 1 else "", "+" if self.ex_dice_type == 1 else "-", self.ex_dice) self.result += ex_result return self._ex_result def roll(self): r = "%d次投掷：" % self.times if self.times != 1: r += "\n" for _ in range(self.times): xdyr = self.xdy() self._ex_handle() self._head = "%sD%d%s=" % ( "" if self.dices == 1 else str(self.dices), self.faces, "" if not self.ex_dice else ( ("+" if self.ex_dice_type == 1 else "-") + str(self.ex_dice) if type(self.ex_dice) == int else (str(self.ex_dice.dices)+"D"+self.ex_dice.faces)) ) self._mid = "%s%s=" % (xdyr, self._ex_result) self._end = "%d%s%s" % ( self.result, self._bp_result, self._a_check_result) r += "%s%s%s" % (self._head, self._mid if self.dices != 1 or self.ex_dice else "", self._end) self.times -= 1 if self.times: r += "\n" return r def prework(args: list, start=0): for i in range(start, len(args), 1): if not re.search("\\d+", args[i]) and len(args[i]) > 1: p = args.pop(i) for j in list(p): args.insert(i, j) i += 1 if prework(args, i): break return True def rd(arg: str): try: h = False dices = Dices() args = re.split("(\\d+)", arg.lower()) prework(args) args = Mylist(args) for i in range(len(args)): if args[i] == "a": dices.a = True elif args[i] == "#" and re.search("\\d+", args.next(i)): dices.times = int(args[i+1]) elif args[i] == "b": dices.bp += 1 elif args[i] == "p": dices.bp -= 1 elif args[i] == "d" and re.search("\\d+", args.next(i)): dices.faces = int(args[i+1]) elif args[i] == " " and re.search("\\d+", args.next(i)) and dices.a: dices.anum = int(args[i+1]) elif args[i] == "h": h = True elif re.search("\\d+", args[i]): if args.next(i) == "d": dices.dices = int(args[i]) elif args[i-1] == " " and dices.a: dices.anum = int(args[i]) elif args[i] in ["-", "+"]: dices.ex_dice_type = (-1 if args[i] == "-" else 1) if args.next(i+1) == "d": dices.ex_dice = Dices() dices.ex_dice.dices = int(args.next(i)) dices.ex_dice.faces = int(args.next(i+2)) elif args.next(i): dices.ex_dice = int(args.next(i)) if h: return [dices.roll()] return dices.roll() except: return r_help_message if __name__ == "__main__": rd("2d100")