CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-22 02:37:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

[V3 Core] Add optional async setup function (#1314)

Browse Source
This commit is contained in:
Will
2018-02-18 21:12:58 -05:00
committed by GitHub
parent 86b5932c8f
commit b871241eac
3 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
redbot/core/bot.py
View File

@@ -172,7 +172,7 @@ class RedBase(BotBase, RpcMethodMixin):
while pkg_name in curr_pkgs:
curr_pkgs.remove(pkg_name)
def load_extension(self, spec: ModuleSpec):
async def load_extension(self, spec: ModuleSpec):
name = spec.name.split('.')[-1]
if name in self.extensions:
return
@@ -182,7 +182,11 @@ class RedBase(BotBase, RpcMethodMixin):
del lib
raise discord.ClientException('extension does not have a setup function')
lib.setup(self)
if asyncio.iscoroutinefunction(lib.setup):
await lib.setup(self)
else:
lib.setup(self)
self.extensions[name] = lib
def unload_extension(self, name):