blob: 2ab694411afdaa78d4c5af45389c5268eefa0804 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from discord.ext import commands
from typing import Literal
import utils.config as config
class Toggle(commands.Cog):
def __init__(self, bot):
self.bot = bot
@commands.command()
@commands.dm_only()
@commands.is_owner()
async def toggle(
self, ctx, action: Literal["disable", "enable", "broken"]
):
"""Toggle YouTube links"""
if action == "disable":
config.YOUTUBE_SUPPORT = False
config.YOUTUBE_BROKEN = False
return await ctx.send("YouTube has been disabled.")
if action == "enable":
config.YOUTUBE_SUPPORT = True
config.YOUTUBE_BROKEN = False
return await ctx.send("YouTube has been enabled.")
if action == "broken":
config.YOUTUBE_SUPPORT = False
config.YOUTUBE_BROKEN = True
return await ctx.send("YouTube has been marked as broken.")
async def setup(bot):
await bot.add_cog(Toggle(bot))
|
