Switch to using @Utils.hook and docstrings for event hooks

1 files changed, 6 insertions, 5 deletions

diff --git a/modules/dns.py b/modules/dns.py
index e5552aa8..19ef7ea0 100644
--- a/modules/dns.py
+++ b/modules/dns.py
@@ -1,13 +1,14 @@
 import socket
+from src import ModuleManager, Utils
 
-class Module(object):
+class Module(ModuleManager.BaseModule):
     _name = "DNS"
-    def __init__(self, bot, events, exports):
-        events.on("received.command.dns").hook(self.dns, min_args=1,
-            help="Get all addresses for a given hostname (IPv4/IPv6)",
-            usage="<hostname>")
 
+    @Utils.hook("received.command.dns", min_args=1, usage="<hostname>")
     def dns(self, event):
+        """
+        Get all addresses for a given hostname (IPv4/IPv6)
+        """
         hostname = event["args_split"][0]
         try:
             address_info = socket.getaddrinfo(hostname, 1, 0,