django-fast-treenode 2.0.5__py3-none-any.whl → 2.0.6__py3-none-any.whl

{django_fast_treenode-2.0.5.dist-info → django_fast_treenode-2.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: django-fast-treenode
-Version: 2.0.5
+Version: 2.0.6
 Summary: Application for supporting tree (hierarchical) data structure in Django projects
 Home-page: https://github.com/TimurKady/django-fast-treenode
 Author: Timur Kady

{django_fast_treenode-2.0.5.dist-info → django_fast_treenode-2.0.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 treenode/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-treenode/admin.py,sha256=x5vJDiZjqcsUAVlK1qkZPK66lQeJq8zYgWwwejentYM,14077
+treenode/admin.py,sha256=ltpCmPCpZpIAOX-ZIuxrleEUvBiUj7AX-jygUjGBHEs,15209
 treenode/apps.py,sha256=M0O9IKEnJZFfhfz12v4wksYJ-0ECyj1Cy3qXrfywos8,472
 treenode/cache.py,sha256=Z_FpaS0vTKXqAI4n1QkZ7A_ILsLU3Q8rLgerA6pYyAA,7210
 treenode/forms.py,sha256=imoLzr7qFlOBdeMQK12rAFp6hINpe0PS_U4-V8RRcDU,2802
@@ -34,8 +34,8 @@ treenode/utils/__init__.py,sha256=jAVT-uNGQDKPVq5ET4JE2GhRpzbxFdHU25w9sLC3_To,13
 treenode/utils/base36.py,sha256=ydgu9hqDaK-WyS8zG-mtSWo7hJqbB4iHqkGz4-IVrb4,834
 treenode/utils/exporter.py,sha256=l-IEsk2fK8xe-E-Hi7SMOV7ZIwxM6mZih5-0rhvsx9E,5396
 treenode/utils/importer.py,sha256=n23PBFPXN-6S_Dl8Qmx56sp0eLy_lWfpxrzJqwygQf0,14190
-django_fast_treenode-2.0.5.dist-info/LICENSE,sha256=GiiEe4Y9oOCbn9eGuNew1mMYHU_bJWaCK9zOusnKvvU,1091
-django_fast_treenode-2.0.5.dist-info/METADATA,sha256=sjucboSzhcI6bto4f0rvLjtKk2c5-2qQwohYSOQJWsc,23109
-django_fast_treenode-2.0.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-django_fast_treenode-2.0.5.dist-info/top_level.txt,sha256=fmgxHbXyx1O2MPi_9kjx8aL9L-8TmV0gre4Go8XgqFk,9
-django_fast_treenode-2.0.5.dist-info/RECORD,,
+django_fast_treenode-2.0.6.dist-info/LICENSE,sha256=GiiEe4Y9oOCbn9eGuNew1mMYHU_bJWaCK9zOusnKvvU,1091
+django_fast_treenode-2.0.6.dist-info/METADATA,sha256=DGr3UvNjD2IXwN9RETGgHsUwhgb417Z9drd2DIcrSko,23109
+django_fast_treenode-2.0.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+django_fast_treenode-2.0.6.dist-info/top_level.txt,sha256=fmgxHbXyx1O2MPi_9kjx8aL9L-8TmV0gre4Go8XgqFk,9
+django_fast_treenode-2.0.6.dist-info/RECORD,,

treenode/admin.py

@@ -6,7 +6,7 @@ This module provides Django admin integration for the TreeNode model.
 It includes custom tree-based sorting, optimized queries, and
 import/export functionality for hierarchical data structures.
 
-Version: 2.0.0
+Version: 2.0.6
 Author: Timur Kady
 Email: kaduevtr@gmail.com
 """
@@ -81,7 +81,7 @@ class TreeNodeAdminModel(admin.ModelAdmin):
     TREENODE_DISPLAY_MODE_INDENTATION = 'indentation'
 
     treenode_display_mode = TREENODE_DISPLAY_MODE_ACCORDION
-
+    import_export = False  # Track import/export availability
     change_list_template = "admin/tree_node_changelist.html"
     ordering = []
     list_per_page = 1000
@@ -107,10 +107,25 @@ class TreeNodeAdminModel(admin.ModelAdmin):
         """Динамически добавляем поле `tn_order` в `list_display`."""
         super().__init__(model, admin_site)
 
-        # Если `list_display` пустой, берем все `fields`
+        # If `list_display` is empty, take all `fields`
         if not self.list_display:
             self.list_display = [field.name for field in model._meta.fields]
 
+        # Check for necessary dependencies
+        self.import_export = all(
+            importlib.util.find_spec(pkg) is not None
+            for pkg in ["openpyxl", "pyyaml", "pandas"]
+        )
+
+        if self.import_export:
+            from .utils import TreeNodeImporter, TreeNodeExporter
+
+            self.TreeNodeImporter = TreeNodeImporter
+            self.TreeNodeExporter = TreeNodeExporter
+        else:
+            self.TreeNodeImporter = None
+            self.TreeNodeExporter = None
+
     def get_queryset(self, request):
         """Override get_ueryset()."""
         queryset = super().get_queryset(request)
@@ -182,10 +197,7 @@ class TreeNodeAdminModel(admin.ModelAdmin):
     def changelist_view(self, request, extra_context=None):
         """Changelist View."""
         extra_context = extra_context or {}
-        extra_context['import_export_enabled'] = all(
-            importlib.util.find_spec(pkg)
-            is not None for pkg in ["openpyxl", "pyyaml", "pandas"]
-        )
+        extra_context['import_export_enabled'] = self.import_export
         return super().changelist_view(request, extra_context=extra_context)
 
     def get_ordering(self, request):
@@ -250,12 +262,19 @@ class TreeNodeAdminModel(admin.ModelAdmin):
         return form
 
     def get_urls(self):
-        """Extend admin URLs with custom import/export routes."""
+        """
+        Extend admin URLs with custom import/export routes.
+
+        Register these URLs only if all the required packages are installed.
+        """
         urls = super().get_urls()
-        custom_urls = [
-            path('import/', self.import_view, name='tree_node_import'),
-            path('export/', self.export_view, name='tree_node_export'),
-        ]
+        if self.import_export:
+            custom_urls = [
+                path('import/', self.import_view, name='tree_node_import'),
+                path('export/', self.export_view, name='tree_node_export'),
+            ]
+        else:
+            custom_urls = []
         return custom_urls + urls
 
     def import_view(self, request):
@@ -265,6 +284,14 @@ class TreeNodeAdminModel(admin.ModelAdmin):
         File upload processing, auto-detection of format, validation and data
         import.
         """
+        if not self.import_export:
+            self.message_user(
+                request,
+                "Import functionality is disabled because required \
+packages are not installed."
+            )
+            return redirect("..")
+
         if request.method == 'POST':
             if 'file' not in request.FILES:
                 return render(
@@ -317,6 +344,14 @@ class TreeNodeAdminModel(admin.ModelAdmin):
         link for manual download,
         and a button to go to the model page.
         """
+        if not self.import_export:
+            self.message_user(
+                request,
+                "Export functionality is disabled because required \
+packages are not installed."
+            )
+            return redirect("..")
+
         # If the download parameter is present, we give the file directly
         if 'download' in request.GET:
             # Get file format