Documentation, code-formatting, and refactoring.

* Added some docs, updated existing docs.
* Use single-quotes for strings when possible.
* Color.pexception() prints exception and stack trace.

wifite/attack/wps.py

@@ -4,8 +4,6 @@
 from ..model.attack import Attack
 from ..util.color import Color
 from ..config import Configuration
-from ..tools.bully import Bully
-from ..tools.reaver import Reaver
 
 class AttackWPS(Attack):
     def __init__(self, target):
@@ -22,26 +20,36 @@ class AttackWPS(Attack):
             self.success = False
             return self.success
 
-        ###################
-        # Pixie-Dust attack
         if Configuration.use_bully:
-            # Bully: Pixie-dust
-            bully = Bully(self.target)
-            bully.run()
-            bully.stop()
-            self.crack_result = bully.crack_result
-            self.success = self.crack_result is not None
-            return self.success
+            return self.run_bully()
         else:
-            reaver = Reaver(self.target)
-            if reaver.is_pixiedust_supported():
-                # Reaver: Pixie-dust
-                reaver = Reaver(self.target)
-                reaver.run()
-                self.crack_result = reaver.crack_result
-                self.success = self.crack_result is not None
-                return self.success
-            else:
-                Color.pl("{!} {R}your version of 'reaver' does not support the {O}WPS pixie-dust attack{W}")
+            return self.run_reaver()
 
         return False
+
+
+    def run_bully(self):
+        # Bully: Pixie-dust
+        from ..tools.bully import Bully
+        bully = Bully(self.target)
+        bully.run()
+        bully.stop()
+        self.crack_result = bully.crack_result
+        self.success = self.crack_result is not None
+        return self.success
+
+
+    def run_reaver(self):
+        from ..tools.reaver import Reaver
+        reaver = Reaver(self.target)
+        if not reaver.is_pixiedust_supported():
+            Color.pl("{!} {R}your version of 'reaver' does not support the {O}WPS pixie-dust attack{W}")
+            return False
+        else:
+            # Reaver: Pixie-dust
+            reaver = Reaver(self.target)
+            reaver.run()
+            self.crack_result = reaver.crack_result
+            self.success = self.crack_result is not None
+            return self.success
+