Merge branch 'master' of https://github.com/Teahouse-Studios/akari-bot

modules/coin/__init__.py

@@ -0,0 +1,75 @@
+import secrets
+from core.builtins.message import MessageSession
+from core.component import on_command,on_regex
+
+MAX_COIN_NUM = 100
+FACE_UP_RATE = 4975  # n/10000 
+FACE_DOWN_RATE = 4975
+COIN_DROP_PLACES = ["地上","桌子上","桌子底下","门口","窗户外","月球"]  # 硬币可能掉落的位置
+
+coin = on_command('coin', developers=['Light-Beacon'], desc='抛n枚硬币')
+
+@coin.handle('[<amount>] {抛n枚硬币}',)
+async def _(msg: MessageSession):
+    amount = msg.parsed_msg.get('<amount>', '1')
+    if not amount.isdigit():
+        await msg.finish('发生错误：无效的硬币个数：' + amount)
+    else:
+        await msg.finish(await flipCoins(int(amount)))
+
+coinrgex = on_regex('coin_regex',
+                  desc='打开后将在发送的聊天内容匹配以下信息时执行对应命令：\n'
+                       '[丢/抛](n)[个/枚]硬币', developers=['Light-Beacon'])
+
+@coinrgex.handle(r"[丢|抛]([0-9]*)?[个|枚]?硬币")
+async def _(message: MessageSession):
+    groups = message.matched_msg.groups()
+    count = groups[0] if groups[0] else '1'
+    count = int(count)
+    await message.finish(await flipCoins(count))
+
+async def flipCoins(count:int):
+    if count > MAX_COIN_NUM:
+        return f"发生错误：你最多只能抛 {MAX_COIN_NUM} 个硬币"
+    if count <= 0:
+        return f"发生错误：你抛了个空气，什么也没发生..."
+    if FACE_UP_RATE + FACE_DOWN_RATE > 10000 or FACE_UP_RATE < 0 or FACE_DOWN_RATE < 0:
+        raise OverflowError("发生错误：硬币概率错误")
+    faceUp = 0
+    faceDown = 0
+    stand = 0
+    for i in range(count):
+        randnum = secrets.randbelow(10000)
+        if randnum < FACE_UP_RATE:
+            faceUp += 1
+        elif randnum < FACE_UP_RATE + FACE_DOWN_RATE:
+            faceDown += 1
+        else:
+            stand += 1
+    head = f"你抛了 {count} 枚硬币，"
+    if count == 1:
+        drop_place = COIN_DROP_PLACES[secrets.randbelow(len(COIN_DROP_PLACES))]
+        head += f"它掉到了{drop_place}...\n"
+        if faceUp:
+            return head + "...是正面！"
+        elif faceDown:
+            return head + "...是反面！"
+        else:
+            return head + "...它立起来了！"
+    else:
+        if not (stand or faceDown):
+            return head + "它们...\n...全是正面！"
+        if not (stand or faceUp):
+            return head + "它们...\n...全是反面！"
+        if not (faceUp or faceDown):
+            return head + "它们...\n...全都立起来了？！"
+        output = head + "其中...\n有"
+        if faceUp:
+            output += f" {faceUp} 枚是正面，"
+        if faceDown:
+            output += f" {faceDown} 枚是反面"
+        if stand:
+            output += f"...还有 {stand} 枚立起来了！"
+        else:
+            output += f"。"
+        return output

modules/random/__init__.py

@@ -3,8 +3,7 @@ import secrets
 from core.builtins import Bot
 from core.component import on_command
 
-r = on_command('random', alias={'rand': 'random', 'rng': 'random', 'coin': 'random choice 正面 反面',
-                                'random coin': 'random choice 正面 反面'}, developers=['Dianliang233'], desc='随机数生成器（密码学安全）', )
+r = on_command('random', alias={'rand': 'random', 'rng': 'random'}, developers=['Dianliang233'], desc='随机数生成器（密码学安全）', )
 
 
 @r.handle('number <min> <max> {生成区间内的随机整数}', )