merge

Author: foodszhang

swagger_py_codegen/jsonschema.py

@@ -5,7 +5,7 @@
 from inspect import getsource
 
 from .base import Code, CodeGenerator
-from .parser import RefNode
+from .parser import schema_var_name
 
 
 class Schema(Code):
@@ -89,8 +89,10 @@ def build_data(swagger):
                 scopes[(endpoint, method)] = list(security.values()).pop()
                 break
 
+    schemas = OrderedDict([(schema_var_name(path), swagger.get(path)) for path in swagger.definitions])
+
     data = dict(
-        definitions={'definitions':swagger.origin_data.get('definitions', {})},
+        schemas=schemas,
         validators=validators,
         filters=filters,
         scopes=scopes,
@@ -107,7 +109,7 @@ def _process(self):
         yield Schema(build_data(self.swagger))
 
 
-def merge_default(schema, value, get_first=True, resolver=None):
+def merge_default(schema, value, get_first=True):
     # TODO: more types support
     type_defaults = {
         'integer': 9573,
@@ -117,17 +119,17 @@ def merge_default(schema, value, get_first=True, resolver=None):
         'boolean': False
     }
 
-    results = normalize(schema, value, type_defaults, resolver=resolver)
+    results = normalize(schema, value, type_defaults)
     if get_first:
         return results[0]
     return results
 
 
-def build_default(schema, resolver=None):
-    return merge_default(schema, None, resolver=resolver)
+def build_default(schema):
+    return merge_default(schema, None)
 
 
-def normalize(schema, data, required_defaults=None, resolver=None):
+def normalize(schema, data, required_defaults=None):
     if required_defaults is None:
         required_defaults = {}
     errors = []
@@ -215,7 +217,7 @@ def _normalize_dict(schema, data):
 
     def _normalize_list(schema, data):
         result = []
-        if hasattr(data, '__iter__') and not isinstance(data, (dict, RefNode)):
+        if hasattr(data, '__iter__') and not isinstance(data, dict):
             for item in data:
                 result.append(_normalize(schema.get('items'), item))
         elif 'default' in schema:
@@ -228,15 +230,6 @@ def _normalize_default(schema, data):
         else:
             return data
 
-    def _normalize_ref(schema, data):
-        if resolver == None:
-            raise TypeError("resolver must be provided")
-        ref = schema.get(u"$ref")
-        scope, resolved = resolver.resolve(ref)
-        return _normalize(resolved, data)
-
-
-
     def _normalize(schema, data):
         if schema is True or schema == {}:
             return data
@@ -246,13 +239,10 @@ def _normalize(schema, data):
             'object': _normalize_dict,
             'array': _normalize_list,
             'default': _normalize_default,
-            'ref': _normalize_ref
         }
         type_ = schema.get('type', 'object')
         if type_ not in funcs:
             type_ = 'default'
-        if schema.get(u'$ref', None):
-            type_ = 'ref'
 
         return funcs[type_](schema, data)
 

swagger_py_codegen/templates/falcon/validators.tpl

@@ -14,7 +14,7 @@ from werkzeug.datastructures import MultiDict, Headers
 from jsonschema import Draft4Validator
 
 from .schemas import (
-    validators, filters, scopes, resolver, security, base_path, normalize)
+    validators, filters, scopes, security, base_path, normalize)
 
 
 if six.PY3:
@@ -44,7 +44,7 @@ class JSONEncoder(json.JSONEncoder):
 class FalconValidatorAdaptor(object):
 
     def __init__(self, schema):
-        self.validator = Draft4Validator(schema, resolver=resolver)
+        self.validator = Draft4Validator(schema)
 
     def validate_number(self, type_, value):
         try:
@@ -87,7 +87,7 @@ class FalconValidatorAdaptor(object):
     def validate(self, value):
         value = self.type_convert(value)
         errors = {e.path[0]: e.message for e in self.validator.iter_errors(value)}
-        return normalize(self.validator.schema, value, resolver=resolver)[0], errors
+        return normalize(self.validator.schema, value)[0], errors
 
 
 def request_validate(req, resp, resource, params):
@@ -154,15 +154,15 @@ def response_filter(req, resp, resource):
             'Not defined',
             description='`%d` is not a defined status code.' % status)
 
-    _resp, errors = normalize(schemas['schema'], req.context['result'], resolver=resolver)
+    _resp, errors = normalize(schemas['schema'], req.context['result'])
     if schemas['headers']:
         headers, header_errors = normalize(
-            {'properties': schemas['headers']}, headers, resolver=resolver)
+            {'properties': schemas['headers']}, headers)
         errors.extend(header_errors)
     if errors:
         raise falcon.HTTPInternalServerError(title='Expectation Failed',
                                              description=errors)
 
     if 'result' not in req.context:
         return
