Eterna2/fix relative ref

.gitignore

@@ -2,4 +2,9 @@ _cache
 _static
 _templates
 
+.eggs
+.tox
+
+*.py[cod]
+
 TODO

jsonschema/tests/test_validators.py

@@ -1664,6 +1664,39 @@ def test_it_retrieves_local_refs_via_urlopen(self):
         with self.resolver.resolving(ref) as resolved:
             self.assertEqual(resolved, "bar")
 
+    def test_it_retrieves_local_relative_refs_via_open(self):
+        with tempfile.NamedTemporaryFile(delete=False, mode="wt") as tempf:
+            self.addCleanup(os.remove, tempf.name)
+            json.dump({"foo": "bar"}, tempf)
+
+        ref = "{}#foo".format(os.path.relpath(tempf.name))
+        with self.resolver.resolving(ref) as resolved:
+            self.assertEqual(resolved, "bar")
+
+    def test_it_retrieves_relative_refs_via_urlopen(self):
+        ref = "fuzz.json#baz"
+        schema = {"baz": 12}
+        resolver = validators.RefResolver(
+            "http://foo.com/bar.json", self.referrer, self.store,
+        )
+        if "requests" in sys.modules:
+            self.addCleanup(
+                sys.modules.__setitem__, "requests", sys.modules["requests"],
+            )
+        sys.modules["requests"] = None
+
+        @contextmanager
+        def fake_urlopen(url):
+            self.assertEqual(url, "http://foo.com/fuzz.json")
+            yield BytesIO(json.dumps(schema).encode("utf8"))
+
+        self.addCleanup(setattr, validators, "urlopen", validators.urlopen)
+        validators.urlopen = fake_urlopen
+
+        with resolver.resolving(ref) as resolved:
+            pass
+        self.assertEqual(resolved, 12)
+
     def test_it_can_construct_a_base_uri_from_a_schema(self):
         schema = {"id": "foo"}
         resolver = validators.RefResolver.from_schema(

jsonschema/validators.py

@@ -4,6 +4,7 @@
 import contextlib
 import json
 import numbers
+import os
 
 from six import add_metaclass
 
@@ -25,6 +26,7 @@
     urljoin,
     urlopen,
     urlsplit,
+    urlunsplit,
 )
 
 # Sigh. https://gitlab.com/pycqa/flake8/issues/280
@@ -826,9 +828,22 @@ def resolve_remote(self, uri):
             # json over http
             result = requests.get(uri).json()
         else:
-            # Otherwise, pass off to urllib and assume utf-8
-            with urlopen(uri) as url:
-                result = json.loads(url.read().decode("utf-8"))
+            try:
+                # Otherwise, pass off to urllib and assume utf-8
+                with urlopen(uri) as url:
+                    result = json.loads(url.read().decode("utf-8"))
+            except Exception:
+                try:
+                    # try relative path on local fs
+                    with open(os.path.join(self.base_uri, uri)) as fin:
+                        result = json.load(fin)
+                except IOError:
+                    scheme, network, path, qs, frag = urlsplit(self.base_uri)
+                    path = "/".join(path.split("/")[0:-1])  # base path
+                    base_url = urlunsplit((scheme, network, path, qs, frag))
+                    # point to relative url
+                    with urlopen(urljoin(base_url, uri)) as url:
+                        result = json.loads(url.read().decode("utf-8"))
 
         if self.cache_remote:
             self.store[uri] = result