Make bot.py Matrix backend quit after 5 seconds

This commit is contained in:
Anthony Wang 2022-07-18 13:44:49 -05:00
parent 36baee4925
commit 40bf86aaff
Signed by: a
GPG key ID: BC96B00AEC5F2D76

18
bot.py
View file

@ -1,4 +1,5 @@
from argparse import ArgumentParser
from asyncio import create_task, sleep, run
from random import randint, choice
from re import sub
@ -96,6 +97,9 @@ while True:
post = sub('(@[^ ]*)@[^ ]*', '\\1', post)
if args.yes:
quit()
# Prompt the user
res = input('Post/Retry/New input/Custom input/Quit: ')
if res not in 'prnPRNcC':
@ -134,6 +138,16 @@ for backend, instance, token in zip(args.backend, args.instance, args.token):
@bot.listener.on_startup
async def room_joined(room_id):
await bot.api.send_text_message(room_id=room_id, message=post)
quit()
bot.run()
async def wait_quit():
await sleep(5)
quit()
async def run_bot():
run = create_task(bot.main())
wait = create_task(wait_quit())
await run
await wait
run(run_bot())