add

Author: foodszhang

swagger_py_codegen/sanic.py

@@ -4,6 +4,7 @@
 
 from .base import Code, CodeGenerator
 from .jsonschema import build_default, build_data
+import six
 
 SUPPORT_METHODS = ['get', 'post', 'put', 'delete', 'patch', 'options', 'head']
 
@@ -109,8 +110,12 @@ def _type(parameters):
     return url, params
 
 
-def _remove_characters(text, deletechars):
-    return text.translate({ord(x): None for x in deletechars})
+if six.PY3:
+    def _remove_characters(text, deletechars):
+        return text.translate({ord(x): None for x in deletechars})
+else:
+    def _remove_characters(text, deletechars):
+        return text.translate(None, deletechars)
 
 
 def _path_to_endpoint(swagger_path):
@@ -150,18 +155,18 @@ def _dependence_callback(self, code):
         # use sanic endpoint to replace default validator's key,
         # example: `('some_path_param', 'method')`
         validators = OrderedDict()
-        for k, v in schemas.data['validators'].items():
-            locations = {_location(loc): val for loc, val in v.items()}
+        for k, v in six.iteritems(schemas.data['validators']):
+            locations = {_location(loc): val for loc, val in six.iteritems(v)}
             validators[(_path_to_endpoint(k[0]), k[1])] = locations
 
         # filters
         filters = OrderedDict()
-        for k, v in schemas.data['filters'].items():
+        for k, v in six.iteritems(schemas.data['filters']):
             filters[(_path_to_endpoint(k[0]), k[1])] = v
 
         # scopes
         scopes = OrderedDict()
-        for k, v in schemas.data['scopes'].items():
+        for k, v in six.iteritems(schemas.data['scopes']):
             scopes[(_path_to_endpoint(k[0]), k[1])] = v
 
         schemas.data['validators'] = validators
@@ -191,7 +196,7 @@ def _process_data(self):
                 if validator:
                     methods[method]['requests'] = list(validator.keys())
 
-                for status, res_data in data[method].get('responses', {}).items():
+                for status, res_data in six.iteritems(data[method].get('responses', {})):
                     if isinstance(status, int) or status.isdigit():
                         example = res_data.get('examples', {}).get('application/json')
 

swagger_py_codegen/templates/sanic/schemas.tpl

@@ -7,9 +7,7 @@ import six
 
 base_path = '{{base_path}}'
 
-{% for name, value in schemas.items() %}
-{{ name }} = {{ value }}
-{%- endfor %}
+definitions = {{ definitions }}
 
 validators = {
 {%- for name, value in validators.items() %}
@@ -30,6 +28,8 @@ scopes = {
 }
 
 
+resolver = RefResolver.from_schema(definitions)
+
 class Current(object):
 
     request = None