Syndication - export external sources

1 files changed, 76 insertions, 8 deletions

diff --git a/ishtar_common/rest.py b/ishtar_common/rest.py
index f85061b69..e489e5f80 100644
--- a/ishtar_common/rest.py
+++ b/ishtar_common/rest.py
@@ -1,7 +1,10 @@
-import json
+import datetime
+import requests
 
 from django.conf import settings
 from django.db.models import Q
+from django.http import HttpResponse
+from django.shortcuts import reverse
 from django.utils.translation import activate, deactivate
 
 from rest_framework import authentication, permissions, generics
@@ -10,8 +13,8 @@ from rest_framework.views import APIView
 
 from ishtar_common import models_rest
 from ishtar_common.models_common import GeneralType
+from ishtar_common.models_imports import Importer
 from ishtar_common.views_item import get_item
-from ishtar_common.serializers_utils import generic_get_results
 
 
 class IpModelPermission(permissions.BasePermission):
@@ -39,15 +42,24 @@ class SearchAPIView(APIView):
             content_type__model=self.model._meta.model_name,
         )
 
-    def get(self, request, format=None):
+    def get(self, request, slug=None, format=None, data_type="json"):
+        search_model = self.search_model_query(request).all()[0]
+        # get default importer
+        if slug:
+            q = search_model.export.filter(slug=slug)
+            if not q.count():
+                return HttpResponse('Page not found', status=404)
+            table_cols, col_names = q.all()[0].get_columns()
+        else:
+            table_cols, col_names = self.model.get_columns(dict_col_labels=False)
+
         _get_item = get_item(
             self.model,
             "get_" + self.model.SLUG,
             self.model.SLUG,
-            no_permission_check=True
-            # TODO: own_table_cols=get_table_cols_for_ope() - adapt columns
+            no_permission_check=True,
+            own_table_cols=table_cols
         )
-        search_model = self.search_model_query(request).all()[0]
         if search_model.limit_query:
             query = search_model.limit_query
             if request.GET.get("search_vector", None):
@@ -55,13 +67,30 @@ class SearchAPIView(APIView):
             request.GET._mutable = True
             request.GET["search_vector"] = query
             request.GET._mutable = False
-        data_type = "json"
+
+        # source-1-689-source-1-1853 -> 689-1853
+        if "selected_ids" in request.GET:
+            value = request.GET["selected_ids"]
+            values = []
+            for idx, k in enumerate(value.split("-")):
+                if idx % 3 == 2:
+                    values.append(k)
+            request.GET._mutable = True
+            request.GET["selected_ids"] = "-".join(values)
+            request.GET._mutable = False
+
         if request.GET.get("data_type", None):
             data_type = request.GET.get("data_type")
-        response = _get_item(request, data_type=data_type)
+        response = _get_item(request, col_names=col_names, data_type=data_type,
+                             type=data_type)
         return response
 
 
+class ExportAPIView(SearchAPIView):
+    def get(self, request, slug=None, format=None, data_type="csv"):
+        return super().get(request, slug=slug, format=format, data_type=data_type)
+
+
 class FacetAPIView(APIView):
     authentication_classes = (authentication.TokenAuthentication,)
     permission_classes = (permissions.IsAuthenticated, IpModelPermission)
@@ -78,6 +107,45 @@ class FacetAPIView(APIView):
     def get(self, request, format=None):
         values = {}
         base_queries = self._get_base_search_model_queries()
+
+        # config
+        values["config"] = {
+            "search_columns": "",
+            "search_columns_label": "",
+            "exports": "",
+            "exports_label": ""
+        }
+        for model in self.models:
+            model_name = f"{model._meta.app_label}-{model._meta.model_name}-"
+            q = models_rest.ApiSearchModel.objects.filter(
+                user=request.user.apiuser,
+                content_type__app_label=model._meta.app_label,
+                content_type__model=model._meta.model_name
+            )
+            if not q.count():
+                continue
+            search_model = q.all()[0]
+            # cols, col_names = search_model.table_format.get_columns()
+            cols, col_names = model.get_columns(dict_col_labels=False)
+            cols, col_names = [c for c in cols if c], [c for c in col_names if c]
+            if cols and col_names:
+                if values["config"]["search_columns"]:
+                    values["config"]["search_columns"] += "||"
+                    values["config"]["search_columns_label"] += "||"
+                values["config"]["search_columns"] += "||".join([
+                    model_name + (col[0] if isinstance(col, list) else col)
+                    for col in cols
+                ])
+                values["config"]["search_columns_label"] += "||".join([
+                    model_name + col for col in col_names
+                ])
+            for export in search_model.export.all():
+                if values["config"]["exports"]:
+                    values["config"]["exports"] += "||"
+                    values["config"]["exports_label"] += "||"
+                values["config"]["exports"] += model_name + export.slug
+                values["config"]["exports_label"] += model_name + export.name
+
         for idx, select_form in enumerate(self.select_forms):
             # only send types matching permissions
             model, q = base_queries[idx]