feat(root):add create export

core/utils/te/__init__.py

@@ -62,6 +62,9 @@ class NewTableEngine:
         if self.return_query:
             query = await self.exe()
             return query
+        if self.input.e:
+            query = await self.exe()
+            return query
         rows, count = await self.exe()
         self.input.extra_filter = None
         del self.engine

core/utils/te/engines/Mongo.py

@@ -2,6 +2,13 @@ import datetime
 import uuid
 from typing import Union, List
 from uuid import UUID
+import os
+import xlsxwriter
+import tempfile
+import string
+import copy
+import jdatetime
+
 
 # import dateutil.parser
 from celery import shared_task
@@ -34,12 +41,10 @@ class MongoEngine(AbstractEngineClass):
 
     async def exe(self):
         _f = self.filters = self.set_filters()
-        print("_f  ====>", _f)
         if self.unselect:
             tmp = self.query_class.find(_f, self.unselect)
         else:
             tmp = self.query_class.find(_f)
-
         sort_by = self.set_sort()
         if isinstance(sort_by, dict):
             tmp = tmp.sort(**sort_by)
@@ -49,7 +54,11 @@ class MongoEngine(AbstractEngineClass):
         self.rows = tmp.limit(
             self.input.get_limit()).skip(self.input.skip())
         if self.input.e and self.input.e.ot:
-            self.input.e.d = self.generate_output(self.query_class, _f)
+            copy_list = copy.deepcopy(tmp)
+            res = await copy_list.to_list(None)
+            self.input.e.d = self.generate_output(res)
+            return {'query': _f, 'sort': self.set_sort(), 'path': self.input.e.d}
+
         if self.return_query:
             return {'query':_f, 'sort':self.set_sort(), 'limit':self.input.get_limit(), 'page':self.input.skip()}
         return [self.model(**document) async for document in self.rows], await self.count()
@@ -249,15 +258,11 @@ class MongoEngine(AbstractEngineClass):
     def set_joins(self):
         pass
 
-    def generate_output(self, query, _f):
-        print(_f.__dict__)
-        return ''
+    def generate_output(self, _f):
+        print(self.input.e.ot)
         if self.input.e.ot == OutPutType.EXEL:
             file_name = str(uuid.uuid4().hex) + '.xlsx'
-            return _f.dict()
-            MongoEngine.output_generator(str(query).replace('false', '0'), params,
-                                         str(self.query_class.bind.url), file_name=file_name, cols=self.input.e.c)
-            return file_name
+            return self.output_generator(_f, file_name=file_name, cols=self.input.e.c)
         elif self.input.e.ot == OutPutType.JSON:
             pass
         elif self.input.e.ot == OutPutType.PDF:
@@ -265,7 +270,65 @@ class MongoEngine(AbstractEngineClass):
         else:
             raise Exception("Output type not supported.")
 
-    @staticmethod
-    @shared_task(name='output_generator_task_mongo')
-    def output_generator(query, params, url, file_name, cols):
-        pass
+    def _excel_temp_dir(self):
+        """پوشه temp پروژه برای ذخیره فایل‌های اکسل (همان مسیری که برای دانلود سرو می‌شود)."""
+        temp_dir = os.path.join(os.getcwd(), 'temp')
+        os.makedirs(temp_dir, exist_ok=True)
+        return temp_dir
+
+    def output_generator(self,query, file_name, cols):
+        datas = query
+        file_dir = os.path.join(self._excel_temp_dir(), file_name)
+        head_cols = list(str(string.ascii_uppercase))
+        headers = list(cols.keys())
+
+        workbook = xlsxwriter.Workbook(file_dir)
+        worksheet = workbook.add_worksheet()
+
+        # نوشتن هدرها
+        bold_format = workbook.add_format({'bold': True})
+
+        # نوشتن هدرها
+        for col_index, header in enumerate(headers):
+            worksheet.write(0, col_index, cols[header], bold_format)
+
+        # نوشتن داده‌ها
+        for row_index, item in enumerate(datas, start=1):
+            for col_index, col in enumerate(headers):
+                val = self.get_nested_value(item, col)
+                worksheet.write(row_index, col_index, val)
+        workbook.close()
+        return file_name
+
+    def get_nested_value(self, data, path, default='-'):
+        """
+        دریافت مقدار از دیکشنری تو در تو با مسیر dot notation
+        مثال: path = "design.title"
+        """
+        if not isinstance(path, str):
+            return default
+
+        keys = path.split('.')
+        val = data
+
+        for key in keys:
+            if isinstance(val, dict):
+                val = val.get(key, default)
+            else:
+                return default
+
+            if val is None:
+                return default
+
+        # تبدیل datetime و UUID به رشته برای اکسل
+        if isinstance(val, datetime.datetime):
+            return self.create_date(val)
+        if isinstance(val, uuid.UUID):
+            return str(val)
+
+        return val
+
+    def create_date(self, date):
+        shamsi_dt = jdatetime.datetime.fromgregorian(datetime=date)
+        return shamsi_dt.strftime("%Y/%m/%d - %H:%M:%S")
+

core/utils/te/utils/Models.py

@@ -35,7 +35,7 @@ class Cr(str, Enum):
 
 class OutPutType(str, Enum):
     JSON = 'json'
-    EXEL = 'exel'
+    EXEL = 'excel'
     PDF = 'pdf'