From 0cbfb5628696886d75a3a16d3de8e71f81f55968 Mon Sep 17 00:00:00 2001
From: Narbeh Arakil <narbeh@protonmail.com>
Date: Sat, 19 Sep 2020 18:05:10 +0430
Subject: [PATCH] Fix Import as Module

---
 README.md      |  8 +++++++-
 ssl_checker.py | 16 ++++++++++++++--
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index fc0394c..89afe36 100644
--- a/README.md
+++ b/README.md
@@ -271,7 +271,13 @@ Simply import the `ssl_checker.py` into your python script and use it as a modul
 
 ```
 from ssl_checker import SSLChecker
-SSLCheckerObject = SSLChecker()  
+
+SSLChecker = SSLChecker()
+args = {
+    'hosts': ['google.com', 'cisco.com']
+}
+
+SSLChecker.show_result(SSLChecker.get_args(json_args=args))
 ```
 
 
diff --git a/ssl_checker.py b/ssl_checker.py
index 9e07501..e14aa86 100755
--- a/ssl_checker.py
+++ b/ssl_checker.py
@@ -313,11 +313,23 @@ class SSLChecker:
 
         return host, port
 
-    def get_args(self):
+    def get_args(self, json_args={}):
         """Set argparse options."""
         parser = ArgumentParser(prog='ssl_checker.py', add_help=False,
                                 description="""Collects useful information about given host's SSL certificates.""")
 
+        if len(json_args) > 0:
+            args = parser.parse_args()
+            setattr(args, 'json_true', True)
+            setattr(args, 'verbose', False)
+            setattr(args, 'csv_enabled', False)
+            setattr(args, 'html_true', False)
+            setattr(args, 'json_save_true', False)
+            setattr(args, 'socks', False)
+            setattr(args, 'analyze', False)
+            setattr(args, 'hosts', json_args['hosts'])
+            return args
+
         group = parser.add_mutually_exclusive_group(required=True)
         group.add_argument('-H', '--host', dest='hosts', nargs='*',
                            required=False, help='Hosts as input separated by space')
@@ -369,4 +381,4 @@ class SSLChecker:
 
 if __name__ == '__main__':
     SSLCheckerObject = SSLChecker()
-    SSLCheckerObject.show_result(SSLCheckerObject.get_args())
+    SSLCheckerObject.show_result(SSLCheckerObject.get_args(json_args={}))