From 0f91bd86fbb8cea315c1b8333c8a950e4ee3ef52 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cau=C3=AA=20Napier?= Date: Fri, 3 Nov 2023 11:03:56 +0000 Subject: [PATCH 1/2] Update __init__.py --- valispace/__init__.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/valispace/__init__.py b/valispace/__init__.py index a698820..78d0656 100644 --- a/valispace/__init__.py +++ b/valispace/__init__.py @@ -732,6 +732,19 @@ def create_dataset_and_set_values(self, vali_id, input_data): return dataset_id + def get_requirements(self, date=None, project_id=None, spec_id=None): + if date is not None: + url = f"requirements/complete/as_of/{date}/" + else: + url = f"requirements/complete/" + if project_id is not None: + url += f"?project={project_id}" + if spec_id == None: + return self.request("GET", url) + else: + requirements = self.request("GET", url) + spec_requirements = [requirement for requirement in requirements if requirement["specification"]==spec_id] + return spec_requirements def vali_import_dataset(self, vali_id, data, headers=None): if headers is None: From fe820446bd9255f25be1c68b241f6755cd7eb7f6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cau=C3=AA=20Napier?= Date: Fri, 3 Nov 2023 13:59:05 +0000 Subject: [PATCH 2/2] Update __init__.py --- valispace/__init__.py | 31 +++++++++++++++---------------- 1 file changed, 15 insertions(+), 16 deletions(-) diff --git a/valispace/__init__.py b/valispace/__init__.py index 3d3428f..76caee3 100644 --- a/valispace/__init__.py +++ b/valispace/__init__.py @@ -148,9 +148,20 @@ def get_specifications(self, project_id): def get_groups(self, project_id): return self.get(f"requirements/groups?project={project_id}") - - def get_requirements(self, project_id): - return self.get(f"requirements/?project={project_id}") + + def get_requirements(self, date=None, project_id=None, spec_id=None): + if date is not None: + url = f"requirements/complete/as_of/{date}/" + else: + url = f"requirements/complete/" + if project_id is not None: + url += f"?project={project_id}" + if spec_id == None: + return self.request("GET", url) + else: + requirements = self.request("GET", url) + spec_requirements = [requirement for requirement in requirements if requirement["specification"]==spec_id] + return spec_requirements def get_vali_list(self, workspace_id=None, workspace_name=None, project_id=None, project_name=None, parent_id=None, parent_name=None, tag_id=None, tag_name=None, vali_marked_as_impacted=None): @@ -751,19 +762,7 @@ def create_dataset_and_set_values(self, vali_id, input_data): return dataset_id - def get_requirements(self, date=None, project_id=None, spec_id=None): - if date is not None: - url = f"requirements/complete/as_of/{date}/" - else: - url = f"requirements/complete/" - if project_id is not None: - url += f"?project={project_id}" - if spec_id == None: - return self.request("GET", url) - else: - requirements = self.request("GET", url) - spec_requirements = [requirement for requirement in requirements if requirement["specification"]==spec_id] - return spec_requirements + def vali_import_dataset(self, vali_id, data, headers=None): if headers is None: