From ee77d888b56f2da6fe0ea962a6f0e4355147089d Mon Sep 17 00:00:00 2001 From: Adrian Manteza Date: Tue, 22 Mar 2022 20:29:03 +0000 Subject: [PATCH] an id is an int --- app/helpers/service_helper.py | 4 ++-- app/models/solver_run.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/app/helpers/service_helper.py b/app/helpers/service_helper.py index 67304c7..a8ad6f7 100644 --- a/app/helpers/service_helper.py +++ b/app/helpers/service_helper.py @@ -99,12 +99,12 @@ def solution_items(variables, solver_run): if v.varValue > 0: if 'Item_' in v.name: item_id = v.name.replace('Item_', '') - item = solver_run.get_item(item_id) + item = solver_run.get_item(int(item_id)) # add item to list and then remove from master item list if item: form_items.append(item) elif 'Bundle_' in v.name: bundle_id = v.name.replace('Bundle_', '') - bundle = solver_run.get_bundle(bundle_id) + bundle = solver_run.get_bundle(int(bundle_id)) if bundle: for item in bundle.items: diff --git a/app/models/solver_run.py b/app/models/solver_run.py index f03d67e..b3fbe8c 100644 --- a/app/models/solver_run.py +++ b/app/models/solver_run.py @@ -26,12 +26,12 @@ class SolverRun(BaseModel): def get_item(self, item_id: int) -> Item or None: for item in self.items: - if str(item.id) == item_id: + if item.id == item_id: return item def get_bundle(self, bundle_id: int) -> Bundle or None: for bundle in self.bundles: - if str(bundle.id) == bundle_id: + if bundle.id == bundle_id: return bundle def get_constraint_by_type(self, type: str) -> Constraint or None: