Reformat Python scripts with black

openqa-powermanagement.py

@@ -13,8 +13,8 @@
 
 machine_list_idle = []
 machine_list_offline = []
-machine_list_broken= []
-machine_list_busy= []
+machine_list_broken = []
+machine_list_busy = []
 machines_to_power_on = []
 
 jobs_worker_classes = []
@@ -26,13 +26,13 @@
 openqa_server = "http://localhost"
 
 # Manage cmdline options
-if __name__ == '__main__':
+if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument('--config')
-    parser.add_argument('--dry-run', action='store_true')
-    parser.add_argument('--host')
-    parser.add_argument('--osd', action='store_true')
-    parser.add_argument('--o3', action='store_true')
+    parser.add_argument("--config")
+    parser.add_argument("--dry-run", action="store_true")
+    parser.add_argument("--host")
+    parser.add_argument("--osd", action="store_true")
+    parser.add_argument("--o3", action="store_true")
     args = parser.parse_args()
     if args.config is not None and len(args.config):
         config_file = args.config
@@ -50,37 +50,47 @@
 print("")
 
 # Scheduled/blocked jobs
-scheduled_list_file = requests.get(openqa_server + '/tests/list_scheduled_ajax').content
+scheduled_list_file = requests.get(openqa_server + "/tests/list_scheduled_ajax").content
 scheduled_list_data = json.loads(scheduled_list_file)
-print("Processing " + str(len(scheduled_list_data['data'])) + " job(s) in scheduled/blocked state... (will take about " + str(int(len(scheduled_list_data['data']) * 0.2)) + " seconds)")
-
+print(
+    "Processing "
+    + str(len(scheduled_list_data["data"]))
+    + " job(s) in scheduled/blocked state... (will take about "
+    + str(int(len(scheduled_list_data["data"]) * 0.2))
+    + " seconds)"
+)
+
 # Create list of WORKER_CLASS needed
-for job in scheduled_list_data['data']:
-    response = requests.get(openqa_server + '/api/v1/jobs/' + str(job['id']))
+for job in scheduled_list_data["data"]:
+    response = requests.get(openqa_server + "/api/v1/jobs/" + str(job["id"]))
     job_data = json.loads(response.content)
-    jobs_worker_classes.append(job_data['job']['settings']['WORKER_CLASS'])
+    jobs_worker_classes.append(job_data["job"]["settings"]["WORKER_CLASS"])
 
 jobs_worker_classes = sorted(set(jobs_worker_classes))
-print("Found " + str(len(jobs_worker_classes)) + " different WORKER_CLASS in scheduled jobs: " + str(jobs_worker_classes))
-
+print(
+    "Found "
+    + str(len(jobs_worker_classes))
+    + " different WORKER_CLASS in scheduled jobs: "
+    + str(jobs_worker_classes)
+)
 
 
 # Workers
-workers_list_file = requests.get(openqa_server + '/api/v1/workers').content
+workers_list_file = requests.get(openqa_server + "/api/v1/workers").content
 workers_list_data = json.loads(workers_list_file)
 
 # Create list of hosts which may need to powered up/down
-for worker in workers_list_data['workers']:
-    if worker['status'] in ['idle']:
-        machine_list_idle.append(worker['host'])
-    elif worker['status'] in ['dead']: # Looks like 'dead' means 'offline'
-        machine_list_offline.append(worker['host'])
-    elif worker['status'] in ['running']: # Looks like 'running' means 'working'
-        machine_list_busy.append(worker['host'])
-    elif worker['status'] in ['broken']:
-        machine_list_broken.append(worker['host'])
+for worker in workers_list_data["workers"]:
+    if worker["status"] in ["idle"]:
+        machine_list_idle.append(worker["host"])
+    elif worker["status"] in ["dead"]:  # Looks like 'dead' means 'offline'
+        machine_list_offline.append(worker["host"])
+    elif worker["status"] in ["running"]:  # Looks like 'running' means 'working'
+        machine_list_busy.append(worker["host"])
+    elif worker["status"] in ["broken"]:
+        machine_list_broken.append(worker["host"])
     else:
-        print("Unhandle worker status: " + str(worker['status']))
+        print("Unhandle worker status: " + str(worker["status"]))
 
 # Clean-up the lists
 machine_list_idle = sorted(set(machine_list_idle))
