mirror of https://github.com/stijndcl/didier
29 lines
714 B
Python
29 lines
714 B
Python
|
from data.constants import prefixes
|
||
|
|
from discord.ext import commands
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
fallback = os.urandom(32).hex()
|
||
|
|
|
||
|
|
|
||
|
|
def get_prefix(bot: commands.Bot, message):
|
||
|
|
content = message.content.lower()
|
||
|
|
mention = "<@!{}>".format(bot.user.id)
|
||
|
|
|
||
|
|
# Used @Didier
|
||
|
|
if content.startswith(mention):
|
||
|
|
if content.startswith(mention + " "):
|
||
|
|
return mention + " "
|
||
|
|
return mention
|
||
|
|
|
||
|
|
# Used a prefix
|
||
|
|
for prefix in prefixes:
|
||
|
|
# Find correct prefix
|
||
|
|
if content.startswith(prefix):
|
||
|
|
# Check if a space has to be added to invoke commands
|
||
|
|
if content.startswith(prefix + " "):
|
||
|
|
return prefix + " "
|
||
|
|
return prefix
|
||
|
|
|
||
|
|
return fallback
|