royalnet/royalpack/commands/magicktreasure.py

from typing import *
import royalnet
import royalnet.commands as rc
import royalnet.utils as ru
from ..tables import Treasure


class MagicktreasureCommand(rc.Command):
    name: str = "magicktreasure"

    description: str = "Crea un nuovo Treasure di Fiorygi (senza spendere i tuoi)."

    syntax: str = "{codice} {valore}"

    async def _permission_check(self, author, code, value, data, session):
        if "banker" not in author.roles:
            raise rc.UserError("Non hai permessi sufficienti per eseguire questo comando.")
        return author

    async def _create_treasure(self, author, code, value, data, session):
        TreasureT = self.alchemy.get(Treasure)

        treasure = await ru.asyncify(session.query(TreasureT).get, code)
        if treasure is not None:
            raise rc.UserError("Esiste già un Treasure con quel codice.")

        treasure = TreasureT(
            code=code,
            value=value,
            redeemed_by=None
        )

        return treasure

    async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None:
        await data.delete_invoking()

        async with data.session_acm() as session:
            author = await data.find_author(session=session, required=True)

            code = args[0].lower()
            try:
                value = int(args[1])
            except ValueError:
                raise rc.InvalidInputError("Il valore deve essere maggiore o uguale a 0.")
            if value < 0:
                raise rc.InvalidInputError("Il valore deve essere maggiore o uguale a 0.")

            await self._permission_check(author, code, value, data, session)
            treasure = await self._create_treasure(author, code, value, data, session)
            session.add(treasure)
            await ru.asyncify(session.commit)

        await data.reply("✅ Treasure creato!")