Initial commit: Django inventory system
This commit is contained in:
70
myproject/test_category_tree.py
Normal file
70
myproject/test_category_tree.py
Normal file
@@ -0,0 +1,70 @@
|
||||
"""
|
||||
Test script to create category hierarchy for testing the tree view
|
||||
"""
|
||||
import os
|
||||
import django
|
||||
|
||||
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproject.settings')
|
||||
django.setup()
|
||||
|
||||
from products.models import ProductCategory
|
||||
|
||||
print("Creating test category hierarchy...")
|
||||
print("="*60)
|
||||
|
||||
# Get existing categories
|
||||
existing = list(ProductCategory.objects.all()[:3])
|
||||
|
||||
if len(existing) >= 3:
|
||||
cat1, cat2, cat3 = existing[0], existing[1], existing[2]
|
||||
|
||||
# Make sure they are root categories first
|
||||
cat1.parent = None
|
||||
cat2.parent = None
|
||||
cat3.parent = None
|
||||
cat1.save()
|
||||
cat2.save()
|
||||
cat3.save()
|
||||
|
||||
print(f"Root categories:")
|
||||
print(f" 1. {cat1.name} (id={cat1.pk})")
|
||||
print(f" 2. {cat2.name} (id={cat2.pk})")
|
||||
print(f" 3. {cat3.name} (id={cat3.pk})")
|
||||
|
||||
# Create child for cat2 (if doesn't exist)
|
||||
child1, created = ProductCategory.objects.get_or_create(
|
||||
name="Test Child 1",
|
||||
defaults={'parent': cat2, 'is_active': True}
|
||||
)
|
||||
if not created:
|
||||
child1.parent = cat2
|
||||
child1.save()
|
||||
|
||||
print(f"\nChild category:")
|
||||
print(f" - {child1.name} (id={child1.pk}, parent={cat2.name})")
|
||||
|
||||
# Create grandchild for child1 (if doesn't exist)
|
||||
grandchild1, created = ProductCategory.objects.get_or_create(
|
||||
name="Test Grandchild 1",
|
||||
defaults={'parent': child1, 'is_active': True}
|
||||
)
|
||||
if not created:
|
||||
grandchild1.parent = child1
|
||||
grandchild1.save()
|
||||
|
||||
print(f"\nGrandchild category:")
|
||||
print(f" - {grandchild1.name} (id={grandchild1.pk}, parent={child1.name})")
|
||||
|
||||
print("\n" + "="*60)
|
||||
print("Hierarchy created successfully!")
|
||||
print("\nExpected tree structure:")
|
||||
print(f"{cat1.name}")
|
||||
print(f"{cat2.name}")
|
||||
print(f" {child1.name}")
|
||||
print(f" {grandchild1.name}")
|
||||
print(f"{cat3.name}")
|
||||
print("\nVisit http://127.0.0.1:8000/products/categories/ to see the tree view")
|
||||
|
||||
else:
|
||||
print("Not enough categories to create hierarchy")
|
||||
print("Please create at least 3 categories first")
|
||||
Reference in New Issue
Block a user