You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
874 B
28 lines
874 B
4 years ago
|
from typing import TYPE_CHECKING, Union
|
||
|
|
|
||
|
|
import discord
|
||
|
|
from discord.ext.commands import BadArgument, Converter
|
||
|
|
from redbot.core import commands
|
||
|
|
|
||
|
|
from werewolf.player import Player
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
PlayerConverter = Union[int, discord.Member]
|
||
|
|
CronConverter = str
|
||
|
|
else:
|
||
|
|
|
||
|
|
class PlayerConverter(Converter):
|
||
|
|
async def convert(self, ctx, argument) -> Player:
|
||
|
|
|
||
|
|
try:
|
||
|
|
target = await commands.MemberConverter().convert(ctx, argument)
|
||
|
|
except BadArgument:
|
||
|
|
try:
|
||
|
|
target = int(argument)
|
||
|
|
assert target >= 0
|
||
|
|
except (ValueError, AssertionError):
|
||
|
|
raise BadArgument
|
||
|
|
|
||
|
|
# TODO: Get the game for context without making a new one
|
||
|
|
# TODO: Get player from game based on either ID or member object
|
||
|
|
return target
|