From 19dfee6e24662aa1ad6393f46cbf2aa1bcad1f97 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Markus=20Clau=C3=9F?= Date: Fri, 2 Jun 2023 09:27:51 +0200 Subject: [PATCH] remove org_id, fix error --- src/paveit/datamodels/machines.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/paveit/datamodels/machines.py b/src/paveit/datamodels/machines.py index 232b6a6..214accb 100755 --- a/src/paveit/datamodels/machines.py +++ b/src/paveit/datamodels/machines.py @@ -59,18 +59,16 @@ class MachineBase(Document): return data - def get_config(self, org_id: ObjectId, testname: LabtestsEnum): - test_id = Labtest.objects(org_id=org_id, test=testname).first() + def get_config(self, testname: LabtestsEnum): + test_id = Labtest.objects(org_id=self.org_id, test=testname).first() if not test_id: return [] test_id = test_id.id - machine_data = self.objects(tests__test=test_id).only('tests').first() - config_array = [] - for test in machine_data.tests: + for test in self.tests: if test.test.id == test_id: config_array = test.config