Added towel roll counting

2018-09-14 01:43:36 -04:00
parent b925854cc5
commit e2399b4f21
1 changed files with 5 additions and 0 deletions
@@ -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))