nemubot/modules/bonneannee.py

# coding=utf-8

"""Wishes Happy New Year when the time comes"""

from datetime import datetime, timezone

from hooks import hook
from tools.countdown import countdown_format

nemubotversion = 3.4

from more import Response

yr = datetime.now(timezone.utc).year
yrn = datetime.now(timezone.utc).year + 1

def load(context):
    d = datetime(yrn, 1, 1, 0, 0, 0, 0, timezone.utc) - datetime.now(timezone.utc)
    add_event(ModuleEvent(interval=0, offset=d.total_seconds(), call=bonneannee))

def bonneannee():
    txt = "Bonne année %d !" % yrn
    print (txt)
    send_response("localhost:2771", Response(txt, "#epitagueule"))
    send_response("localhost:2771", Response(txt, "#yaka"))
    send_response("localhost:2771", Response(txt, "#epita2014"))
    send_response("localhost:2771", Response(txt, "#ykar"))
    send_response("localhost:2771", Response(txt, "#42sh"))
    send_response("localhost:2771", Response(txt, "#nemubot"))

@hook("cmd_hook", "newyear")
@hook("cmd_hook", str(yrn), yrn)
def cmd_newyear(msg, yr):
    return Response(countdown_format(datetime(yr, 1, 1, 0, 0, 1, 0, timezone.utc),
                         "Il reste %s avant la nouvelle année.",
                         "Nous faisons déjà la fête depuis %s !"),
                    channel=msg.channel)

@hook("cmd_rgxp", data=yrn, regexp="^[0-9]{4}$")
def cmd_timetoyear(msg, cur):
    yr = int(msg.cmds[0])

    if yr == cur:
        return None

    return Response(countdown_format(datetime(yr, 1, 1, 0, 0, 1, 0, timezone.utc),
                         "Il reste %s avant %d." % ("%s", yr),
                         "Le premier janvier %d est passé depuis %s !" % (yr, "%s")),
                    channel=msg.channel)