def test_elec(self):
electronics = PartCategory(self.api, 1)
# This is a top-level category, should not have a parent!
self.assertIsNone(electronics.getParentCategory())
self.assertEqual(electronics.name, "Electronics")
children = electronics.getChildCategories()
self.assertGreaterEqual(len(children), 3)
for child in children:
self.assertEqual(child.parent, 1)
child = PartCategory(self.api, pk=3)
self.assertEqual(child.name, 'Capacitors')
self.assertEqual(child.getParentCategory().pk, electronics.pk)
# Grab all child categories
children = PartCategory.list(self.api, parent=child.pk)
n = len(children)
for c in children:
self.assertEqual(c.parent, child.pk)
# Create some new categories under this one
for idx in range(3):
name = f"Subcategory {n+idx}"
cat = PartCategory.create(
self.api, {
"parent": child.pk,
"name": name,
"description": "A new subcategory",
})
self.assertEqual(cat.parent, child.pk)
self.assertEqual(cat.name, name)
# Edit the name of the new location
cat.save({
"name": f"{name}_suffix",
})
# Reload from server, and check
cat.reload()
self.assertEqual(cat.name, f"{name}_suffix")
# Number of children should have increased!
self.assertEqual(len(child.getChildCategories()), n + 3)
def is_new_part(category_id: int, part_info: dict) -> int:
''' Check if part exists based on parameters (or description) '''
global inventree_api
# Get category object
part_category = PartCategory(inventree_api, category_id)
# Fetch all parts from category and subcategories
part_list = []
part_list.extend(part_category.getParts())
for subcategory in part_category.getChildCategories():
part_list.extend(subcategory.getParts())
# Extract parameter from part info
new_part_parameters = part_info['parameters']
template_list = ParameterTemplate.list(inventree_api)
def fetch_template_name(template_id):
for item in template_list:
if item.pk == template_id:
return item.name
# Retrieve parent category name for parameters compare
try:
category_name = part_category.getParentCategory().name
except AttributeError:
category_name = part_category.name
filters = config_interface.load_category_parameters_filters(
category=category_name,
supplier_config_path=settings.CONFIG_PARAMETERS_FILTERS)
# cprint(filters)
for part in part_list:
# Get part parameters
db_part_parameters = part.getParameters()
part_parameters = {}
for parameter in db_part_parameters:
parameter_name = fetch_template_name(parameter.template)
parameter_value = parameter.data
part_parameters[parameter_name] = parameter_value
if new_part_parameters:
# Compare database part with new part
compare = part_tools.compare(
new_part_parameters=new_part_parameters,
db_part_parameters=part_parameters,
include_filters=filters)
else:
# Compare with name and description
compare = part_info['name'] == part.name or part_info[
'description'] == part.description
if compare:
cprint(f'\n[TREE]\tFound part match in database (pk = {part.pk})',
silent=settings.HIDE_DEBUG)
return part.pk
cprint(f'\n[TREE]\tNo match found in database', silent=settings.HIDE_DEBUG)
return 0