Skip to content

Add decorator interface for slash commands #4

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

Merged
merged 4 commits into from
Aug 29, 2021
Merged

Add decorator interface for slash commands #4

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
01b9eff
Add decorator interface for slash commands
Dorukyum Aug 29, 2021
1a03b72
Fix circular import
Dorukyum Aug 29, 2021
fe50978
Fix function typehint
Dorukyum Aug 29, 2021
edf87eb
Use `typing.Callable` instead of `callable`
Dorukyum Aug 29, 2021
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
24 changes: 15 additions & 9 deletions discord/bot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,9 @@

import asyncio
import inspect
from typing import Callable

from .client import *
from .client import Client


class ApplicationCommand:
Expand All @@ -52,11 +53,7 @@ def __init__(self, func, *args, **kwargs):
self.callback = func

def to_dict(self):
_data = {
"name": self.name,
"description": self.description
}
return _data
return {"name": self.name, "description": self.description}


class ApplicationMixin:
Expand All @@ -69,16 +66,25 @@ def add_application_command(self, command):
async def register_application_commands(self):
if len(self.application_commands) == 0:
return
dbg = await self.http.bulk_upsert_global_commands(self.user.id, [i.to_dict for i in self.application_commands])
print(dbg)
await self.http.bulk_upsert_global_commands(
self.user.id, [i.to_dict() for i in self.application_commands]
)


class BotBase(ApplicationMixin):
pass


class Bot(BotBase, Client):
pass
def interaction_command(self, **kwargs):
def wrap(func: Callable) -> ApplicationCommand:
command = ApplicationCommand(func, **kwargs)
self.add_application_command(command)
return command

return wrap

command = slash = interaction_command


class AutoShardedBot(BotBase, Client):
Expand Down