gdq.py

import requests
from bs4 import BeautifulSoup
from datetime import datetime, timedelta, date, timezone
import sopel
import re

def GDQdatetime():
    now = datetime.utcnow()
    now = now.replace(tzinfo=timezone.utc)
    try:
        url = "https://gamesdonequick.com"
        req = requests.get(url).content
        bs = BeautifulSoup(req, "html.parser")
        dtext = bs.h5.findNext("p").text
    except:
        if datetime.now().month >= 5:
            nextgdqstartest = "Early January"
        else:
            nextgdqstartest = "Early June"
        delta = None
        return now, delta, nextgdqstartest
    begdtext = re.sub(r' - .*?,',',',dtext)
    fdtext = re.sub(r'(?<=\d)(st|nd|rd|th)','',begdtext)
    try:
        gdqs = (datetime.strptime(fdtext, "%B %d, %Y")).replace(tzinfo=timezone.utc)
        delta = gdqs - now
        nextgdqstartest = fdtext.split(',')[0]
    except:
        nextgdqstartest = fdtext
        delta = None
    return now, delta, nextgdqstartest

def getinfo(run,now):
    schedule = run.find_all('tr',attrs={'class':None})
    game,runner,console,comment,eta,nextgame,nextrunner,nextconsole,nexteta,nextcomment = '','','','','','','','','',''
    for item in schedule:
        group = item.find_all('td')
        try:
            group2 = item.find_next_sibling().find_all('td')
        except:
            nextgame = False
            return (game, runner, console, comment, eta, nextgame, nextrunner, nexteta, nextconsole, nextcomment)
        st = group[0].getText()
        #estfix = timedelta(hours=-5)
        starttime = datetime.strptime(st, '%Y-%m-%dT%H:%M:%SZ' )
        starttime = starttime.replace(tzinfo=timezone.utc)
        #starttime = starttime + estfix
        try:
            offset = datetime.strptime(group2[0].getText().strip(), "%H:%M:%S")
            endtime = starttime + timedelta(hours = offset.hour, minutes = offset.minute, seconds=offset.second)
        except:
            endtime = datetime(2011,1,1,12,00)
        if starttime < now and endtime > now:
            game = group[1].getText()
            runner = group[2].getText()
            #console = group[3].getText()
            comment = group2[1].getText()
            eta = group2[0].getText().strip()
        if starttime > now:
            nextgame = group[1].getText()
            nextrunner = group[2].getText()
            #nextconsole = group[3].getText()
            nexteta = group2[0].getText().strip()
            nextcomment = group2[1].getText()
            break
        else:
            nextgame = 'done'
            nextrunner = 'done'
    return (game, runner, console, comment, eta, nextgame, nextrunner, nexteta, nextconsole, nextcomment)

@sopel.module.commands('gdq','sgdq','agdq')
def gdq(bot, trigger):
    #now = datetime.utcnow()
    #now = now.replace(tzinfo=timezone.utc)
    #delta = datetime(2018,1,7,16,30,tzinfo=timezone.utc) - now
    #textdate = "January 7"
    now, delta, textdate = GDQdatetime()
    url = 'https://gamesdonequick.com/schedule'
    try:
        x = requests.get(url).content
        bs = BeautifulSoup(x)
        run = bs.find("table",{"id":"runTable"}).tbody
        gdqstart = datetime.strptime(run.td.getText(), '%Y-%m-%dT%H:%M:%SZ')
        gdqstart = gdqstart.replace(tzinfo=timezone.utc)
        (game, runner, console, comment, eta, nextgame, nextrunner, nexteta, nextconsole, nextcomment) = getinfo(run,now)
    except:
        if not delta:
            return bot.say("🎮 next GDQ: {}".format(textdate))
        return bot.say("🎮 GDQ is {0} days away ({1})".format(delta.days, textdate))
    if not nextgame:
        if not delta:
            return bot.say("🎮 next GDQ: {}".format(textdate))
        return bot.say("🎮 GDQ is {0} days away ({1})".format(delta.days,textdate))
    if now < gdqstart:
        tts = gdqstart - now
        if tts.days <= 2:
            return bot.say("🎮 GDQ is only {0}hrs {1}mins away. The first game will be: {2} by {3} 🕙 ETA: {4} 💬 comment: {5} | https://gamesdonequick.com/schedule".format(int(tts.total_seconds() // 3600),int((tts.total_seconds() % 3600) // 60), nextgame, nextrunner, nexteta, nextcomment))
        else:
            return bot.say("🎮 GDQ is {0} days away ({1}) | https://gamesdonequick.com/schedule | I will have more details for you once we get close.".format(tts.days,gdqstart.strftime('%d.%m.%Y')))
    if nextgame == 'done':
        return bot.say("🎮 GDQ is {0} days away ({1} [estimated])".format(delta.days,textdate))
    if game:
        if comment:
            bot.say("🎮 current game: {0} by {1} 🕙 ETA: {2} 💬 comment: {3} | 🔜 next game: {4} by {5} | https://www.twitch.tv/gamesdonequick | https://gamesdonequick.com/schedule".format(game, runner, eta, comment, nextgame, nextrunner))
        else:
            bot.say("🎮 current game: {0} by {1} 🕙 ETA: {2} | 🔜 next game: {3} by {4} | https://www.twitch.tv/gamesdonequick | https://gamesdonequick.com/schedule".format(game, runner, eta, nextgame, nextrunner))
    else:
        bot.say("🎮 current game: 🚧 setting things up for the next runners 🚧 | 🔜 next game {0} by {1} | https://www.twitch.tv/gamesdonequick | https://gamesdonequick.com/schedule".format(nextgame, nextrunner))