1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
import click
import shodan
from shodan.cli.helpers import get_api_key, humanize_api_plan
@click.group()
def org():
"""Manage your organization's access to Shodan"""
pass
@org.command()
@click.option('--silent', help="Don't send a notification to the user", default=False, is_flag=True)
@click.argument('user', metavar='<username or email>')
def add(silent, user):
"""Add a new member"""
key = get_api_key()
api = shodan.Shodan(key)
try:
api.org.add_member(user, notify=not silent)
except shodan.APIError as e:
raise click.ClickException(e.value)
click.secho('Successfully added the new member', fg='green')
@org.command()
def info():
"""Show an overview of the organization"""
key = get_api_key()
api = shodan.Shodan(key)
try:
organization = api.org.info()
except shodan.APIError as e:
raise click.ClickException(e.value)
click.secho(organization['name'], fg='cyan')
click.secho('Access Level: ', nl=False, dim=True)
click.secho(humanize_api_plan(organization['upgrade_type']), fg='magenta')
if organization['domains']:
click.secho('Authorized Domains: ', nl=False, dim=True)
click.echo(', '.join(organization['domains']))
click.echo('')
click.secho('Administrators:', dim=True)
for admin in organization['admins']:
click.echo(u' > {:30}\t{:30}'.format(
click.style(admin['username'], fg='yellow'),
admin['email'])
)
click.echo('')
if organization['members']:
click.secho('Members:', dim=True)
for member in organization['members']:
click.echo(u' > {:30}\t{:30}'.format(
click.style(member['username'], fg='yellow'),
member['email'])
)
else:
click.secho('No members yet', dim=True)
@org.command()
@click.argument('user', metavar='<username or email>')
def remove(user):
"""Remove and downgrade a member"""
key = get_api_key()
api = shodan.Shodan(key)
try:
api.org.remove_member(user)
except shodan.APIError as e:
raise click.ClickException(e.value)
click.secho('Successfully removed the member', fg='green')
|
