From 22e96539c2bd8895b188d35b549bca2197437041 Mon Sep 17 00:00:00 2001 From: eeintech Date: Thu, 4 Mar 2021 12:24:53 -0500 Subject: [PATCH] Removed CategoryParameterTemplate class and fixed fetching of category parameters --- inventree/base.py | 8 +++++++- inventree/part.py | 11 ++++------- 2 files changed, 11 insertions(+), 8 deletions(-) diff --git a/inventree/base.py b/inventree/base.py index 60b7fcc9..a6004554 100644 --- a/inventree/base.py +++ b/inventree/base.py @@ -66,7 +66,13 @@ def list(cls, api, **kwargs): # Dict of query params to send to the API params = kwargs - response = api.get(cls.URL, params=params, **kwargs) + # Check if custom URL is present in request arguments + if 'url' in kwargs: + url = kwargs.pop('url') + else: + url = cls.URL + + response = api.get(url=url, params=params, **kwargs) if response is None: return None diff --git a/inventree/part.py b/inventree/part.py index 701da78c..68e825ed 100644 --- a/inventree/part.py +++ b/inventree/part.py @@ -31,15 +31,12 @@ def get_category_parameter_templates(self, fetch_parent=True): """ fetch_parent: enable to fetch templates for parent categories """ - return CategoryParameterTemplate.list(self._api, - category=self.pk, - fetch_parent=fetch_parent) - -class CategoryParameterTemplate(inventree.base.InventreeObject): - """ Class representing the PartCategoryParameterTemplate database model """ + parameters_url = f'part/category/{self.pk}/parameters' - URL = 'part/category/parameters' + return self.list(self._api, + url=parameters_url, + fetch_parent=fetch_parent) class Part(inventree.base.InventreeObject):