@@ -100,41 +110,63 @@
         machine_list_offline.remove(machine)
 
 # Print an overview
-print(str(len(machine_list_idle)) + " workers listed fully idle: " + str(machine_list_idle))
-print(str(len(machine_list_offline)) + " workers listed offline/dead: " + str(machine_list_offline))
-print(str(len(machine_list_broken)) + " workers listed broken: " + str(machine_list_broken))
+print(
+    str(len(machine_list_idle))
+    + " workers listed fully idle: "
+    + str(machine_list_idle)
+)
+print(
+    str(len(machine_list_offline))
+    + " workers listed offline/dead: "
+    + str(machine_list_offline)
+)
+print(
+    str(len(machine_list_broken))
+    + " workers listed broken: "
+    + str(machine_list_broken)
+)
 print(str(len(machine_list_busy)) + " workers listed busy: " + str(machine_list_busy))
 
 # Get WORKER_CLASS for each workers of each machines (idle and offline) and compare to WORKER_CLASS required by scheduled/blocked jobs
-for worker in workers_list_data['workers']:
-    if worker['host'] in machine_list_offline:
+for worker in workers_list_data["workers"]:
+    if worker["host"] in machine_list_offline:
         for classes in jobs_worker_classes:
-            if set(classes.split(',')).issubset(worker['properties']['WORKER_CLASS'].split(',')):
-                machines_to_power_on.append(worker['host'])
-
-    if worker['host'] in machine_list_idle:
-        if worker['properties']['WORKER_CLASS'] in jobs_worker_classes:
+            if set(classes.split(",")).issubset(
+                worker["properties"]["WORKER_CLASS"].split(",")
+            ):
+                machines_to_power_on.append(worker["host"])
+
+    if worker["host"] in machine_list_idle:
+        if worker["properties"]["WORKER_CLASS"] in jobs_worker_classes:
             # Warning: scheduled (blocked?) job could be run on idle machine!
             print("Warning: scheduled (blocked?) job could be run on idle machine!")
 
 # Power on machines which can run scheduled jobs
 for machine in sorted(set(machines_to_power_on)):
     if machine in machine_list_broken:
-        print("Removing '" + machine + "' from the list to power ON since some workers are broken there")
+        print(
+            "Removing '"
+            + machine
+            + "' from the list to power ON since some workers are broken there"
+        )
     elif args.dry_run:
         print("Would power ON '" + machine + "' - Dry run mode")
-    elif 'power_management' in config and config['power_management'].get(machine + "_POWER_ON"):
+    elif "power_management" in config and config["power_management"].get(
+        machine + "_POWER_ON"
+    ):
         print("Powering ON: " + machine)
-        subprocess.call(config['power_management'][machine + "_POWER_ON"])
+        subprocess.call(config["power_management"][machine + "_POWER_ON"])
     else:
         print("Unable to power ON '" + machine + "' - No command for that")
 
 # Power off machines which are idle or broken (TODO: add a threshold, e.g. idle since more than 15 minutes. Does API provide this information?)
 for machine in machine_list_idle + machine_list_broken:
     if args.dry_run:
         print("Would power OFF '" + machine + "' - Dry run mode")
-    elif 'power_management' in config and config['power_management'].get(machine + "_POWER_OFF"):
+    elif "power_management" in config and config["power_management"].get(
+        machine + "_POWER_OFF"
+    ):
         print("Powering OFF: " + machine)
-        subprocess.call(config['power_management'][machine + "_POWER_OFF"])
+        subprocess.call(config["power_management"][machine + "_POWER_OFF"])
     else:
         print("Unable to power OFF '" + machine + "' - No command for that")

tests/test_trigger_bisect_jobs.py

@@ -204,7 +204,13 @@ def test_problems():
     openqa.log.info = MagicMock()
     openqa.main(args)
     openqa.log.info.assert_called_with("Job 101 (foo) is passed, skipping bisection")
-    assert call('http://openqa.opensuse.org/tests/101/investigation_ajax', request_type='json') not in openqa.fetch_url.mock_calls
+    assert (
+        call(
+            "http://openqa.opensuse.org/tests/101/investigation_ajax",
+            request_type="json",
+        )
+        not in openqa.fetch_url.mock_calls
+    )
     openqa.openqa_clone.assert_not_called()