Merge pull request #94 from guokr/sanic

fix current_scopes request undefined

Author: fantastao

swagger_py_codegen/sanic.py

@@ -3,11 +3,18 @@
 from collections import OrderedDict
 
 from .base import Code, CodeGenerator
-from .jsonschema import Schema, SchemaGenerator, build_default
+from .jsonschema import build_default, build_data
 
 SUPPORT_METHODS = ['get', 'post', 'put', 'delete', 'patch', 'options', 'head']
 
 
+class Schema(Code):
+
+    template = 'sanic/schemas.tpl'
+    dest_template = '%(package)s/%(module)s/schemas.py'
+    override = True
+
+
 class Router(Code):
 
     template = 'sanic/routers.tpl'
@@ -66,6 +73,12 @@ class UIIndex(Code):
     dest_template = '%(package)s/static/swagger-ui/index.html'
 
 
+class SchemaGenerator(CodeGenerator):
+
+    def _process(self):
+        yield Schema(build_data(self.swagger))
+
+
 def _swagger_to_sanic_url(url, swagger_path_node):
     types = {
         'integer': 'int',

swagger_py_codegen/templates/sanic/blueprint.tpl

@@ -8,7 +8,7 @@ from .validators import security
 
 
 @security.scopes_loader
-def current_scopes():
+def current_scopes(request):
     return {{ scopes_supported }}
 
 bp = Blueprint('{{ blueprint }}', __name__)

swagger_py_codegen/templates/sanic/schemas.tpl

@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+
+# TODO: datetime support
+
+{% include '_do_not_change.tpl' %}
+
+base_path = '{{base_path}}'
+
+{% for name, value in schemas.items() %}
+{{ name }} = {{ value }}
+{%- endfor %}
+
+validators = {
+{%- for name, value in validators.items() %}
+    {{ name }}: {{ value }},
+{%- endfor %}
+}
+
+filters = {
+{%- for name, value in filters.items() %}
+    {{ name }}: {{ value }},
+{%- endfor %}
+}
+
+scopes = {
+{%- for name, value in scopes.items() %}
+    {{ name }}: {{ value }},
+{%- endfor %}
+}
+
+
+class Current(object):
+
+    request = None
+
+
+current = Current()
+
+
+class Security(object):
+
+    def __init__(self):
+        super(Security, self).__init__()
+        self._loader = lambda x: []
+
+    @property
+    def scopes(self):
+        return self._loader(current.request)
+
+    def scopes_loader(self, func):
+        self._loader = func
+        return func
+
+security = Security()
+
+
+{{ merge_default }}
+
+{{ normalize }}
+

swagger_py_codegen/templates/sanic/validators.tpl

@@ -17,7 +17,7 @@ from sanic.request import RequestParameters
 from jsonschema import Draft4Validator
 
 from .schemas import (
-    validators, filters, scopes, security, base_path, normalize)
+    validators, filters, scopes, security, base_path, normalize, current)
 
 
 def unpack(value):
@@ -115,6 +115,7 @@ def request_validate(view):
     def wrapper(*args, **kwargs):
         request = args[1]
         endpoint = _path_to_endpoint(request.uri_template)
+        current.request = request
         # scope
         if (endpoint, request.method) in scopes and not set(
                 scopes[(endpoint, request.method)]).issubset(set(security.scopes)):
@@ -144,7 +145,7 @@ def response_filter(view):
         request = args[1]
         resp = view(*args, **kwargs)
 
-	    from inspect import isawaitable
+        from inspect import isawaitable
         if isawaitable(resp):
             resp = await resp
         if isinstance(resp, HTTPResponse):