-    resp.body = json.dumps(_resp)
+    resp.body = json.dumps(_resp)

swagger_py_codegen/templates/flask/validators.tpl

@@ -15,7 +15,7 @@ from flask_restful.utils import unpack
 from jsonschema import Draft4Validator
 
 from .schemas import (
-    validators, filters, scopes, resolver, security, merge_default, normalize)
+    validators, filters, scopes, security, merge_default, normalize)
 
 
 class JSONEncoder(json.JSONEncoder):
@@ -29,7 +29,7 @@ class JSONEncoder(json.JSONEncoder):
 class FlaskValidatorAdaptor(object):
 
     def __init__(self, schema):
-        self.validator = Draft4Validator(schema, resolver=resolver)
+        self.validator = Draft4Validator(schema)
 
     def validate_number(self, type_, value):
         try:
@@ -72,7 +72,7 @@ class FlaskValidatorAdaptor(object):
     def validate(self, value):
         value = self.type_convert(value)
         errors = list(e.message for e in self.validator.iter_errors(value))
-        return normalize(self.validator.schema, value, resolver=resolver)[0], errors
+        return normalize(self.validator.schema, value)[0], errors
 
 
 def request_validate(view):
@@ -136,10 +136,10 @@ def response_filter(view):
             # return resp, status, headers
             abort(500, message='`%d` is not a defined status code.' % status)
 
-        resp, errors = normalize(schemas['schema'], resp, resolver=resolver)
+        resp, errors = normalize(schemas['schema'], resp)
         if schemas['headers']:
             headers, header_errors = normalize(
-                {'properties': schemas['headers']}, headers, resolver=resolver)
+                {'properties': schemas['headers']}, headers)
             errors.extend(header_errors)
         if errors:
             abort(500, message='Expectation Failed', errors=errors)

swagger_py_codegen/templates/jsonschema/schemas.tpl

@@ -1,17 +1,16 @@
 # -*- coding: utf-8 -*-
 
 import six
-from jsonschema import RefResolver
-from swagger_py_codegen.parser import RefNode
 
 # TODO: datetime support
 
-
 {% include '_do_not_change.tpl' %}
 
 base_path = '{{base_path}}'
 
-definitions = {{ definitions }}
+{% for name, value in schemas.items() %}
+{{ name }} = {{ value }}
+{%- endfor %}
 
 validators = {
 {%- for name, value in validators.items() %}
@@ -31,7 +30,6 @@ scopes = {
 {%- endfor %}
 }
 
-resolver = RefResolver.from_schema(definitions)
 
 class Security(object):
 

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, resolver, base_path, normalize, current)
+    validators, filters, scopes, security, base_path, normalize, current)
 
 
 def unpack(value):
@@ -63,7 +63,7 @@ class JSONEncoder(json.JSONEncoder):
 class SanicValidatorAdaptor(object):
 
     def __init__(self, schema):
-        self.validator = Draft4Validator(schema, resolver=resolver)
+        self.validator = Draft4Validator(schema)
 
     def validate_number(self, type_, value):
         try:
@@ -106,7 +106,7 @@ class SanicValidatorAdaptor(object):
     def validate(self, value):
         value = self.type_convert(value)
         errors = list(e.message for e in self.validator.iter_errors(value))
-        return normalize(self.validator.schema, value, resolver=resolver)[0], errors
+        return normalize(self.validator.schema, value)[0], errors
 
 
 def request_validate(view):
@@ -175,10 +175,10 @@ def response_filter(view):
             # return resp, status, headers
             raise ServerError('`%d` is not a defined status code.' % status, 500)
 
-        resp, errors = normalize(schemas['schema'], resp, resolver=resolver)
+        resp, errors = normalize(schemas['schema'], resp)
         if schemas['headers']:
             headers, header_errors = normalize(
-                {'properties': schemas['headers']}, headers, resolver=resolver)
+                {'properties': schemas['headers']}, headers)
             errors.extend(header_errors)
         if errors:
             raise ServerError('Expectation Failed', 500)

swagger_py_codegen/templates/tornado/validators.tpl

@@ -11,13 +11,13 @@ import six
 from functools import wraps
 from jsonschema import Draft4Validator
 
-from .schemas import validators, scopes, resolver, normalize, filters
+from .schemas import validators, scopes, normalize, filters
 
 
 class ValidatorAdaptor(object):
 
     def __init__(self, schema):
