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
|
#!/usr/bin/env python
import argparse
import os
import globus_sdk
from globus_sdk.tokenstorage import SimpleJSONFileAdapter
MY_FILE_ADAPTER = SimpleJSONFileAdapter(
os.path.expanduser("~/.sdk-manage-projects.json")
)
SCOPES = [globus_sdk.AuthClient.scopes.manage_projects, "openid", "email"]
RESOURCE_SERVER = globus_sdk.AuthClient.resource_server
# tutorial client ID
# we recommend replacing this with your own client for any production use-cases
CLIENT_ID = "61338d24-54d5-408f-a10d-66c06b59f6d2"
NATIVE_CLIENT = globus_sdk.NativeAppAuthClient(CLIENT_ID)
def do_login_flow():
NATIVE_CLIENT.oauth2_start_flow(requested_scopes=SCOPES)
authorize_url = NATIVE_CLIENT.oauth2_get_authorize_url()
print(f"Please go to this URL and login:\n\n{authorize_url}\n")
auth_code = input("Please enter the code here: ").strip()
tokens = NATIVE_CLIENT.oauth2_exchange_code_for_tokens(auth_code)
return tokens.by_resource_server[RESOURCE_SERVER]
def get_auth_client():
tokens = do_login_flow()
return globus_sdk.AuthClient(
authorizer=globus_sdk.AccessTokenAuthorizer(tokens["access_token"])
)
def create_project(args):
auth_client = get_auth_client()
userinfo = auth_client.oauth2_userinfo()
print(
auth_client.create_project(
args.name,
contact_email=userinfo["email"],
admin_ids=userinfo["sub"],
)
)
def list_projects():
auth_client = get_auth_client()
for project in auth_client.get_projects():
print(f"name: {project['display_name']}")
print(f"id: {project['id']}")
print()
def main():
parser = argparse.ArgumentParser()
parser.add_argument("action", choices=["create", "list"])
parser.add_argument("-n", "--name", help="Project name for create")
args = parser.parse_args()
execute(parser, args)
def execute(parser, args):
if args.action == "create":
if args.name is None:
parser.error("create requires --name")
create_project(args)
elif args.action == "list":
list_projects()
else:
raise NotImplementedError()
if __name__ == "__main__":
main()
|
