lesscode-flask 0.0.79__tar.gz → 0.0.81__tar.gz

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lesscode-flask
-Version: 0.0.79
+Version: 0.0.81
 Summary: lesscode-flask 是基于flask的web开发脚手架项目，该项目初衷为简化开发过程，让研发人员更加关注业务。
 Home-page: https://lesscode-flask
 Author: Chao.yy

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/lesscode_flask/__init__.py

@@ -1,10 +1,12 @@
-__version__ = '0.0.79'
+__version__ = '0.0.81'
 
 import functools
 import logging
 import traceback
 
 from flask import Blueprint
+
+from lesscode_flask.export_data.data_download_handler import download_func_dict
 from lesscode_flask.utils.decorator.cache import deal_cache
 
 
@@ -15,11 +17,13 @@ class SQ_Blueprint(Blueprint):
         super().__init__(name=name, url_prefix=url_prefix, **kwargs)
 
     def decorator_handler(self, title: str, url: str = None, cache_enalbe: bool = False, cache_ex: int = 3600 * 10,
-                          content_type: str = "json", methods=['POST']):
+                          content_type: str = "json", methods=['POST'], export_enable: bool = False):
         options = {"methods": methods}
 
         def decorator(func):
             path = url if url else "/{}".format(func.__name__)
+            download_key = func.__module__ + "." + func.__name__
+            download_func_dict[download_key] = func
 
             @functools.wraps(func)
             def wrapper(*args, **kwargs):
@@ -31,9 +35,13 @@ class SQ_Blueprint(Blueprint):
                     except Exception as e:
                         logging.error(traceback.format_exc())
                         data = func(*args, **kwargs)
-                    return data
-                # 如果没有开启缓存，或者缓存未命中，则执行原始函数
-                data = func(*args, **kwargs)
+                else:
+                    # 如果没有开启缓存，或者缓存未命中，则执行原始函数
+                    data = func(*args, **kwargs)
+                if isinstance(data, dict) and export_enable:
+                    data.update({
+                        "download_key": download_key
+                    })
                 return data
 
             wrapper._title = title
@@ -45,11 +53,11 @@ class SQ_Blueprint(Blueprint):
         return decorator
 
     def post_route(self, title: str, url: str = None, cache_enalbe: bool = False, cache_ex: int = 3600 * 10,
-                   content_type: str = "json", methods=['POST']):
-        decorator = self.decorator_handler(title, url, cache_enalbe, cache_ex, content_type, methods)
+                   content_type: str = "json", methods=['POST'], export_enable: bool = False):
+        decorator = self.decorator_handler(title, url, cache_enalbe, cache_ex, content_type, methods, export_enable)
         return decorator
 
     def get_route(self, title: str, url: str = None, cache_enalbe: bool = False, cache_ex: int = 3600 * 10,
                   content_type: str = "json", methods=['GET']):
         decorator = self.decorator_handler(title, url, cache_enalbe, cache_ex, content_type, methods)
-        return decorator
+        return decorator

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/lesscode_flask/app.py

@@ -16,7 +16,7 @@ from werkzeug.middleware.proxy_fix import ProxyFix
 
 from lesscode_flask.model.response_result import ResponseResult
 from lesscode_flask.setup import setup_blueprint, setup_logging, setup_query_runner, setup_swagger, setup_sql_alchemy, \
-    setup_redis, setup_login_manager, setup_resource_register
+    setup_redis, setup_login_manager, setup_resource_register, setup_data_download
 from lesscode_flask.utils.helpers import inject_args, generate_uuid, app_config
 from lesscode_flask.utils.json.NotSortJSONProvider import NotSortJSONProvider
 from lesscode_flask.utils.redis.redis_helper import RedisHelper
@@ -147,6 +147,9 @@ class Lesscoder(Flask):
         for blueprint_name, blueprint in blueprint_map.items():
             self.register_blueprint(blueprint)
         setup_resource_register(self)
+        setup_data_download(self)
+      #  setup_static(self)
+
 
     @staticmethod
     def handle_exception(e):

lesscode_flask-0.0.81/lesscode_flask/export_data/data_download_handler.py

