From 25dfdf43887d15854ad0be8a9253dffd71853f49 Mon Sep 17 00:00:00 2001 From: Noah Moroze Date: Wed, 4 Dec 2024 10:04:56 +0000 Subject: [PATCH] [manuf] Clean up --cp-only handling in orchestrator Signed-off-by: Noah Moroze --- .../orchestrator/src/orchestrator.py | 21 ++++++++++--------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/sw/host/provisioning/orchestrator/src/orchestrator.py b/sw/host/provisioning/orchestrator/src/orchestrator.py index f36e3c230679d..b7e91c4656519 100644 --- a/sw/host/provisioning/orchestrator/src/orchestrator.py +++ b/sw/host/provisioning/orchestrator/src/orchestrator.py @@ -163,18 +163,19 @@ def main(args_in): fpga=args.fpga, require_confirmation=not args.non_interactive) dut.run_cp() - if not args.cp_only: - dut.run_ft() + if args.cp_only: + logging.info("FT skipped since --cp-only was provided") + return - db_path = Path(args.db_path) - db_handle = db.DB(db.DBConfig(db_path=db_path)) - db.DeviceRecord.create_table(db_handle) + dut.run_ft() - device_record = db.DeviceRecord.from_dut(dut) - device_record.upsert(db_handle) - logging.info(f"Added DeviceRecord to database: {device_record}") - else: - logging.info("FT skipped since --cp-only was provided") + db_path = Path(args.db_path) + db_handle = db.DB(db.DBConfig(db_path=db_path)) + db.DeviceRecord.create_table(db_handle) + + device_record = db.DeviceRecord.from_dut(dut) + device_record.upsert(db_handle) + logging.info(f"Added DeviceRecord to database: {device_record}") if __name__ == "__main__":