diff options
Diffstat (limited to 'cli/utils.py')
| -rw-r--r-- | cli/utils.py | 37 |
1 files changed, 20 insertions, 17 deletions
diff --git a/cli/utils.py b/cli/utils.py index 208be68..b267207 100644 --- a/cli/utils.py +++ b/cli/utils.py @@ -7,25 +7,28 @@ import json + def scope_lookup(scope_id, scope_type, d): - if scope_id == d.get('id') and scope_type == d.get('type'): - return d - for scope in d['scope']: - lookup = scope_lookup(scope_id, scope_type, scope) - if lookup is not None: - return lookup - return None + if scope_id == d.get('id') and scope_type == d.get('type'): + return d + for scope in d['scope']: + lookup = scope_lookup(scope_id, scope_type, scope) + if lookup is not None: + return lookup + return None + def ips_in_scope(scope): - if scope is None: - return [] - if 'ip' in scope: - return [scope['ip']] - ips = [] - for child in scope['scope']: - ips += ips_in_scope(child) - return ips + if scope is None: + return [] + if 'ip' in scope: + return [scope['ip']] + ips = [] + for child in scope['scope']: + ips += ips_in_scope(child) + return ips + def print_json(text): - payload = json.loads(text) - print(json.dumps(payload, sort_keys=True, indent=2)) + payload = json.loads(text) + print(json.dumps(payload, sort_keys=True, indent=2)) |