nonebot-plugin-slot-machine 0.0.4__tar.gz

nonebot_plugin_slot_machine-0.0.4/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 万俊辉
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

nonebot_plugin_slot_machine-0.0.4/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-slot-machine
+Version: 0.0.4
+Summary: NoneBot slot machine plugin
+Requires-Python: <4.0,>=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiosqlite>=0.22.1
+Requires-Dist: nonebot-adapter-milky>=1.2.0
+Requires-Dist: nonebot-plugin-alconna>=0.62.0
+Requires-Dist: nonebot-plugin-localstore>=0.7.4
+Requires-Dist: nonebot2[aiohttp,fastapi]>=2.5.0
+Requires-Dist: pillow>=10.0.0
+Provides-Extra: dev
+Requires-Dist: pyright[nodejs]; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Dynamic: license-file
+
+# slot_machine
+
+## How to start
+
+1. generate project using `nb create` .
+2. create your plugin using `nb plugin create` .
+3. writing your plugins under `slot_machine/plugins` folder.
+4. run your bot using `nb run --reload` .
+
+## Documentation
+
+See [Docs](https://nonebot.dev/)

nonebot_plugin_slot_machine-0.0.4/README.md

@@ -0,0 +1,12 @@
+# slot_machine
+
+## How to start
+
+1. generate project using `nb create` .
+2. create your plugin using `nb plugin create` .
+3. writing your plugins under `slot_machine/plugins` folder.
+4. run your bot using `nb run --reload` .
+
+## Documentation
+
+See [Docs](https://nonebot.dev/)

nonebot_plugin_slot_machine-0.0.4/nonebot_plugin_slot_machine.egg-info/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-slot-machine
+Version: 0.0.4
+Summary: NoneBot slot machine plugin
+Requires-Python: <4.0,>=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiosqlite>=0.22.1
+Requires-Dist: nonebot-adapter-milky>=1.2.0
+Requires-Dist: nonebot-plugin-alconna>=0.62.0
+Requires-Dist: nonebot-plugin-localstore>=0.7.4
+Requires-Dist: nonebot2[aiohttp,fastapi]>=2.5.0
+Requires-Dist: pillow>=10.0.0
+Provides-Extra: dev
+Requires-Dist: pyright[nodejs]; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Dynamic: license-file
+
+# slot_machine
+
+## How to start
+
+1. generate project using `nb create` .
+2. create your plugin using `nb plugin create` .
+3. writing your plugins under `slot_machine/plugins` folder.
+4. run your bot using `nb run --reload` .
+
+## Documentation
+
+See [Docs](https://nonebot.dev/)

nonebot_plugin_slot_machine-0.0.4/nonebot_plugin_slot_machine.egg-info/SOURCES.txt

@@ -0,0 +1,15 @@
+LICENSE
+README.md
+pyproject.toml
+nonebot_plugin_slot_machine.egg-info/PKG-INFO
+nonebot_plugin_slot_machine.egg-info/SOURCES.txt
+nonebot_plugin_slot_machine.egg-info/dependency_links.txt
+nonebot_plugin_slot_machine.egg-info/requires.txt
+nonebot_plugin_slot_machine.egg-info/top_level.txt
+slot_machine/plugins/slot_machine/__init__.py
+slot_machine/plugins/slot_machine/algorithm.py
+slot_machine/plugins/slot_machine/constants.py
+slot_machine/plugins/slot_machine/database.py
+slot_machine/plugins/slot_machine/utils.py
+slot_machine/plugins/slot_machine/image/image.png
+slot_machine/plugins/slot_machine/plugins/screw_work/__init__.py

nonebot_plugin_slot_machine-0.0.4/nonebot_plugin_slot_machine.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

nonebot_plugin_slot_machine-0.0.4/nonebot_plugin_slot_machine.egg-info/requires.txt

@@ -0,0 +1,10 @@
+aiosqlite>=0.22.1
+nonebot-adapter-milky>=1.2.0
+nonebot-plugin-alconna>=0.62.0
+nonebot-plugin-localstore>=0.7.4
+nonebot2[aiohttp,fastapi]>=2.5.0
+pillow>=10.0.0
+
+[dev]
+pyright[nodejs]
+ruff

nonebot_plugin_slot_machine-0.0.4/nonebot_plugin_slot_machine.egg-info/top_level.txt

@@ -0,0 +1 @@
+slot_machine

nonebot_plugin_slot_machine-0.0.4/pyproject.toml

@@ -0,0 +1,117 @@
+[project]
+name = "nonebot-plugin-slot-machine"
+version = "0.0.4"
+description = "NoneBot slot machine plugin"
+readme = "README.md"
+requires-python = ">=3.11, <4.0"
+dependencies = [
+    "aiosqlite>=0.22.1",
+    "nonebot-adapter-milky>=1.2.0",
+    "nonebot-plugin-alconna>=0.62.0",
+    "nonebot-plugin-localstore>=0.7.4",
+    "nonebot2[aiohttp,fastapi]>=2.5.0",
+    "pillow>=10.0.0",
+]
+
+[build-system]
+requires = ["setuptools>=69"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+include = ["slot_machine*"]
+
+[tool.setuptools.package-data]
+"slot_machine.plugins.slot_machine" = ["image/*.png"]
+
+[project.optional-dependencies]
+dev = [
+    "pyright[nodejs]",
+    "ruff"
+]
+
+[tool.nonebot]
+plugin_dirs = ["slot_machine/plugins"]
+builtin_plugins = []
+
+[tool.nonebot.adapters]
+nonebot-adapter-milky = [
+    { name = "nonebot-adapter-milky", module_name = "nonebot.adapters.milky" }
+]
+"@local" = []
+
+[tool.nonebot.plugins]
+"@local" = []
+nonebot-plugin-alconna = ["nonebot_plugin_alconna"]
+[tool.ruff]
+line-length = 88
+target-version = "py311"
+
+[tool.ruff.format]
+line-ending = "lf"
+
+[tool.ruff.lint]
+# For more rules, see https://docs.astral.sh/ruff/rules/.
+select = [
+    "F",     # Pyflakes
+    "W",     # pycodestyle warnings
+    "E",     # pycodestyle errors
+    "I",     # isort
+    "C90",   # mccabe
+    "N",     # pep8-naming
+    "PL",    # pylint
+    "UP",    # pyupgrade
+    "YTT",   # flake8-2020
+    "ANN",   # flake8-annotations
+    "ASYNC", # flake8-async
+    "BLE",   # flake8-blind-except
+    "FBT",   # flake8-boolean-trap
+    "B",     # flake8-bugbear
+    "A",     # flake8-builtins
+    "COM",   # flake8-commas
+    "C4",    # flake8-comprehensions
+    "DTZ",   # flake8-datetimez
+    "T10",   # flake8-debugger
+    "ICN",   # flake8-import-conventions
+    "PIE",   # flake8-pie
+    "T20",   # flake8-print
+    "PYI",   # flake8-pyi
+    "Q",     # flake8-quotes
+    "RSE",   # flake8-raise
+    "RET",   # flake8-return
+    "SIM",   # flake8-simplify
+    "SLOT",  # flake8-slots
+    "TID",   # flake8-tidy-imports
+    "TC",    # flake8-type-checking
+    "ARG",   # flake8-unused-arguments
+    "PTH",   # flake8-use-pathlib
+    # "ERA",   # eradicate
+    "FAST",  # FastAPI
+    "PERF",  # Perflint
+    "PGH",   # pygrep-hooks
+    "FURB",  # refurb
+    "TRY",   # tryceratops
+    "RUF",   # Ruff-specific rules
+]
+ignore = [
+    "E402",    # module-import-not-at-top-of-file  # nonebot2 require() violates this
+    "B008",    # function-call-in-default-argument  # nonebot2 Depends() without Annotated violates this
+    "UP037",   # quoted-annotation
+    # "RUF001",  # ambiguous-unicode-character-string
+    # "RUF002",  # ambiguous-unicode-character-docstring
+    # "RUF003",  # ambiguous-unicode-character-comment
+    # "ANN201",  # missing-return-type-undocumented-public-function
+    "ANN202",  # missing-return-type-private-function
+    "ANN401",  # any-type
+    "COM812",  # missing-trailing-comma
+    "PLC0415", # import-outside-top-level
+]
+allowed-confusables = ["，", "。", "“", "”", "：", "；", "？", "！", "【", "】", "《", "》", "…", "—", "（", "）", "、"]
+
+[tool.ruff.lint.pyupgrade]
+keep-runtime-typing = true
+
+[tool.pyright]
+# For Pylance/Pyright configurations, see https://microsoft.github.io/pyright/#/configuration.
+pythonVersion = "3.11"
+pythonPlatform = "All"
+typeCheckingMode = "standard"

nonebot_plugin_slot_machine-0.0.4/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

nonebot_plugin_slot_machine-0.0.4/slot_machine/plugins/slot_machine/__init__.py

@@ -0,0 +1,307 @@
+import asyncio
+from decimal import Decimal, InvalidOperation
+from pathlib import Path
+
+from nonebot import get_driver, load_plugins, logger, on_command
+from nonebot.adapters.milky import MessageSegment
+from nonebot.adapters.milky.event import MessageEvent
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import Alconna, Args, CommandMeta, on_alconna
+
+from .algorithm import (
+    BetConfigError,
+    CascadeResult,
+    calculate_allowed_win_probability,
+    format_bet_summary,
+    parse_bet_config,
+    resolve_controlled_spin,
+)
+from .constants import REGISTRATION_REWARD
+from .database import (
+    apply_spin_result,
+    get_bet_setting,
+    get_user,
+    initialize_database,
+    register_user,
+    transfer_user_coins,
+    upsert_bet_setting,
+)
+from .utils import (
+    GeneratedSpinImage,
+    SpinMessageContext,
+    build_forward_message,
+    draw_spin_result_image,
+    format_decimal,
+)
+
+__plugin_meta__ = PluginMetadata(
+    name="slot_machine",
+    description="一个老虎机游戏插件",
+    usage=(
+        "注册老虎机/注册：领取初始金币\n"
+        "设置投注 <投注大小> <投注倍数>：保存投注设置\n"
+        "查询老虎机/查询：查看账号和投注信息\n"
+        "开始旋转：开始抽奖"
+    ),
+)
+
+load_plugins(str(Path(__file__).parent / "plugins"))
+
+slot_register = on_command(
+    "注册老虎机",
+    aliases={"注册"},
+    block=True,
+)
+slot_setting = on_alconna(
+    Alconna(
+        "设置投注",
+        Args["bet_size", str]["multiplier", int],
+        meta=CommandMeta(description="设置老虎机投注大小和投注倍数"),
+    ),
+    aliases={"setslot"},
+    block=True,
+)
+slot_transfer = on_alconna(
+    Alconna(
+        "转账",
+        Args["receiver_account", str]["amount", str],
+        meta=CommandMeta(description="向其他老虎机账号转账金币"),
+    ),
+    block=True,
+)
+slot_query = on_command(
+    "查询老虎机",
+    aliases={"查询"},
+    block=True,
+)
+slot_machine = on_command(
+    "开始旋转",
+    block=True,
+)
+
+@get_driver().on_startup
+async def startup_slot_machine() -> None:
+    await initialize_database()
+
+
+async def send_spin_result(event: MessageEvent, context: SpinMessageContext) -> None:
+    images: list[GeneratedSpinImage] = []
+    try:
+        images = list(
+            await asyncio.gather(
+                *(
+                    draw_spin_result_image(
+                        context=context,
+                        cascade=cascade,
+                        cascade_index=index,
+                    )
+                    for index, cascade in enumerate(context.all_cascades, start=1)
+                )
+            )
+        )
+
+        if len(images) <= 3:  # noqa: PLR2004
+            for image in images[:-1]:
+                await slot_machine.send(MessageSegment.image(raw=image.data))
+            await slot_machine.finish(MessageSegment.image(raw=images[-1].data))
+
+        await slot_machine.finish(build_forward_message(event, context, images))
+    finally:
+        for image in images:
+            image.path.unlink(missing_ok=True)
+
+
+async def send_miss_result(context: SpinMessageContext, board: list[list[str]]) -> None:
+    image = await draw_spin_result_image(
+        context,
+        CascadeResult(
+            board=board,
+            highlighted_positions=frozenset(),
+            bonus_multiplier=1,
+            payout=Decimal(0),
+        ),
+    )
+    try:
+        await slot_machine.finish(MessageSegment.image(raw=image.data))
+    finally:
+        image.path.unlink(missing_ok=True)
+
+
+@slot_register.handle()
+async def handle_slot_register(event: MessageEvent) -> None:
+    account = event.get_user_id()
+    registered = await register_user(account)
+    user = await get_user(account)
+
+    if user is None:
+        await slot_register.finish("注册失败，请稍后再试。")
+
+    if not registered:
+        await slot_register.finish(
+            "你已经注册过了。\n"
+            f"账号：{account}\n"
+            f"当前金币：{format_decimal(user.coins)}\n"
+            f"抽奖次数：{user.spin_count}"
+        )
+
+    await slot_register.finish(
+        "注册成功。\n"
+        f"账号：{account}\n"
+        f"赠送金币：{format_decimal(REGISTRATION_REWARD)}\n"
+        f"当前金币：{format_decimal(user.coins)}"
+    )
+
+
+@slot_setting.handle()
+async def handle_slot_setting(
+    event: MessageEvent,
+    bet_size: str,
+    multiplier: int,
+) -> None:
+    account = event.get_user_id()
+    user = await get_user(account)
+
+    if user is None:
+        await slot_setting.finish("你还没有注册。\n请先发送：注册老虎机")
+
+    try:
+        bet_config = parse_bet_config(f"{bet_size} {multiplier}")
+    except BetConfigError:
+        await slot_setting.finish(
+            "请输入：设置投注 <投注大小> <投注倍数>，例如：设置投注 0.2 5\n"
+            "投注大小只能是 0.02、0.2 或 1，投注倍数只能是 1 到 10 的整数。"
+        )
+
+    await upsert_bet_setting(account, bet_config)
+    await slot_setting.finish(f"已保存你的投注设置。\n{format_bet_summary(bet_config)}")
+
+
+@slot_transfer.handle()
+async def handle_slot_transfer(
+    event: MessageEvent,
+    receiver_account: str,
+    amount: str,
+) -> None:
+    sender_account = event.get_user_id()
+    sender = await get_user(sender_account)
+    if sender is None:
+        await slot_transfer.finish("你还没有注册。\n请先发送：注册老虎机")
+
+    try:
+        transfer_amount = Decimal(amount)
+    except InvalidOperation:
+        await slot_transfer.finish("请输入正确金额，例如：转账 123456 10")
+
+    if transfer_amount <= 0:
+        await slot_transfer.finish("转账金额必须大于 0。")
+
+    receiver = await get_user(receiver_account)
+    if receiver is None:
+        await slot_transfer.finish("对方还没有注册老虎机账号。")
+
+    if sender_account == receiver_account:
+        await slot_transfer.finish("不能给自己转账。")
+
+    if sender.coins < transfer_amount:
+        await slot_transfer.finish(
+            "金币不足，无法转账。\n"
+            f"当前金币：{format_decimal(sender.coins)}\n"
+            f"转账金额：{format_decimal(transfer_amount)}"
+        )
+
+    updated_sender, updated_receiver = await transfer_user_coins(
+        sender_account,
+        receiver_account,
+        transfer_amount,
+    )
+    await slot_transfer.finish(
+        "转账成功。\n"
+        f"收款账号：{receiver_account}\n"
+        f"转账金额：{format_decimal(transfer_amount)} 金币\n"
+        f"你的剩余金币：{format_decimal(updated_sender.coins)}\n"
+        f"对方当前金币：{format_decimal(updated_receiver.coins)}"
+    )
+
+
+@slot_query.handle()
+async def handle_slot_query(event: MessageEvent) -> None:
+    account = event.get_user_id()
+    user = await get_user(account)
+
+    if user is None:
+        await slot_query.finish("你还没有注册。\n请先发送：注册老虎机")
+
+    bet_config = await get_bet_setting(account)
+    bet_text = (
+        format_bet_summary(bet_config)
+        if bet_config is not None
+        else "尚未设置投注"
+    )
+    await slot_query.finish(
+        "老虎机账号信息\n"
+        f"账号：{account}\n"
+        f"当前金币：{format_decimal(user.coins)}\n"
+        f"抽奖次数：{user.spin_count}\n"
+        f"中奖次数：{user.win_count}\n"
+        f"累计获得：{format_decimal(user.total_payout)} 金币\n\n"
+        f"当前投注：\n{bet_text}"
+    )
+
+
+@slot_machine.handle()
+async def handle_slot_machine(event: MessageEvent) -> None:
+    account = event.get_user_id()
+    user = await get_user(account)
+
+    if user is None:
+        await slot_machine.finish("你还没有注册。\n请先发送：注册老虎机")
+
+    bet_config = await get_bet_setting(account)
+    if bet_config is None:
+        await slot_machine.finish(
+            "你还没有设置投注。\n"
+            "请先发送：设置投注 <投注大小> <投注倍数>\n"
+            "例如：设置投注 0.2 5"
+        )
+
+    if user.coins < bet_config.total_bet:
+        await slot_machine.finish(
+            "金币不足，无法抽奖。\n"
+            f"当前金币：{format_decimal(user.coins)}\n"
+            f"本次需要：{format_decimal(bet_config.total_bet)}"
+        )
+
+    allowed_probability = calculate_allowed_win_probability(
+        user.coins,
+        user.win_count,
+        user.total_payout,
+    )
+    logger.info(
+        f"老虎机抽奖概率 | 账号：{account} | "
+        f"金币：{format_decimal(user.coins)} | "
+        f"中奖次数：{user.win_count} | "
+        f"累计获得：{format_decimal(user.total_payout)} | "
+        f"本次概率：{allowed_probability:.2%}"
+    )
+
+    spin_result = resolve_controlled_spin(
+        bet_config,
+        user.coins,
+        user.win_count,
+        user.total_payout,
+    )
+    updated_user = await apply_spin_result(
+        account, bet_config.total_bet, spin_result.total_payout
+    )
+    context = SpinMessageContext(
+        account=account,
+        total_bet=format_decimal(bet_config.total_bet),
+        total_payout=format_decimal(spin_result.total_payout),
+        remaining_coins=format_decimal(updated_user.coins),
+        all_cascades=list(spin_result.cascades),
+    )
+
+    if not spin_result.cascades:
+        await send_miss_result(context, spin_result.final_board)
+
+    await send_spin_result(event, context)