优化GenerateResponse

module/core.py

@@ -19,29 +19,34 @@ class GenerateResponse:
     TYPE = Union[str, Response]
 
     def __init__(self):
-        self.response = {}
-        self.onlyJSON = False
-
-    def json(self) -> TYPE:
-        responseJSON = json.dumps(self.response, ensure_ascii=False)
-        if self.stream:
+        self.code = 0
+        self.message = ''
+        self.data = None
+        self.streamFormat = False
+        self.streamResponse = False
+
+    def generate(self) -> TYPE:
+        responseJSON = json.dumps({
+            'code': self.code,
+            'message': self.message,
+            'data': self.data
+        }, ensure_ascii=False)
+        if self.streamFormat:
             return f'data: {responseJSON}\n\n'
+        if self.streamResponse:
+            return Response(f'data: {responseJSON}\n\n', media_type='text/event-stream')
         return Response(responseJSON, media_type='application/json')
 
-    def error(self, code: int, message: str, stream=False) -> TYPE:
-        self.response = {
-            'code': code,
-            'message': message,
-            'data': None
-        }
-        self.stream = stream
-        return self.json()
+    def error(self, code: int, message: str, streamFormat=False, streamResponse=False) -> TYPE:
+        self.code = code
+        self.message = message
+        self.streamFormat = streamFormat
+        self.streamResponse = streamResponse
+        return self.generate()
 
-    def success(self, data: Any, stream=False) -> TYPE:
-        self.response = {
-            'code': 200,
-            'message': 'success',
-            'data': data
-        }
-        self.stream = stream
-        return self.json()
+    def success(self, data: Any, streamFormat=False) -> TYPE:
+        self.code = 200
+        self.message = 'success'
+        self.data = data
+        self.streamFormat = streamFormat
+        return self.generate()