deflax/television
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

refactor announement

Browse source
This commit is contained in:
deflax 2024-01-17 05:27:30 +02:00
parent eacea567ba
commit b3480c8bd2
1 changed files with 27 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/discordbot/discordbot.py
View file

@ -72,28 +72,33 @@ async def update_database():
response = requests.get(db_url) response = requests.get(db_url)
response.raise_for_status() response.raise_for_status()
database = response.json() database = response.json()
else:
logger_discord.error('Cannot connect to the database!')
return
# Search for live streams
if database != {}: if database != {}:
for key, value in database.items(): for key, value in database.items():
stream_name = value['name'] stream_name = value['name']
stream_start_at = value['start_at'] stream_start_at = value['start_at']
if stream_start_at == 'now': if stream_start_at == 'now':
try: # Check if the job already exists
job = scheduler.get_job('announce_live_channel') if scheduler.get_job('announce_live_channel') is None:
if not job: # Job doesn't exist, so add it
logger_discord.info(f'{stream_name} live stream detected!') logger_discord.info(f'{stream_name} live stream detected!')
scheduler.add_job(func=announce_live_channel, trigger='interval', seconds=60, id='announce_live_channel', args=(stream_name,)) scheduler.add_job(func=announce_live_channel, trigger='interval', seconds=60, id='announce_live_channel', args=(stream_name,))
# Exit the loop since we found one
return return
except JobLookupError: else:
pass # Exit the loop since we already have a announcement job
try: return
job = scheduler.get_job('announce_live_channel')
if job: # Cleanup the announce job
if scheduler.get_job('announce_live_channel') is not None:
scheduler.remove_job('announce_live_channel') scheduler.remove_job('announce_live_channel')
live_channel = bot.get_channel(announce_channel_id) #live_channel = bot.get_channel(announce_channel_id)
logger_discord.info(f'{stream_name} finished') logger_discord.info(f'{stream_name} finished')
await live_channel.send(f'{stream_name} finished') #await live_channel.send(f'{stream_name} finished')
except JobLookupError:
pass
async def announce_live_channel(stream_name): async def announce_live_channel(stream_name):
if announce_channel_id == 'disabled': if announce_channel_id == 'disabled':