Fix Module Import

2020-12-09 18:05:54 +03:30 · 2020-12-09 18:05:54 +03:30 · 94c9a42b72
commit 94c9a42b72
parent bf262c224c
1 changed files with 4 additions and 0 deletions
--- a/ssl_checker.py
+++ b/ssl_checker.py
@ -274,6 +274,10 @@ class SSLChecker:
        if user_args.html_true:
            self.export_html(context)

+        # While using the script as a module
+        if __name__ != '__main__':
+            return json.dumps(context)
+
        # Enable JSON output if -j/--json argument specified
        if user_args.json_true:
            print(json.dumps(context))