From d2fbca5bc0cfcc13d9196b2c08de4b589e996922 Mon Sep 17 00:00:00 2001 From: rclarke0 Date: Tue, 27 Feb 2024 16:13:27 -0500 Subject: [PATCH] lint round 3 --- hardware-testing/hardware_testing/abr_tools/__init__.py | 1 + .../hardware_testing/abr_tools/abr_run_logs.py | 9 +++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/hardware-testing/hardware_testing/abr_tools/__init__.py b/hardware-testing/hardware_testing/abr_tools/__init__.py index e69de29bb2d..419248402b5 100644 --- a/hardware-testing/hardware_testing/abr_tools/__init__.py +++ b/hardware-testing/hardware_testing/abr_tools/__init__.py @@ -0,0 +1 @@ +"""Download run logs from ABR robots.""" diff --git a/hardware-testing/hardware_testing/abr_tools/abr_run_logs.py b/hardware-testing/hardware_testing/abr_tools/abr_run_logs.py index a7d1bf8536e..f97f2d03fc6 100644 --- a/hardware-testing/hardware_testing/abr_tools/abr_run_logs.py +++ b/hardware-testing/hardware_testing/abr_tools/abr_run_logs.py @@ -10,9 +10,7 @@ def get_run_ids_from_storage(storage_directory: str) -> Set[str]: - """Read all files in long term storage directory and read run id. - Add run id to a set. Return run id set. - """ + """Read all files in storage directory, extracts run id, adds to set.""" os.makedirs(storage_directory, exist_ok=True) list_of_files = os.listdir(storage_directory) run_ids = set() @@ -109,7 +107,9 @@ def save_runs(runs_to_save: Set[str], ip: str, storage_directory: str) -> None: def get_all_run_logs(storage_directory: str) -> None: - """Connect to each ABR robot to read run log data. + """GET ALL RUN LOGS. + + Connect to each ABR robot to read run log data. Read each robot's list of unique run log IDs and compare them to all IDs in storage. Any ID that is not in storage, download the run log and put it in storage. """ @@ -125,6 +125,7 @@ def get_all_run_logs(storage_directory: str) -> None: if __name__ == "__main__": + """Get run logs.""" parser = argparse.ArgumentParser(description="Pulls run logs from ABR robots.") parser.add_argument( "storage_directory",