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

refactor bot record annoncement

Browse source
This commit is contained in:
deflax 2024-01-18 02:02:19 +02:00
parent b62411f89d
commit 15ab90c00e
1 changed files with 50 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

87
src/discordbot/discordbot.py
View file

@ -101,51 +101,64 @@ async def query_database():
logger_discord.error('Cannot connect to the database!') logger_discord.error('Cannot connect to the database!')
return return
if database == {}:
logger_discord.error('Database is empty!')
return
# Search for live streams # Search for live streams
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': # Check if the job already exists
# Check if the job already exists if scheduler.get_job('announce_live_channel') is None:
if scheduler.get_job('announce_live_channel') is None: # Job doesn't exist, so add it
# 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', minutes=int(live_channel_update), id='announce_live_channel', args=(stream_name,))
scheduler.add_job(func=announce_live_channel, trigger='interval', minutes=int(live_channel_update), id='announce_live_channel', args=(stream_name,))
# Manually execute the job once immediately
# Manually execute the job once immediately scheduler.get_job('announce_live_channel').modify(next_run_time=datetime.now())
scheduler.get_job('announce_live_channel').modify(next_run_time=datetime.now())
# Set global rechead
# Set global rechead rec_url = f'https://{scheduler_hostname}/rechead'
rec_url = f'https://{scheduler_hostname}/rechead' if requests.get(rec_url).status_code == 200:
if requests.get(rec_url).status_code == 200: response = requests.get(rec_url)
response = requests.get(rec_url) response.raise_for_status()
response.raise_for_status() rechead = response.json()
rechead = response.json()
# Exit since we found one
# Exit since we found one return
return else:
else: # Exit since we already have a announcement job
# Exit since we already have a announcement job return
return
# Cleanup the announce job # Cleanup the announce job
if scheduler.get_job('announce_live_channel') is not None: if scheduler.get_job('announce_live_channel') is not None:
scheduler.remove_job('announce_live_channel') scheduler.remove_job('announce_live_channel')
if live_channel_id != 0:
live_channel = bot.get_channel(int(live_channel_id))
if rechead != {}: if rechead != {}:
video_filename = rechead['video'] video_filename = rechead['video']
thumb_filename = rechead['thumb'] thumb_filename = rechead['thumb']
rechead = {} rechead = {}
vod_url = f'https://{scheduler_hostname}/video/{video_filename}' video_url = f'https://{scheduler_hostname}/video/{video_filename}'
thumb_url = f'https://{scheduler_hostname}/thumb/{thumb_filename}' thumb_url = f'https://{scheduler_hostname}/thumb/{thumb_filename}'
offline_msg = f'Live stream is now offline. [Download VoD recording]({vod_url}) {thumb_url}' offline_msg = f'Live stream is now offline. [Download VoD recording]({vod_url}) {thumb_url}'
else:
offline_msg = f'Live stream is now offline.' # Creating an embed
logger_discord.info(offline_msg) embed = discord.Embed(
if live_channel_id != 0: title='Download Example',
live_channel = bot.get_channel(int(live_channel_id)) description='Click the link below to download the MP4 file.',
await live_channel.send(offline_msg) color=discord.Color.green()
)
embed.set_thumbnail(url=thumb_url)
embed.add_field(name='Download MP4', value='[Download Video](video_url)', inline=False)
embed.set_author(name='DeflaxTV', icon_url=bot.user.avatar_url)
# Sending the embed to the channel
await live_channel.send(embed=embed)
else:
logger_discord.info('Live stream is now offline.')
async def announce_live_channel(stream_name): async def announce_live_channel(stream_name):
logger_discord.info(f'{stream_name} is live!') logger_discord.info(f'{stream_name} is live!')