add security option

lazy-labs · Dec 11, 2019 · 1694188 · 1694188
1 parent 4979c5f
commit 1694188
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@ def get_packages(package):
         "starlette<=1",
         "apispec<3",
     ],
-    version="0.0.10",
+    version="0.0.11",
     url="https://github.com/slv0/start_resty",
     license="BSD",
     description="The web framework",

diff --git a/star_resty/apidocs/setup.py b/star_resty/apidocs/setup.py
@@ -103,8 +103,12 @@ def setup_operation(endpoint: Method, version=2):
         'summary': options.summary,
         'produces': [endpoint.serializer.media_type],
         'parameters': resolve_parameters(meta),
-        'responses': resolve_responses(endpoint)
+        'responses': resolve_responses(endpoint),
     }
+
+    if options.security is not None:
+        res['security'] = options.security
+
     if version > 2:
         res['requestBody'] = resolve_request_body(meta)
     else:

diff --git a/star_resty/operation/schema.py b/star_resty/operation/schema.py
@@ -6,3 +6,7 @@ class Operation(NamedTuple):
     description: Optional[str] = None
     summary: Optional[str] = None
     errors: Sequence[Any] = ()
+    security: Optional[Sequence] = None
+
+    def update(self, **kwargs):
+        return self._replace(**kwargs)