h/plugins/pony.py

from util import hook
from ddate.base import DDate

import datetime
import ponyapi

def get_time(ep):
    now = datetime.datetime(2006, 1, 1)
    now = now.now()
    then = now.fromtimestamp(int(ep["air_date"]))
    td = then-now

    return now, then, td

@hook.command
def when(inp, say=None):
    #"Shows the countdown to the new episode of My Little Pony: Friendship is Magic!"

    ep = ponyapi.newest()
    now, then, td = get_time(ep)
    seasonep = ""

    if inp == "discord":
        return "%s will air on %s" % (ep["name"], DDate(then))

    if ep["is_movie"]:
        seasonep = "(a movie)"
    else:
        seasonep = "(season %d episode %d)" % (ep["season"], ep["episode"])

    reply = "%s %s will air on %s in %d days!" % (
                ep["name"], seasonep, then.strftime("%a, %d %b %Y %H:%M:%S"),
                td.days)

    return reply

@hook.command
def randomep(inp):
    #"Shows a random episode of My Little Pony: Friendship is Magic"
    ep = ponyapi.random()

    seasonep = ""

    if ep["is_movie"]:
        seasonep = "(a movie)"
    else:
        seasonep = "(season %d episode %d)" % (ep["season"], ep["episode"])

    return "%s %s" % (ep["name"], seasonep)