aboutsummaryrefslogtreecommitdiff
path: root/crosperf/compare_machines.py
diff options
context:
space:
mode:
authorGeorge Burgess IV <gbiv@google.com>2022-09-02 16:59:27 -0700
committerChromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com>2022-09-07 21:15:07 +0000
commit74bd380a27f4f0e8e90ff2dc1cef0b502d74961b (patch)
treebe028f89ec1e2eca735bb4aa1610530147a53625 /crosperf/compare_machines.py
parent8448c60a6a2337ec993923837e1d55b41f49dabc (diff)
downloadtoolchain-utils-74bd380a27f4f0e8e90ff2dc1cef0b502d74961b.tar.gz
Autoformat all Python code
This autoformats all Python code with our new Python formatter, `black`. BUG=b:244644217 TEST=None Change-Id: I15ee49233d98fb6295c0c53c129bbf8e78e0d9ff Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/third_party/toolchain-utils/+/3877337 Tested-by: George Burgess <gbiv@chromium.org> Reviewed-by: Jordan Abrahams-Whitehead <ajordanr@google.com> Commit-Queue: George Burgess <gbiv@chromium.org>
Diffstat (limited to 'crosperf/compare_machines.py')
-rw-r--r--crosperf/compare_machines.py99
1 files changed, 52 insertions, 47 deletions
diff --git a/crosperf/compare_machines.py b/crosperf/compare_machines.py
index c25fd5ab..338c039b 100644
--- a/crosperf/compare_machines.py
+++ b/crosperf/compare_machines.py
@@ -7,61 +7,66 @@
from __future__ import print_function
+import argparse
import os.path
import sys
-import argparse
from machine_manager import CrosMachine
def PrintUsage(msg):
- print(msg)
- print('Usage: ')
- print('\n compare_machines.py --chromeos_root=/path/to/chroot/ '
- 'machine1 machine2 ...')
+ print(msg)
+ print("Usage: ")
+ print(
+ "\n compare_machines.py --chromeos_root=/path/to/chroot/ "
+ "machine1 machine2 ..."
+ )
def Main(argv):
- parser = argparse.ArgumentParser()
- parser.add_argument(
- '--chromeos_root',
- default='/path/to/chromeos',
- dest='chromeos_root',
- help='ChromeOS root checkout directory')
- parser.add_argument('remotes', nargs=argparse.REMAINDER)
-
- options = parser.parse_args(argv)
-
- machine_list = options.remotes
- if len(machine_list) < 2:
- PrintUsage('ERROR: Must specify at least two machines.')
- return 1
- elif not os.path.exists(options.chromeos_root):
- PrintUsage('Error: chromeos_root does not exist %s' % options.chromeos_root)
- return 1
-
- chroot = options.chromeos_root
- cros_machines = []
- test_machine_checksum = None
- for m in machine_list:
- cm = CrosMachine(m, chroot, 'average')
- cros_machines = cros_machines + [cm]
- test_machine_checksum = cm.machine_checksum
-
- ret = 0
- for cm in cros_machines:
- print('checksum for %s : %s' % (cm.name, cm.machine_checksum))
- if cm.machine_checksum != test_machine_checksum:
- ret = 1
- print('Machine checksums do not all match')
-
- if ret == 0:
- print('Machines all match.')
-
- return ret
-
-
-if __name__ == '__main__':
- retval = Main(sys.argv[1:])
- sys.exit(retval)
+ parser = argparse.ArgumentParser()
+ parser.add_argument(
+ "--chromeos_root",
+ default="/path/to/chromeos",
+ dest="chromeos_root",
+ help="ChromeOS root checkout directory",
+ )
+ parser.add_argument("remotes", nargs=argparse.REMAINDER)
+
+ options = parser.parse_args(argv)
+
+ machine_list = options.remotes
+ if len(machine_list) < 2:
+ PrintUsage("ERROR: Must specify at least two machines.")
+ return 1
+ elif not os.path.exists(options.chromeos_root):
+ PrintUsage(
+ "Error: chromeos_root does not exist %s" % options.chromeos_root
+ )
+ return 1
+
+ chroot = options.chromeos_root
+ cros_machines = []
+ test_machine_checksum = None
+ for m in machine_list:
+ cm = CrosMachine(m, chroot, "average")
+ cros_machines = cros_machines + [cm]
+ test_machine_checksum = cm.machine_checksum
+
+ ret = 0
+ for cm in cros_machines:
+ print("checksum for %s : %s" % (cm.name, cm.machine_checksum))
+ if cm.machine_checksum != test_machine_checksum:
+ ret = 1
+ print("Machine checksums do not all match")
+
+ if ret == 0:
+ print("Machines all match.")
+
+ return ret
+
+
+if __name__ == "__main__":
+ retval = Main(sys.argv[1:])
+ sys.exit(retval)