diff --git a/docs/examples/timer_management/create_timer.py b/docs/examples/timer_management/create_timer.py index f988b08ab..3dfdbcdc3 100644 --- a/docs/examples/timer_management/create_timer.py +++ b/docs/examples/timer_management/create_timer.py @@ -1,14 +1,10 @@ -#!/usr/bin/env python - import argparse import datetime import globus_sdk -from globus_sdk.experimental.globus_app import UserApp # Tutorial Client ID - NATIVE_CLIENT_ID = "61338d24-54d5-408f-a10d-66c06b59f6d2" -USER_APP = UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) def main(): @@ -31,8 +27,12 @@ def main(): ) args = parser.parse_args() - client = globus_sdk.TimersClient(app=USER_APP) + with globus_sdk.UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) as app: + with globus_sdk.TimersClient(app=app) as client: + create_timer(client, args) + +def create_timer(client: globus_sdk.TimersClient, args: argparse.Namespace) -> None: body = globus_sdk.TransferData( source_endpoint=args.SOURCE_COLLECTION, destination_endpoint=args.DESTINATION_COLLECTION, diff --git a/docs/examples/timer_management/create_timer_data_access.py b/docs/examples/timer_management/create_timer_data_access.py index 56fd03dc3..bc4f2c36f 100644 --- a/docs/examples/timer_management/create_timer_data_access.py +++ b/docs/examples/timer_management/create_timer_data_access.py @@ -1,14 +1,10 @@ -#!/usr/bin/env python - import argparse import datetime import globus_sdk -from globus_sdk.experimental.globus_app import UserApp # Tutorial Client ID - NATIVE_CLIENT_ID = "61338d24-54d5-408f-a10d-66c06b59f6d2" -USER_APP = UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) def uses_data_access(transfer_client, collection_id): @@ -40,9 +36,19 @@ def main(): ) args = parser.parse_args() - timers_client = globus_sdk.TimersClient(app=USER_APP) - transfer_client = globus_sdk.TransferClient(app=USER_APP) + with globus_sdk.UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) as app: + with ( + globus_sdk.TimersClient(app=app) as timers_client, + globus_sdk.TransferClient(app=app) as transfer_client, + ): + create_timer(timers_client, transfer_client, args) + +def create_timer( + timers_client: globus_sdk.TimersClient, + transfer_client: globus_sdk.TransferClient, + args: argparse.Namespace, +) -> None: # check if the source or destination use 'data_access' scopes # if so, register these requirements with the app if uses_data_access(transfer_client, args.SOURCE_COLLECTION): diff --git a/docs/examples/timer_management/delete_timer.py b/docs/examples/timer_management/delete_timer.py index 560acd831..e659b84eb 100644 --- a/docs/examples/timer_management/delete_timer.py +++ b/docs/examples/timer_management/delete_timer.py @@ -1,13 +1,9 @@ -#!/usr/bin/env python - import argparse import globus_sdk -from globus_sdk.experimental.globus_app import UserApp # Tutorial Client ID - NATIVE_CLIENT_ID = "61338d24-54d5-408f-a10d-66c06b59f6d2" -USER_APP = UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) def main(): @@ -15,10 +11,10 @@ def main(): parser.add_argument("TIMER_ID") args = parser.parse_args() - client = globus_sdk.TimersClient(app=USER_APP) - - client.delete_job(args.TIMER_ID) - print("Finished deleting timer.") + with globus_sdk.UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) as app: + with globus_sdk.TimersClient(app=app) as client: + client.delete_job(args.TIMER_ID) + print("Finished deleting timer.") if __name__ == "__main__": diff --git a/docs/examples/timer_management/list_timers.py b/docs/examples/timer_management/list_timers.py index a100e0a8f..f9d75e5dc 100644 --- a/docs/examples/timer_management/list_timers.py +++ b/docs/examples/timer_management/list_timers.py @@ -1,16 +1,16 @@ -#!/usr/bin/env python - import globus_sdk -from globus_sdk.experimental.globus_app import UserApp # Tutorial Client ID - NATIVE_CLIENT_ID = "61338d24-54d5-408f-a10d-66c06b59f6d2" -USER_APP = UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) def main(): - client = globus_sdk.TimersClient(app=USER_APP) + with globus_sdk.UserApp("manage-timers-example", client_id=NATIVE_CLIENT_ID) as app: + with globus_sdk.TimersClient(app=app) as client: + list_timers(client) + +def list_timers(client: globus_sdk.TimersClient) -> None: first = True for record in client.list_jobs(query_params={"filter_active": True})["jobs"]: if not first: