first_commit .gitignore

plugins/nonebot_plugin_cocdicer/san_check.py

@@ -0,0 +1,38 @@
+from .data_source import Dices
+from .messages import sc_help_message
+
+import re
+
+
+def number_or_dice(arg: str):
+    if "d" in arg:
+        d = Dices()
+        if dices := re.search(r"\d+d", arg):
+            d.dices = int(dices.group()[:-1])
+        if faces := re.search(r"d\d+", arg):
+            d.faces = int(faces.group()[1:])
+        d.roll()
+        return d
+    else:
+        return int(arg)
+
+
+def sc(arg: str) -> str:
+    a_num = re.search(r" \d+", arg)
+    success = re.search(r"\d*d\d+|\d+", arg)
+    failure = re.search(r"[\/]+(\d*d\d+|\d+)", arg)
+    if not (a_num and success and failure):
+        return sc_help_message
+    check_dice = Dices()
+    check_dice.a = True
+    check_dice.anum = int(a_num.group()[1:])
+    success = number_or_dice(success.group())
+    failure = number_or_dice(failure.group()[1:])
+    r = "San Check" + check_dice.roll()[4:]
+    result = success if check_dice.result <= check_dice.anum else failure
+    r += "\n理智降低了"
+    if type(result) == int:
+        r += "%d点" % result
+    else:
+        r = r + result._head + str(result.result)
+    return r