Blackfur/SheepBotFrame
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Improved Argument Parser to stop the bot instantly

Browse source
This commit is contained in:
BuildTools 2021-01-13 11:42:06 +01:00
parent c1a364445a
commit b758ee7c7d
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
run.py
View file

@ -57,21 +57,25 @@ async def train():
async def on_ready(): async def on_ready():
print("Logged in!") print("Logged in!")
args = sys.argv args = sys.argv
args.pop(0)
for arg in args: for arg in args:
if arg == "--help" or "-?": if arg == "--help" or arg == "-?":
printhelp() printhelp()
await bot.close() await bot.close()
if arg == "--version" or "-v": raise SystemExit(0)
if arg == "--version" or arg == "-v":
printversion() printversion()
await bot.close() await bot.close()
if arg == "--download" or "-d": raise SystemExit(0)
if arg == "--download" or arg == "-d":
tasks[0] = True tasks[0] = True
if arg == "--train" or "-t": if arg == "--train" or arg == "-t":
tasks[1] = True tasks[1] = True
else: else:
print(f"Unexspected {arg}") print(f"Unexspected {arg}")
printhelp() printhelp()
await bot.close() await bot.close()
raise SystemExit(1)
if tasks[0]: if tasks[0]:
await download() await download()
if tasks[1]: if tasks[1]: