advance_streams.py

import sys
import json
import discord
from datetime import timezone, datetime, timedelta

import core.schedule as schedule

if "-h" in sys.argv or len(sys.argv) != 5:
    print("""Usage: {} <data sheet.xlsx> <day> <time end> <time start>

    Advance the streams to the next broadcasts to be made live, and make those broadcasts live.
    Will take offline the broadcasts that ended within the [<time end>, <time start>] interval
    and make those starting within the [<time end>, <time start>] interval live

    Options:
        <time end>          Specify the earliest end time of the prev sessions that are live now
                            that should be taken offline. Specify as HHMM or none to indicate no prior
                            sessions, i.e., the sessions being started are the start of the day

        <time start>        Specify the latest start time of the next sessions that should be
                            made live. Specify as HHMM or none to indicate no following sessions,
                            i.e., the sessions being ended are the end of the day.
    """.format(sys.argv[0]))
    sys.exit(0)

database = schedule.Database(sys.argv[1], youtube=True, use_pickled_credentials=True)
# Fill in the computer stream key IDs
database.populate_stream_key_ids()
day = database.get_day(sys.argv[2])

time_end = None
if sys.argv[3] != "none":
    time_end = datetime(schedule.CONFERENCE_YEAR, day.month, day.day,
            hour=int(sys.argv[3][0:2]), minute=int(sys.argv[3][2:4]), tzinfo=schedule.conf_tz)
else:
    time_end = datetime(schedule.CONFERENCE_YEAR, day.month, day.day,
            hour=0, minute=1, tzinfo=schedule.conf_tz)

time_start = None
if sys.argv[4] != "none":
    time_start = datetime(schedule.CONFERENCE_YEAR, day.month, day.day,
            hour=int(sys.argv[4][0:2]), minute=int(sys.argv[4][2:4]), tzinfo=schedule.conf_tz)   
else:
    time_start = datetime(schedule.CONFERENCE_YEAR, day.month, day.day,
            hour=23, minute=0, tzinfo=schedule.conf_tz)   

print("Ending broadcasts whose sessions end in the interval [{}, {}]".format(time_end, time_start))
print("Advancing streams to broadcasts whose sessions start in the interval [{}, {}]".format(time_end, time_start))

end_sessions = []
start_sessions = []
for k, v in day.get_sessions(False).items():
    time = v.session_time()
    if time[1] >= time_end and time[1] <= time_start:
        end_sessions.append(v)
    elif time[0] >= time_end and time[0] <= time_start:
        start_sessions.append(v)

print("=" * 10 + "\nEnding sessions:")
for s in end_sessions:
    print(s)
    s.stop_streaming()
    non_consent_text = "Does not consent to video recording, Will edit out in post"
    if non_consent_text in s.special_notes():
        print("Make video {} private, non-consenting presentation".format(
            s.timeslot_entry(0, "Youtube Broadcast").value))
    print("-----")

print("=" * 10 + "\nStarting sessions:")
for s in start_sessions:
    print(s)
    s.start_streaming()
    print("-----")

# Annoying but have to run bot to post the starting/ending messages
client = discord.Client()
@client.event
async def on_ready():
    embed = schedule.base_discord_embed()
    embed["title"] = "The Session has Ended"

    for s in end_sessions:
        if not s.has_discord_channel():
            continue
        guild_id, channel_id = s.discord_ids()
        guild = [g for g in client.guilds if str(g.id) == guild_id][0]
        channel = [c for c in guild.text_channels if str(c.id) == channel_id]
        if len(channel) > 0:
            session_info = await channel[0].send(embed=discord.Embed.from_dict(embed))

    embed["title"] = "The Session is Starting"
    for s in start_sessions:
        if not s.has_discord_channel():
            continue
        guild_id, channel_id = s.discord_ids()
        guild = [g for g in client.guilds if str(g.id) == guild_id][0]
        channel = [c for c in guild.text_channels if str(c.id) == channel_id]
        if len(channel) > 0:
            session_info = await channel[0].send(embed=discord.Embed.from_dict(embed))

    print("Start/End messages sent, hit Ctrl-C to kill the bot")

client.run(database.auth.discord["bot_token"])