Adding mute() decorator that suppress functions from printing to sys.stdout.

routersploit/__init__.py

@@ -9,6 +9,7 @@ from routersploit.utils import (
     random_text,
     http_request,
     boolify,
+    mute,
 )
 
 from routersploit import exploits

routersploit/modules/exploits/asmax/ar_1004g_password_disclosure.py

@@ -8,6 +8,7 @@ from routersploit import (
     print_success,
     print_table,
     http_request,
+    mute,
 )
 
 
@@ -62,6 +63,7 @@ class Exploit(exploits.Exploit):
         else:
             print_error("Credentials could not be found")
 
+    @mute
     def check(self):
         url = sanitize_url("{}:{}/password.cgi".format(self.target, self.port))
 

routersploit/utils.py

@@ -87,6 +87,23 @@ def stop_after(space_number):
     return _outer_wrapper
 
 
+class DummyFile(object):
+    """  Mocking file object. Optimalization for the "mute" decorator. """
+    def write(self, x): pass
+
+
+def mute(fn):
+    """ Suppress function from printing to sys.stdout """
+    @wraps(fn)
+    def wrapper(self, *args, **kwargs):
+        sys.stdout = DummyFile()
+        try:
+            return fn(self, *args, **kwargs)
+        finally:
+            sys.stdout = sys.__stdout__
+    return wrapper
+
+
 def __cprint(*args, **kwargs):
     """ Color print()