-        self.validator = Draft4Validator(schema, resolver=resolver)
+        self.validator = Draft4Validator(schema)
 
     def validate_number(self, type_, value):
         try:
@@ -66,7 +66,7 @@ class ValidatorAdaptor(object):
     def validate(self, value):
         value = self.type_convert(value)
         errors = list(e.message for e in self.validator.iter_errors(value))
-        return normalize(self.validator.schema, value, resolver=resolver)[0], errors
+        return normalize(self.validator.schema, value)[0], errors
 
 def request_validate(obj):
     def _request_validate(view):
@@ -134,10 +134,10 @@ def response_filter(obj):
                     raise tornado.web.HTTPError(
                         500, message='`%d` is not a defined status code.' % status)
 
-                resp, errors = normalize(schemas['schema'], resp, resolver=resolver)
+                resp, errors = normalize(schemas['schema'], resp)
                 if schemas['headers']:
                     headers, header_errors = normalize(
-                        {'properties': schemas['headers']}, headers, resolver=resolver)
+                        {'properties': schemas['headers']}, headers)
                     errors.extend(header_errors)
                 if errors:
                     raise tornado.web.HTTPError(
@@ -167,4 +167,4 @@ def unpack(value):
     except ValueError:
         pass
 
-    return value, 200, {}
+    return value, 200, {}

tests/test_jsonschema.py

@@ -1,13 +1,13 @@
 from __future__ import absolute_import
-from swagger_py_codegen.parser import Swagger, RefNode
+from swagger_py_codegen.parser import Swagger
 from swagger_py_codegen.jsonschema import build_data
 
 
 def test_schema_base_01():
     data = {}
     swagger = Swagger(data)
     data = build_data(swagger)
-    assert len(data['definitions']['definitions']) == 0
+    assert len(data['schemas']) == 0
 
 
 def test_schema_base_02():
@@ -23,7 +23,7 @@ def test_schema_base_02():
     }
     swagger = Swagger(data)
     data = build_data(swagger)
-    assert len(data['definitions']['definitions']) == 1
+    assert len(data['schemas']) == 1
 
 
 def test_schema_base_03():
@@ -41,7 +41,7 @@ def test_schema_base_03():
     }
     swagger = Swagger(data)
     data = build_data(swagger)
-    assert len(data['definitions']['definitions']) == 0
+    assert len(data['schemas']) == 0
 
 
 def test_schema_ref_01():
@@ -66,8 +66,8 @@ def test_schema_ref_01():
     }
     swagger = Swagger(data)
     data = build_data(swagger)
-    print("!!!!!!!!!!!!!!!!!!", data['definitions']['definitions'])
-    assert len(data['definitions']['definitions']) == 2
+    assert len(data['schemas']) == 2
+    assert list(data['schemas'].keys())[0] == 'DefinitionsUser'
 
 
 def test_validators():
@@ -116,13 +116,13 @@ def test_validators():
     }
     swagger = Swagger(data)
     data = build_data(swagger)
-    schemas = data['definitions']['definitions']
+    schemas = data['schemas']
     validators = data['validators']
 
     # body parameters
     assert ('/users', 'POST') in validators
     v1 = validators[('/users', 'POST')]['body']
-    assert(v1 == RefNode(schemas['User'], '#/definitions/User'))
+    assert v1 == schemas['DefinitionsUser']
 
     # query parameters
     v2 = validators[('/users', 'POST')]['query']
@@ -136,8 +136,8 @@ def test_validators():
     assert 'path' not in validators[('/users', 'POST')]
 
     # definitions
-    assert 'User' in schemas
-    assert 'Product' in schemas
+    assert 'DefinitionsUser' in schemas
+    assert 'DefinitionsProduct' in schemas
 
     assert len(schemas) == 2
 
@@ -192,7 +192,7 @@ def test_filters():
     }
     swagger = Swagger(data)
     data = build_data(swagger)
-    definitions = data['definitions']['definitions']
+    schemas = data['schemas']
     filters = data['filters']
 
     assert 201 in filters[('/users', 'POST')]
@@ -201,7 +201,7 @@ def test_filters():
     r1 = filters[('/users', 'POST')][201]
     r2 = filters[('/users', 'POST')][422]
 
-    assert r1['schema'] == RefNode(definitions['User'], '#/definitions/User')
+    assert r1['schema'] == schemas['DefinitionsUser']
     assert r2['schema']['properties']['code'] == {'type': 'string'}