Fix Import as Module
This commit is contained in:
parent
0ec1a12991
commit
0cbfb56286
@ -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
|
from ssl_checker import SSLChecker
|
||||||
SSLCheckerObject = SSLChecker()
|
|
||||||
|
SSLChecker = SSLChecker()
|
||||||
|
args = {
|
||||||
|
'hosts': ['google.com', 'cisco.com']
|
||||||
|
}
|
||||||
|
|
||||||
|
SSLChecker.show_result(SSLChecker.get_args(json_args=args))
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -313,11 +313,23 @@ class SSLChecker:
|
|||||||
|
|
||||||
return host, port
|
return host, port
|
||||||
|
|
||||||
def get_args(self):
|
def get_args(self, json_args={}):
|
||||||
"""Set argparse options."""
|
"""Set argparse options."""
|
||||||
parser = ArgumentParser(prog='ssl_checker.py', add_help=False,
|
parser = ArgumentParser(prog='ssl_checker.py', add_help=False,
|
||||||
description="""Collects useful information about given host's SSL certificates.""")
|
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 = parser.add_mutually_exclusive_group(required=True)
|
||||||
group.add_argument('-H', '--host', dest='hosts', nargs='*',
|
group.add_argument('-H', '--host', dest='hosts', nargs='*',
|
||||||
required=False, help='Hosts as input separated by space')
|
required=False, help='Hosts as input separated by space')
|
||||||
@ -369,4 +381,4 @@ class SSLChecker:
|
|||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
SSLCheckerObject = SSLChecker()
|
SSLCheckerObject = SSLChecker()
|
||||||
SSLCheckerObject.show_result(SSLCheckerObject.get_args())
|
SSLCheckerObject.show_result(SSLCheckerObject.get_args(json_args={}))
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user