Category filter filtered with top level

Author: MySecondLanguage

nxtbn/product/admin_filters.py

@@ -60,10 +60,12 @@ class Meta:
 
 class CategoryFilter(filters.FilterSet):
     name = filters.CharFilter(lookup_expr='icontains')
+    parent = filters.ModelChoiceFilter(field_name='parent', queryset=Category.objects.all())
+    is_top_level = filters.BooleanFilter(field_name='parent', lookup_expr='isnull')
 
     class Meta:
         model = Category
-        fields = ('name',)
+        fields = ('name', 'parent', 'is_top_level',)
 
 
 class CollectionFilter(filters.FilterSet):

nxtbn/product/admin_queries.py

@@ -27,8 +27,6 @@ class ProductQuery(graphene.ObjectType):
 
     category = graphene.Field(CategoryType, id=graphene.ID(required=True))
     categories = DjangoFilterConnectionField(CategoryType)
-    category = graphene.Field(CategoryType, id=graphene.ID(required=True))
-
     # All translations
 
     product_translation = graphene.Field(ProductTranslationType, base_product_id=graphene.ID(required=True), lang_code=graphene.String(required=True))
@@ -97,13 +95,17 @@ def resolve_product_variants(root, info, **kwargs):
         return Product.objects.all()
 
     def resolve_category(root, info, id):
-        check_user_permissions(info, any_staff=True)
+        # check_user_permissions(info, any_staff=True)
 
         try:
             return Category.objects.get(pk=id)
         except Category.DoesNotExist:
             return None
 
+    def resolve_categories(root, info, **kwargs):
+        # check_user_permissions(info, any_staff=True)
+        return Category.objects.all()
+        
     # All translations
 
     def resolve_category_translation(root, info, base_category_id, lang_code):

nxtbn/product/admin_types.py

@@ -57,8 +57,12 @@ class Meta:
         filterset_class = CollectionFilter
 
 
+class CategoryChildInfoType(graphene.ObjectType):
+    has_child = graphene.Boolean()
+
 class CategoryType(DjangoObjectType):
     db_id = graphene.Int(source="id")
+    child_info = graphene.Field(CategoryChildInfoType)
     class Meta:
         model = Category
         fields = (
@@ -67,10 +71,18 @@ class Meta:
             'description',
             'meta_title',
             'meta_description',
+            'parent',
         )
         interfaces = (relay.Node,)
         filterset_class = CategoryFilter
 
+    def resolve_child_info(self, info):
+        has_child = self.subcategories.exists()
+        return CategoryChildInfoType(has_child=has_child)
+    
+
+
+
 
 class ProductTagType(DjangoObjectType):
     db_id = graphene.Int(source="id")