@@ -0,0 +1,94 @@
+import inspect
+import logging
+import os
+from datetime import datetime
+
+from flask import current_app
+from openpyxl import Workbook
+from openpyxl.styles import Alignment, Font
+
+download_func_dict = {}
+
+
+def upload_result_url(data_list: list, file_name: str = None, describe_text: str = None):
+    # 创建一个工作簿和一个工作表
+    wb = Workbook()
+    ws = wb.active
+    # 将数据写入工作表
+    for row in data_list:
+        row = ["，".join(list(set(x))) if type(x) == list else str(x) for x in row]
+        ws.append(row)
+    # 自适应列宽
+    for index, col in enumerate(ws.columns):
+        max_length = max([len(str(data[index])) for data in data_list])
+        column = col[0].column_letter  # 获取列字母
+        adjusted_width = min(60, (max_length * 2 + 2))  # 添加一些额外的空白以防止截断
+        ws.column_dimensions[column].width = adjusted_width
+    if describe_text:
+        ws.insert_rows(1, amount=1)
+        ws.merge_cells(f"A1:{column}1")  # 合并第1行单元格
+        ws["A1"] = describe_text
+    # img = Image("./static/img/img.png")
+    # ws.add_image(img, "A1")
+    # 保存工作簿
+    file_key = f'{file_name}-{datetime.now().strftime("%Y%m%d%H%M%S")}.xlsx'
+    path = f'{current_app.config.get("STATIC_PATH", "")}/data_download/{file_key}'
+    wb.save(path)
+    clear_excel_task()
+    return f'{current_app.config.get("OUTSIDE_SCREEN_IP")}/static/data_download/{file_key}', file_key
+
+
+def clear_excel_task():
+    delay = 3600 * 24
+    excel_path = os.path.join(current_app.config.get("STATIC_PATH", ""), "data_download")
+    files = get_files(excel_path)
+    for f in files:
+        now = datetime.now().timestamp()
+        if os.path.exists(f):
+            file_time = os.path.getctime(f)
+            if now - file_time > delay:
+                os.remove(f)
+                logging.info(f'delete file({f}) success')
+
+
+def get_files(path):
+    files = []
+    fs = os.listdir(path)
+    for f in fs:
+        file_path = f'{path}/{f}'
+        if os.path.isfile(file_path):
+            if ".gitkeep" not in file_path:
+                files.append(file_path)
+        else:
+            ft = get_files(file_path)
+            files.extend(ft)
+    return files
+
+
+def format_to_table_download(table_head_list=None, table_body_list=None):
+    if not table_head_list:
+        table_head_list = table_body_list["columns"]
+        table_body_list = table_body_list["dataSource"]
+    data_index_list = [one['dataIndex'] for one in table_head_list]
+    title_list = [one['title'] for one in table_head_list]
+    all_data_list = [title_list]
+    for one in table_body_list:
+        all_data_list.append([one.get(dataIndex, "") for dataIndex in data_index_list])
+    # table_head_map_dict = {one['dataIndex']: one['title'] for one in table_head_list}
+    #
+    # sta_map_dict = {}
+    # for one in table_body_list:
+    #     for tag in data_index_list:
+    #         if tag not in sta_map_dict:
+    #             sta_map_dict[tag] = [one.get(tag, "")]
+    #         else:
+    #             sta_map_dict[tag].append(one.get(tag, ""))
+    # sta_map_dict = {table_head_map_dict[k]: v for k, v in sta_map_dict.items()}
+    return all_data_list
+
+
+def convert_page(offset: int = None, size: int = None, page_num: int = None, page_size: int = None):
+    if not offset and not size:
+        offset = (page_num - 1) * page_size
+        size = page_size
+    return offset, size

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/lesscode_flask/setting/__init__.py

@@ -61,7 +61,7 @@ class BaseConfig:
     # 应用运行根路径
     # APPLICATION_PATH: str = f"{os.path.abspath(os.path.dirname(sys.argv[0]))}"
     # # 静态资源目录
-    STATIC_PATH: str = f"{os.path.abspath(os.path.dirname(sys.argv[0]))}"
+    STATIC_PATH: str = os.path.join(f"{os.path.abspath(os.path.dirname(sys.argv[0]))}", "static")
 
     SECRET_KEY = "423ad5ef841bbd073b415e4ba4136d7c94cac3f5e9bfeec1a21da35cd9ea6b46"
 

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/lesscode_flask/setup/__init__.py

@@ -4,9 +4,11 @@ from datetime import datetime
 from importlib import import_module
 from logging.handlers import TimedRotatingFileHandler
 
-from flask import current_app
+from flask import current_app, send_from_directory
 
+from lesscode_flask import download_func_dict
 from lesscode_flask.db import db
+from lesscode_flask.export_data.data_download_handler import convert_page, format_to_table_download, upload_result_url
 from lesscode_flask.log.access_log_handler import AccessLogHandler
 from lesscode_flask.model.user import AnonymousUser
 from lesscode_flask.service.authentication_service import get_token_user, get_api_user, get_gateway_user
@@ -400,3 +402,38 @@ def setup_resource_register(app):
                 return res
             except Exception as e:
                 raise e
+
+
+# def setup_static(app):
+#     @app.route("/static/<filename>")
+#     def static_resource(filename, **kwargs):
+#         """
+#         方法说明
+#         :param download_key:
+#         :return:
+#         """
+#         return send_from_directory('static', filename)
+
+
+def setup_data_download(app):
+    @app.route(f"{current_app.config.get('ROUTE_PREFIX')}/download/data_download", methods=['POST'])
+    def data_download(download_key, params, column_list=None, page_size=10, file_name="", page_number=1, offset=0,
+                      size=10):
+        """
+        方法说明
+        :param download_key:
+        :return:
+        """
+        if not file_name:
+            file_name = "data_export"
+        func = download_func_dict.get(download_key, {})
+        offset, size = convert_page(offset, size, page_number, page_size)
+        request_param = {"offset": offset, "size": size}
+        signature = inspect.signature(func)
+        for parameter_name, parameter in signature.parameters.items():
+            if params.get(parameter_name):
+                request_param[parameter_name] = params[parameter_name]
+        table_body_list = func(**request_param)
+        sta_map_dict = format_to_table_download(table_head_list=column_list, table_body_list=table_body_list)
+        url, file_key = upload_result_url(sta_map_dict, file_name=file_name)
+        return url

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/lesscode_flask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lesscode-flask
-Version: 0.0.79
+Version: 0.0.81
 Summary: lesscode-flask 是基于flask的web开发脚手架项目，该项目初衷为简化开发过程，让研发人员更加关注业务。
 Home-page: https://lesscode-flask
 Author: Chao.yy

{lesscode_flask-0.0.79 → lesscode_flask-0.0.81}/lesscode_flask.egg-info/SOURCES.txt

@@ -11,6 +11,8 @@ lesscode_flask.egg-info/top_level.txt
 lesscode_flask/db/__init__.py
 lesscode_flask/db/datasource.py
 lesscode_flask/db/executor.py
+lesscode_flask/export_data/__init__.py
+lesscode_flask/export_data/data_download_handler.py
 lesscode_flask/log/access_log_handler.py
 lesscode_flask/model/access_log.py
 lesscode_flask/model/auth_client.py