Skip to content

✨ add draw, dnd and coc 6/7/6d/7d #9

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
RF-Tar-Railt wants to merge 2 commits into
base: master
Choose a base branch
from
Open

✨ add draw, dnd and coc 6/7/6d/7d #9

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bd26279
:recycle: sync changes
RF-Tar-Railt Jan 23, 2023
058b40e
:sparkles: add draw, dnd, coc6/7/6d/7d
RF-Tar-Railt Jan 23, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
171 changes: 122 additions & 49 deletions nonebot_plugin_cocdicer/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,59 +1,71 @@
from .util import MessageEvent, GroupMessageEvent
from .dices import help_message, st, en, rd0
from .madness import ti, li
from .investigator import Investigator
from .san_check import sc
from .cards import _cachepath, cards, cache_cards, set_handler, show_handler, sa_handler, del_handler
from __future__ import annotations

from nonebot import get_driver, get_bot
from nonebot.rule import Rule
from nonebot.matcher import Matcher
from nonebot.plugin import on_startswith
from nonebot import get_bot, get_driver
from nonebot.adapters import Bot as Bot
from nonebot.adapters.onebot.v11 import Bot as V11Bot
from nonebot.adapters.onebot.v12 import Bot as V12Bot
from nonebot.matcher import Matcher
from nonebot.plugin import on_startswith
from nonebot.rule import Rule


import os
from .cards import Cards
from .constant import help_main, helps
from .deck import draw
from .investigator import Investigator
from .pc import coc6, coc6d, coc7, coc7d, dnd
from .rd import dhr, en, expr, long_insane, rd0, roll_success_level, st, temp_insane
from .util import GroupMessageEvent, MessageEvent

driver = get_driver()
coc_cards = Cards("data/coc_cards.json")


@driver.on_startup
async def _(): # 角色卡暂存目录初始化
if not os.path.exists("data"):
os.makedirs("data")
if not os.path.exists(_cachepath):
with open(_cachepath, "w", encoding="utf-8") as f:
f.write("{}")
cards.load()
coc_cards.load()


@driver.on_shutdown
async def _():
coc_cards.save()


def is_group_message() -> Rule:
async def _is_group_message(bot: Bot, event: MessageEvent) -> bool:
return True if type(event) is GroupMessageEvent else False
return type(event) is GroupMessageEvent

return Rule(_is_group_message)


def get_level_uid(event: MessageEvent) -> tuple[str, int]:
return (
str(event.group_id) if isinstance(event, GroupMessageEvent) else "0",
event.sender.user_id or 0,
)


rdhelp = on_startswith(".help", priority=2, block=True)
drawcommand = on_startswith(".draw", priority=2, block=True)
stcommand = on_startswith(".st", priority=2, block=True)
encommand = on_startswith(".en", priority=2, block=True)
ticommand = on_startswith(".ti", priority=2, block=True)
licommand = on_startswith(".li", priority=2, block=True)
dndcommand = on_startswith(".dnd", priority=2, block=True)
coccommand = on_startswith(".coc", priority=2, block=True)
wcoccommand = on_startswith(".wcoc", priority=2, block=True)
sccommand = on_startswith(".sc", priority=2, block=True)
racommand = on_startswith(".ra", priority=5, block=True)
rhcommand = on_startswith(".rh", priority=3, block=True)
rdcommand = on_startswith(".r", priority=4, block=True)
setcommand = on_startswith(".set", priority=5, block=True)
showcommand = on_startswith(".show", priority=5, block=True)
sacommand = on_startswith(".sa", priority=5, block=True)
delcommand = on_startswith(".del", priority=5, block=True)


@rdhelp.handle()
async def rdhelphandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[5:].strip()
await matcher.finish(help_message(args))
await matcher.finish(helps.get(args, help_main))


@stcommand.handle()
Expand All @@ -64,80 +76,141 @@ async def stcommandhandler(matcher: Matcher):
@encommand.handle()
async def enhandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[3:].strip()
await matcher.finish(en(args))
await matcher.finish(en(int(args)))


@drawcommand.handle()
async def drawhandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[5:].strip().split(" ")
key = args.pop(0) if args else ""
cnt = int(args[0]) if args and args[0].isdigit() else None
await matcher.finish(draw(key, cnt))


@racommand.handle()
async def rahandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[3:].strip().lower().split(" ")
level, uid = get_level_uid(event)
name = args.pop(0) if args else ""
exp = int(args[0]) if args and args[0].isdigit() else -1
await matcher.finish(coc_cards.ra_handler(name, exp, level, uid))


