lesscode-flask 0.1.5__tar.gz → 0.1.6__tar.gz

{lesscode_flask-0.1.5 → lesscode-flask-0.1.6}/PKG-INFO

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

{lesscode_flask-0.1.5 → lesscode-flask-0.1.6}/lesscode_flask/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.1.5'
+__version__ = '0.1.6'
 
 import functools
 import logging

lesscode-flask-0.1.6/lesscode_flask/utils/oss/__init__.py

@@ -0,0 +1,92 @@
+import os
+from io import BytesIO
+
+from lesscode_flask.utils.helpers import app_config
+from lesscode_flask.utils.oss.ks3_oss import Ks3Oss
+
+
+class CommonOss:
+    def __init__(self, storage_type, **kwargs):
+        self.storage_type = storage_type
+        self.storage_config = kwargs.get("storage_config", {}) if kwargs.get("storage_config", {}) else app_config.get(
+            "STORAGE_CONFIG", {})
+
+    def upload(self, **kwargs):
+        file_url_list = []
+        files = kwargs.get("files", [])
+        if not files:
+            raise Exception("files is empty")
+        if self.storage_type == "ks3":
+            storage_config = self.storage_config or dict()
+            ks3 = Ks3Oss(**storage_config)
+            for f in files:
+                if not isinstance(f, dict):
+                    file_name = f.filename
+                    stream = f.stream.read()
+                    file_stream = BytesIO(stream)
+                    content = file_stream.getvalue().decode('utf-8')
+                    url = ks3.save(key=file_name, content=content, content_type="string", policy="public-read")
+                    file_url_list.append({"key": file_name, "url": url})
+                else:
+                    key = f.get("key", "")
+                    _steam = f.get("stream")
+                    stream = _steam.stream.read()
+                    file_stream = BytesIO(stream)
+                    content = file_stream.getvalue().decode('utf-8')
+                    url = ks3.save(key=key, content=content, content_type="string", policy="public-read")
+                    file_url_list.append({"key": key, "url": url})
+        elif self.storage_type == "file":
+            storage_dir = kwargs.get("storage_dir", "")
+            if not storage_dir:
+                storage_dir = app_config.get("STORAGE_DIR")
+            if not storage_dir:
+                raise Exception("storage_dir is empty")
+
+            for f in files:
+                if not isinstance(f, dict):
+                    file_name = f.filename
+                    stream = f.stream.read()
+                    file_stream = BytesIO(stream)
+                    file_path = os.path.join(storage_dir, file_name)
+                    with open(file_path, 'wb') as outfile:
+                        outfile.write(file_stream.getvalue())
+                    file_url_list.append({"key": file_name, "url": file_path})
+                else:
+                    key = f.get("key", "")
+                    _steam = f.get("stream")
+                    stream = _steam.stream.read()
+                    file_stream = BytesIO(stream)
+                    key_list = []
+                    if "\\" in key:
+                        key_list = key.split("\\")
+                    if "/" in key:
+                        key_list = key.split("/")
+                    file_path = storage_dir
+                    if key_list:
+                        for k in key_list:
+                            file_path = os.path.join(file_path, k)
+                    with open(file_path, 'wb') as outfile:
+                        outfile.write(file_stream.getvalue())
+                    file_url_list.append({"key": key, "url": file_path})
+        return file_url_list
+
+    def download(self, key, **kwargs):
+        if self.storage_type == "ks3":
+            storage_config = self.storage_config or dict()
+            ks3 = Ks3Oss(**storage_config)
+            return ks3.get_file(key)
+        if self.storage_type == "file":
+            storage_dir = kwargs.get("storage_dir", "")
+            if not storage_dir:
+                storage_dir = app_config.get("STORAGE_DIR")
+            key_list = []
+            if "\\" in key:
+                key_list = key.split("\\")
+            if "/" in key:
+                key_list = key.split("/")
+            file_path = storage_dir
+            if key_list:
+                for k in key_list:
+                    file_path = os.path.join(file_path, k)
+            with open(file_path, 'rb') as f:
+                return f.read()

{lesscode_flask-0.1.5 → lesscode-flask-0.1.6}/lesscode_flask.egg-info/PKG-INFO

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