ticker/ticker.py

import discord
from datetime import datetime
from gtts import gTTS
from forex_python.converter import CurrencyRates
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.cron import CronTrigger
from random import randint
import asyncio
import hashlib
import os.path

ROOT = "/home/bot/ticker"
SEGUIS = 665399296522321942
GOONS = 499822662281854996
MATT = 150791815723876352
DAD = 509547108907614208
SMOKEYS = 673203690672357422
RETARD = 695090043609415740

client = discord.Client()
c = CurrencyRates()

ticker_queue = asyncio.Queue()

# returns a file path from the cache
def to_tts(message, lang = 'ja'):
    # the file we save to is based on the hash of the message
    message_hash = hashlib.md5(message.encode('utf-8')).hexdigest()
    file_name = ROOT + "/cache/" + message_hash + ".mp3"

    # if the hash matches and existing file, well...
    if not os.path.isfile(file_name):
        print("file does not exist, cacheing")
        tts = gTTS(message, lang = lang)
        tts.save(file_name)

    return file_name

# coroutine responsible for handling the queue as events come in
async def ticker_update():
    while True:
        await client.wait_until_ready()

        # blocks until there is a message in the queue
        announce_file, target_voice_channel = await ticker_queue.get()
        print("ticker event has been removed from queue")
 
        # there are three scenarios here
        # 1. we are not connected to a voice channel. we should CONNECT to the target voice channel
        # 2. we are connected to the wrong channel. we should MOVE to the target channel
        # 3. we are already in the correct channel. we should do nothing

        guild = target_voice_channel.guild # the guild this ticker event is for
        voice_client = guild.voice_client # the voice client for this guild, may be None

        # there is no connection, nor an existing voice client
        if not voice_client: 
            voice_client = await target_voice_channel.connect()

        # there is a voice client, but there is no connection
        if not voice_client.is_connected():
            voice_client = await target_voice_channel.connect()

        # we are not already in the correct channel
        if not target_voice_channel.id == voice_client.channel.id: 
            await voice_client.move_to(target_voice_channel)

        # load and play supplied message
        source = discord.FFmpegOpusAudio(announce_file)
        voice_client.play(source)

        # block while the message is playing
        while voice_client.is_playing():
             await asyncio.sleep(0.1)

        # once all ticker events have completed, dc
        if ticker_queue.empty():
             await voice_client.disconnect()

@client.event
async def on_ready():
    print(client.user.name)

# add afk message
@client.event
async def on_voice_state_update(member, before, after):
    if member.id == MATT and after.channel and after.channel.id == SMOKEYS:
        await asyncio.sleep(randint(3, 30))
        await member.move_to(client.get_channel(RETARD))

    # only run on moves
    if before.channel and after.channel:
        # moves to the afk channel
        if member.guild.afk_channel and after.channel.id == member.guild.afk_channel.id:
            file_name = to_tts(member.display_name + "はretard dimensionに移動しました")
            await add_event(file_name, before.channel)

# ban league
@client.event
async def on_member_update(before, after):
    # literally any of these fields might not exist and python has no safe navigation
    # nor does it have pattern matching usable for this purpose
    # guess i'll die
    before_name = None
    if before and before.activity and before.activity.type == discord.ActivityType.playing and before.activity.name:
        before_name = before.activity.name

    after_name = None
    if after and after.activity and after.activity.type == discord.ActivityType.playing and after.activity.name:
        after_name = after.activity.name

    if before_name != after_name:
        if after_name == "League of Legends" and after.voice:
            file_name = to_tts("Stop playing League, " + after.display_name)
            await add_event(file_name, after.voice.channel)

client.loop.create_task(ticker_update()) 

# either matts channel, or the most active channel
def get_best_channel(guild_id):
    guild = client.get_guild(guild_id)
    matt = discord.utils.find(lambda m: m.id == MATT, guild.members)

    # why the fuck does python still not have safe navigation operators?
    matt_voice = ((matt.voice.channel if matt.voice.channel else None) if matt.voice else None) if matt else None
    top_channel = sorted(guild.voice_channels, key = lambda chan: len(chan.members), reverse = True)[0]
    return matt_voice if matt_voice else top_channel

async def add_event(file_name, voice_channel):
    print("ticker event has been added to queue")
    await ticker_queue.put((file_name, voice_channel))

async def ticker():
    voice_channel = get_best_channel(SEGUIS)
    file_name = to_tts("The current JPY exchange rate is " + str(c.get_rate('USD', 'JPY')) + " yen to a dollar")
    await add_event(file_name, voice_channel)

async def flatten():
    voice_channel = get_best_channel(SEGUIS)
    file_name = to_tts("Matt, it is time for your 4 PM dick flattening")
    await add_event(file_name, voice_channel)

async def cummies():
    voice_channel = get_best_channel(GOONS)
    file_name = to_tts("くむづりっぽぷっししりっと")
    await add_event(file_name, voice_channel)

s = AsyncIOScheduler(event_loop = client.loop)
s.add_job(ticker, CronTrigger(hour = '0-19,21-23'))
s.add_job(flatten, CronTrigger(hour = '20'))
s.add_job(cummies, CronTrigger(hour = '*', minute = '30'))
s.start()

with open(ROOT + "/priv/token") as file:
    token = file.readline()
    client.run(token)