improved

Author: luolingchun

flask_openapi3/__init__.py

@@ -2,19 +2,19 @@
 # @Author  : llc
 # @Time    : 2021/4/30 10:14
 
-__version__ = '1.1.2'
-
-import os
+from .__version__ import __version__
 
+from .api_blueprint import APIBlueprint
 from .models.file import FileStorage
 from .models.info import Info
 from .models.oauth import OAuthConfig
-from .models.security import HTTPBase, HTTPBearer, OAuth2, APIKey, OpenIdConnect
-from .models.server import Server, ServerVariable
+from .models.security import APIKey
+from .models.security import HTTPBase
+from .models.security import HTTPBearer
+from .models.security import OAuth2
+from .models.security import OpenIdConnect
+from .models.server import Server
+from .models.server import ServerVariable
 from .models.tag import Tag
 from .models.validation_error import UnprocessableEntity
-from .openapi import APIBlueprint
 from .openapi import OpenAPI
-
-if os.environ.get("WERKZEUG_RUN_MAIN") is None:
-    print(f" * Powered by flask-openapi3 (version: {__version__})")

flask_openapi3/__version__.py

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# @Author  : llc
+# @Time    : 2022/4/30 9:20
+
+__version__ = '1.1.2'

flask_openapi3/do_wrapper.py

@@ -39,15 +39,15 @@ def _do_wrapper(
     request_kwargs = dict()
     try:
         if header:
-            request_headers = dict(request.headers) if request.headers is not None else {}
+            request_headers = dict(request.headers) or {}
             for key, value in header.__annotations__.items():
-                ket_title = key.title()
+                key_title = key.title()
                 # add original key
-                if ket_title in request_headers.keys():
-                    request_headers[key] = request_headers[ket_title]
+                if key_title in request_headers.keys():
+                    request_headers[key] = request_headers[key_title]
             request_kwargs.update({"header": header(**request_headers)})
         if cookie:
-            request_cookies = cookie(**request.cookies if request.cookies is not None else {})
+            request_cookies = cookie(**request.cookies or {})
             request_kwargs.update({"cookie": request_cookies})
         if path:
             request_path = path(**kwargs)
@@ -91,11 +91,9 @@ def _do_wrapper(
         if body:
             if body.__custom_root_type__:
                 # https://pydantic-docs.helpmanual.io/usage/models/#custom-root-types
-                body_ = body(
-                    __root__=request.get_json(silent=True) if request.get_json(silent=True) is not None else {})
+                body_ = body(__root__=request.get_json(silent=True) or {})
             else:
-                body_ = body(
-                    **request.get_json(silent=True) if request.get_json(silent=True) is not None else {})
+                body_ = body(**request.get_json(silent=True) or {})
             request_kwargs.update({"body": body_})
     except ValidationError as e:
         response = make_response(e.json())

flask_openapi3/openapi.py

@@ -159,7 +159,7 @@ def init_doc(self) -> None:
         self.register_blueprint(blueprint)
 
     def export_to_markdown(self) -> Response:
-        """Experimental"""
+        """Export to markdown(Experimental)"""
         md = StringIO()
 
         md.write(openapi_to_markdown(self.api_doc))
@@ -171,7 +171,7 @@ def export_to_markdown(self) -> Response:
 
     @property
     def api_doc(self) -> Dict:
-        """Generate spec json"""
+        """Generate Specification json"""
         spec = APISpec(
             openapi=self.openapi_version,
             info=self.info,