@rhcommand.handle()
async def rhcommandhandler(bot: Bot, event: GroupMessageEvent):
args = str(event.get_message())[3:].strip()
uid = event.get_user_id()
if args and not("." in args):
if args and "." not in args:
print("get here")
_args = args.lower().split(" ")
pat = _args.pop(0)
anum = int(_args[0]) if _args else None
if isinstance(bot, V12Bot):
from nonebot.adapters.onebot.v12 import MessageSegment
await bot.send_message(detail_type="private", user_id=uid, message=[MessageSegment.text(rd0(args))])
from nonebot.adapters.onebot.v12 import MessageSegment

await bot.send_message(
detail_type="private",
user_id=uid,
message=[MessageSegment.text(rd0(pat, anum))],
)
elif isinstance(bot, V11Bot):
await bot.send_private_msg(user_id=uid, message=rd0(args))
await bot.send_private_msg(user_id=uid, message=rd0(pat, anum))


@rdcommand.handle()
async def rdcommandhandler(event: MessageEvent):
async def rdcommandhandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[2:].strip()
if args and not("." in args):
await rdcommand.finish(rd0(args))
if args and "." not in args:
_args = args.lower().split(" ")
pat = _args.pop(0)
anum = int(_args[0]) if _args else None
await matcher.finish(rd0(pat, anum))


@dndcommand.handle()
async def dndhandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[4:].strip()
val = int(args) if args else 1
await matcher.finish(dnd(val))


@coccommand.handle()
async def cochandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[4:].strip()
args = str(event.get_message())[4:].strip().split(" ")
pat = args.pop(0)
val = int(args[0]) if args else 1
if pat.endswith("d"):
if pat.startswith("6"):
await matcher.finish(coc6d())
else:
await matcher.finish(coc7d())
elif pat.startswith("6"):
await matcher.finish(coc6(val))
else:
await matcher.finish(coc7(val))


@wcoccommand.handle()
async def wcochandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[5:].strip()
try:
args = int(args)
except ValueError:
args = 20
inv = Investigator()
await matcher.send(inv.age_change(args))
if 15 <= args < 90:
cache_cards.update(event, inv.__dict__, save=False)
level, uid = get_level_uid(event)
coc_cards.cache_update(inv.dump(), level, uid)
await matcher.finish(inv.output())


@ticommand.handle()
async def ticommandhandler(matcher: Matcher,):
await matcher.finish(ti())
async def ticommandhandler(
matcher: Matcher,
):
await matcher.finish(temp_insane())


@licommand.handle()
async def licommandhandler(matcher: Matcher,):
await matcher.finish(li())
async def licommandhandler(
matcher: Matcher,
):
await matcher.finish(long_insane())


@sccommand.handle()
async def schandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[3:].strip().lower()
await matcher.finish(sc(args, event=event))
args = str(event.get_message())[3:].strip().lower().split(" ")
level, uid = get_level_uid(event)
sf = args.pop(0)
san = int(args[0]) if args else None
await matcher.finish(coc_cards.sc_handler(sf, san, level, uid))


@setcommand.handle()
async def sethandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[4:].strip().lower()
await matcher.finish(set_handler(event, args))
args = str(event.get_message())[4:].strip().lower().split(" ")
level, uid = get_level_uid(event)
name = args.pop(0) if args else None
num = args[0] if args and args[0].isdigit() else None
await matcher.finish(coc_cards.set_handler(name, num, level, uid))


@showcommand.handle()
async def showhandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[5:].strip().lower()
for msg in show_handler(event, args):
await matcher.send(msg)


@sacommand.handle()
async def sahandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[3:].strip().lower()
await matcher.finish(sa_handler(event, args))
level, uid = get_level_uid(event)
if args.startswith("s"):
await matcher.finish(coc_cards.show_skill_handler(level, uid))
else:
for msg in coc_cards.show_handler(level, uid):
await matcher.send(msg)


@delcommand.handle()
async def delhandler(matcher: Matcher, event: MessageEvent):
args = str(event.get_message())[4:].strip().lower()
for msg in del_handler(event, args):
args = str(event.get_message())[4:].strip().lower().split(" ")
level, uid = get_level_uid(event)
for msg in coc_cards.del_handler(args, level, uid):
await matcher.send(msg)
Loading