from django.db import models
 
# Create your models here.
import mptt
from mptt.models import MPTTModel, TreeForeignKey
 
class Folder(MPTTModel):
        name = models.CharField(max_length=50, unique=True, verbose_name=u'Name')
        parent = TreeForeignKey('self', null=True, blank=True, related_name='children', db_index=True)
 
        def __str__(self):
                return self.name
 
        class MPTTMeta:
                parent_attr = 'parent'
 
class Product(models.Model):
        name = models.CharField(max_length=50, db_index=True, verbose_name=u'PName')
        price = models.CharField(max_length=8, db_index=True, verbose_name=u'Price')
        parent = models.ForeignKey(Folder, blank=True, null=True, related_name='child_product')
 
        def __str__(self):
                return self.name
 
mptt.register(Folder)
mptt.register(Product)

from django.contrib import admin
from mptt.admin import MPTTModelAdmin
from .models import Folder, Product
 
admin.site.register(Folder, MPTTModelAdmin)
admin.site.register(Product, MPTTModelAdmin)