From e2399b4f21974cc6b77045d42def9d06e10f1d73 Mon Sep 17 00:00:00 2001 From: Jacob Henry Date: Fri, 14 Sep 2018 01:43:36 -0400 Subject: [PATCH] Added towel roll counting --- main.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/main.py b/main.py index 7e743fb..8b0ae68 100644 --- a/main.py +++ b/main.py @@ -6,6 +6,7 @@ import job_nagger import management_commands import scroll_util import slack_util +import slavestothemachine from dummy import FakeClient # Read api token from file @@ -41,6 +42,10 @@ def main(): # Add kill switch wrap.add_hook(management_commands.reboot_hook) + # Add towel rolling + wrap.add_hook(slavestothemachine.count_work_hook) + wrap.add_hook(slavestothemachine.dump_work_hook) + # Add help help_callback = management_commands.list_hooks_callback_gen(wrap.hooks) wrap.add_hook(slack_util.Hook(help_callback, pattern=management_commands.bot_help_pattern))