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

Make Red shutdown when resetting token (#3358)

Browse Source 
* Update __main__.py

* Update __main__.py
This commit is contained in:
jack1142 2020-01-13 02:26:01 +01:00 committed by Michael H
parent 7bdd177713
commit 088360ec51
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
redbot/__main__.py
View File

@ -16,6 +16,7 @@ import sys
from argparse import Namespace
from copy import deepcopy
from pathlib import Path
from typing import NoReturn
import discord
@ -287,7 +288,7 @@ def handle_edit(cli_flags: Namespace):
sys.exit(0)
async def run_bot(red: Red, cli_flags: Namespace):
async def run_bot(red: Red, cli_flags: Namespace) -> NoReturn:
driver_cls = drivers.get_driver_class()
@ -334,6 +335,7 @@ async def run_bot(red: Red, cli_flags: Namespace):
sys.exit(0)
try:
await red.start(token, bot=True, cli_flags=cli_flags)
# This raises SystemExit in normal use at close
except discord.LoginFailure:
log.critical("This token doesn't seem to be valid.")
db_token = await red._config.token()
@ -341,6 +343,8 @@ async def run_bot(red: Red, cli_flags: Namespace):
if confirm("\nDo you want to reset the token?"):
await red._config.token.set("")
print("Token has been reset.")
sys.exit(0)
sys.exit(1)
def handle_early_exit_flags(cli_flags: Namespace):