# coding=utf-8 """Wishes Happy New Year when the time comes""" from datetime import datetime, timezone from nemubot import context from nemubot.event import ModuleEvent from nemubot.hooks import hook from nemubot.tools.countdown import countdown_format nemubotversion = 4.0 from more import Response yr = datetime.now(timezone.utc).year yrn = datetime.now(timezone.utc).year + 1 def load(context): if not context.config or not context.config.hasNode("sayon"): print("You can append in your configuration some balise to " "automaticaly wish an happy new year on some channels like:\n" "") def bonneannee(): txt = "Bonne année %d !" % yrn print(txt) if context.config and context.config.hasNode("sayon"): for sayon in context.config.getNodes("sayon"): if "hostid" not in sayon or "channel" not in sayon: print("Error: missing hostif or channel") continue srv = sayon["hostid"] chan = sayon["channel"] context.send_response(srv, Response(txt, chan)) d = datetime(yrn, 1, 1, 0, 0, 0, 0, timezone.utc) - datetime.now(timezone.utc) context.add_event(ModuleEvent(interval=0, offset=d.total_seconds(), call=bonneannee)) @hook("cmd_hook", "newyear") @hook("cmd_hook", str(yrn)) def cmd_newyear(msg): return Response(countdown_format(datetime(yrn, 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.cmd) 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)