Add support for `$ref` URIs containing fragments in OpenAPI definitions and JSON schemas (#1751)
Signed-off-by: Kévin Commaille <zecakeh@tedomum.fr>pull/1745/head
Kévin Commaille
8 months ago
committed by
GitHub
parent
0b43b5a343
commit
4d7e33ec26
@ -0,0 +1 @@
|
||||
Add support for `$ref` URIs containing fragments in OpenAPI definitions and JSON schemas.
|
||||
@ -0,0 +1,87 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
# Helpers to resolve $ref recursively in OpenAPI and JSON schemas.
|
||||
|
||||
# Copyright 2016 OpenMarket Ltd
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import json
|
||||
import os
|
||||
import os.path
|
||||
import urllib.parse
|
||||
import yaml
|
||||
|
||||
def resolve_references(path, schema):
|
||||
"""Recurse through a given schema until we find a $ref key. Upon doing so,
|
||||
check that the referenced file exists, then load it up and check all of the
|
||||
references in that file. Continue on until we've hit all dead ends.
|
||||
|
||||
$ref values are deleted from schemas as they are validated, to prevent
|
||||
duplicate work.
|
||||
"""
|
||||
if isinstance(schema, dict):
|
||||
# do $ref first
|
||||
if '$ref' in schema:
|
||||
# Pull the referenced URI from the schema
|
||||
ref_uri = schema['$ref']
|
||||
|
||||
# Join the referenced URI with the URI of the file, to resolve
|
||||
# relative URIs
|
||||
full_ref_uri = urllib.parse.urljoin("file://" + path, ref_uri)
|
||||
|
||||
# Separate the fragment.
|
||||
(full_ref_uri, fragment) = urllib.parse.urldefrag(full_ref_uri)
|
||||
|
||||
# Load the referenced file
|
||||
ref = load_file_from_uri(full_ref_uri)
|
||||
|
||||
if fragment:
|
||||
# The fragment should be a JSON Pointer
|
||||
keys = fragment.strip('/').split('/')
|
||||
for key in keys:
|
||||
ref = ref[key]
|
||||
|
||||
# Check that the references in *this* file are valid
|
||||
result = resolve_references(urllib.parse.urlsplit(full_ref_uri).path, ref)
|
||||
|
||||
# They were valid, and so were the sub-references. Delete
|
||||
# the reference here to ensure we don't pass over it again
|
||||
# when checking other files
|
||||
del schema['$ref']
|
||||
else:
|
||||
result = {}
|
||||
|
||||
for key, value in schema.items():
|
||||
result[key] = resolve_references(path, value)
|
||||
return result
|
||||
elif isinstance(schema, list):
|
||||
return [resolve_references(path, value) for value in schema]
|
||||
else:
|
||||
return schema
|
||||
|
||||
|
||||
def load_file_from_uri(path):
|
||||
"""Load a JSON or YAML file from a file:// URI.
|
||||
"""
|
||||
print("Loading reference: %s" % path)
|
||||
if not path.startswith("file://"):
|
||||
raise Exception("Bad ref: %s" % (path,))
|
||||
path = path[len("file://"):]
|
||||
with open(path, "r") as f:
|
||||
if path.endswith(".json"):
|
||||
return json.load(f)
|
||||
else:
|
||||
# We have to assume it's YAML because some of the YAML examples
|
||||
# do not have file extensions.
|
||||
return yaml.safe_load(f)
|
||||
Loading…
Reference in New Issue