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

query rechead

Browse source
This commit is contained in:
deflax 2024-01-17 22:06:47 +02:00
parent 19271fe016
commit 1dc81ecba4
2 changed files with 22 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/discordbot/discordbot.py
View file

@ -12,6 +12,7 @@ bot_token = os.environ.get('DISCORDBOT_TOKEN', 'token')
live_channel_id = os.environ.get('DISCORDBOT_LIVE_CHANNEL_ID', 0) live_channel_id = os.environ.get('DISCORDBOT_LIVE_CHANNEL_ID', 0)
live_channel_update = os.environ.get('DISCORDBOT_LIVE_CHANNEL_UPDATE', 5) live_channel_update = os.environ.get('DISCORDBOT_LIVE_CHANNEL_UPDATE', 5)
scheduler_hostname = os.environ.get('SCHEDULER_API_HOSTNAME', 'tv.example.com') scheduler_hostname = os.environ.get('SCHEDULER_API_HOSTNAME', 'tv.example.com')
vod_hostname = os.environ.get('VOD_HOSTNAME', 'vod.example.com')
# Discord API Intents # Discord API Intents
intents = discord.Intents.all() intents = discord.Intents.all()
@ -34,6 +35,7 @@ log_level = os.environ.get('SCHEDULER_LOG_LEVEL', 'INFO').upper()
logger_discord.setLevel(log_level) logger_discord.setLevel(log_level)
database = {} database = {}
rechead = {}
# Bot functions # Bot functions
@bot.event @bot.event
@ -90,6 +92,7 @@ async def query_playhead():
async def query_database(): async def query_database():
global database global database
global rechead
db_url = f'https://{scheduler_hostname}/database' db_url = f'https://{scheduler_hostname}/database'
if requests.get(db_url).status_code == 200: if requests.get(db_url).status_code == 200:
response = requests.get(db_url) response = requests.get(db_url)
@ -103,7 +106,7 @@ async def query_database():
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':
# 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:
@ -114,6 +117,13 @@ async def query_database():
# 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
rec_url = f'https://{scheduler_hostname}/rechead'
if requests.get(rec_url).status_code == 200:
response = requests.get(rec_url)
response.raise_for_status()
rechead = response.json()
# Exit since we found one # Exit since we found one
return return
else: else:
@ -123,10 +133,18 @@ async def query_database():
# 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')
logger_discord.info(f'Live stream is offline.')
if rechead != {}:
vod_filename = rechead['file']
rechead = {}
vod_url = f'https://{vod_hostname}/storage/{vod_filename}'
offline_msg = f'Live stream is offline. VOD: {vod_url}'
else:
offline_msg = f'Live stream is offline.'
logger_discord.info(offline_msg)
if live_channel_id != 0: if live_channel_id != 0:
live_channel = bot.get_channel(int(live_channel_id)) live_channel = bot.get_channel(int(live_channel_id))
await live_channel.send('Live stream is offline.') await live_channel.send(offline_msg)
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!')

1
variables.env.dist
View file

@ -9,3 +9,4 @@ SCHEDULER_API_HOSTNAME=tv.example.com
DISCORDBOT_TOKEN=changeme DISCORDBOT_TOKEN=changeme
DISCORDBOT_LIVE_CHANNEL_ID=0 DISCORDBOT_LIVE_CHANNEL_ID=0
DISCORDBOT_LIVE_CHANNEL_UPDATE=5 DISCORDBOT_LIVE_CHANNEL_UPDATE=5
VOD_HOSTNAME=vod.example.com