def get_resource_mappings():
"""
Get the resource mapping instances for the astropy schemas
and manifests. This method is registered with the
asdf.resource_mappings entry point.
Returns
-------
list of collections.abc.Mapping
"""
from . import resources
resources_root = importlib_resources.files(resources)
return [
DirectoryResourceMapping(resources_root / "schemas", "http://astropy.org/schemas/astropy", recursive=True),
DirectoryResourceMapping(resources_root / "manifests", "http://astropy.org/asdf/extensions/astropy/manifests"),
]
def get_resource_mappings():
"""
Get the resource mapping instances for the datamodel schemas
and manifests. This method is registered with the
asdf.resource_mappings entry point.
Returns
-------
list of collections.abc.Mapping
"""
from . import resources
resources_root = importlib_resources.files(resources)
return [
DirectoryResourceMapping(resources_root / "schemas", "asdf://stsci.edu/datamodels/roman/schemas/", recursive=True),
DirectoryResourceMapping(resources_root / "manifests", "asdf://stsci.edu/datamodels/roman/manifests/"),
]
def get_resource_mappings():
"""
Get the resource mapping instances for myschemas
and manifests. This method is registered with the
asdf.resource_mappings entry point.
Returns
-------
list of collections.abc.Mapping
"""
from . import resources
resources_root = importlib_resources.files(resources)
return [
DirectoryResourceMapping(resources_root / "schemas",
"asdf://sunpy.org/sunpy/schemas/"),
DirectoryResourceMapping(resources_root / "manifests",
"asdf://sunpy.org/sunpy/manifests/"),
]
def get_schema_resource_mapping() -> DirectoryResourceMapping:
"""Get the weldx schema resource mapping."""
mapping = DirectoryResourceMapping(
SCHEMA_PATH,
WELDX_SCHEMA_URI_BASE,
recursive=True,
filename_pattern="*.yaml",
stem_filename=True,
)
return mapping
def get_extension_resource_mapping() -> DirectoryResourceMapping:
"""Get the weldx manifest resource mapping."""
mapping = DirectoryResourceMapping(
MANIFEST_PATH,
WELDX_EXTENSION_URI_BASE,
recursive=True,
filename_pattern="*.yaml",
stem_filename=True,
)
return mapping
def get_resource_mappings():
resources_root = importlib_resources.files(
asdf_transform_schemas) / "resources"
if not resources_root.is_dir():
# In an editable install, the resources directory will exist off the
# repository root:
resources_root = Path(__file__).parent.parent.parent / "resources"
if not resources_root.is_dir():
raise RuntimeError("Missing resources directory")
return [
DirectoryResourceMapping(
resources_root / "stsci.edu" / "schemas",
"http://stsci.edu/schemas/asdf/transform/",
),
DirectoryResourceMapping(
resources_root / "asdf-format.org" / "manifests",
"asdf://asdf-format.org/transform/manifests/",
),
]
def test_directory_resource_mapping_with_traversable():
"""
Confirm that DirectoryResourceMapping doesn't use pathlib.Path
methods outside of the Traversable interface.
"""
class MockTraversable(importlib_resources.abc.Traversable):
def __init__(self, name, value):
self._name = name
self._value = value
def iterdir(self):
if isinstance(self._value, dict):
for key, child in self._value.items():
yield MockTraversable(key, child)
def read_bytes(self):
if not isinstance(self._value, bytes):
raise RuntimeError("Not a file")
return self._value
def read_text(self, encoding="utf-8"):
return self.read_bytes().decode(encoding)
def is_dir(self):
return isinstance(self._value, dict)
def is_file(self):
return self._value is not None and not isinstance(
self._value, dict)
def joinpath(self, child):
if isinstance(self._value, dict):
child_value = self._value.get(child)
else:
child_value = None
return MockTraversable(child, child_value)
def __truediv__(self, child):
return self.joinpath(child)
def open(self, mode="r", *args, **kwargs):
if not self.is_file():
raise RuntimeError("Not a file")
if mode == "r":
return io.TextIOWrapper(io.BytesIO(self._value), *args,
**kwargs)
elif mode == "rb":
return io.BytesIO(self._value)
else:
raise "Not a valid mode"
@property
def name(self):
return self._name
root = MockTraversable(
"/path/to/some/root", {
"foo-1.0.0.yaml": b"foo",
"bar-1.0.0.yaml": b"bar",
"baz-1.0.0": b"baz",
"nested": {
"foz-1.0.0.yaml": b"foz"
}
})
mapping = DirectoryResourceMapping(root, "http://somewhere.org/schemas")
assert len(mapping) == 2
assert set(mapping) == {
"http://somewhere.org/schemas/foo-1.0.0",
"http://somewhere.org/schemas/bar-1.0.0"
}
assert "http://somewhere.org/schemas/foo-1.0.0" in mapping
assert mapping["http://somewhere.org/schemas/foo-1.0.0"] == b"foo"
assert "http://somewhere.org/schemas/bar-1.0.0" in mapping
assert mapping["http://somewhere.org/schemas/bar-1.0.0"] == b"bar"
assert "http://somewhere.org/schemas/baz-1.0.0" not in mapping
assert "http://somewhere.org/schemas/nested/foz-1.0.0" not in mapping
mapping = DirectoryResourceMapping(root,
"http://somewhere.org/schemas",
recursive=True)
assert len(mapping) == 3
assert set(mapping) == {
"http://somewhere.org/schemas/foo-1.0.0",
"http://somewhere.org/schemas/bar-1.0.0",
"http://somewhere.org/schemas/nested/foz-1.0.0"
}
assert "http://somewhere.org/schemas/foo-1.0.0" in mapping
assert mapping["http://somewhere.org/schemas/foo-1.0.0"] == b"foo"
assert "http://somewhere.org/schemas/bar-1.0.0" in mapping
assert mapping["http://somewhere.org/schemas/bar-1.0.0"] == b"bar"
assert "http://somewhere.org/schemas/baz-1.0.0" not in mapping
assert "http://somewhere.org/schemas/nested/foz-1.0.0" in mapping
assert mapping["http://somewhere.org/schemas/nested/foz-1.0.0"] == b"foz"
mapping = DirectoryResourceMapping(root,
"http://somewhere.org/schemas",
filename_pattern="baz-*",
stem_filename=False)
assert len(mapping) == 1
assert set(mapping) == {"http://somewhere.org/schemas/baz-1.0.0"}
assert "http://somewhere.org/schemas/foo-1.0.0" not in mapping
assert "http://somewhere.org/schemas/bar-1.0.0" not in mapping
assert "http://somewhere.org/schemas/baz-1.0.0" in mapping
assert mapping["http://somewhere.org/schemas/baz-1.0.0"] == b"baz"
assert "http://somewhere.org/schemas/nested/foz-1.0.0" not in mapping
def test_directory_resource_mapping(tmpdir):
tmpdir.mkdir("schemas")
(tmpdir / "schemas").mkdir("nested")
with (tmpdir / "schemas" / "foo-1.2.3.yaml").open("w") as f:
f.write("id: http://somewhere.org/schemas/foo-1.2.3\n")
with (tmpdir / "schemas" / "nested" / "bar-4.5.6.yaml").open("w") as f:
f.write("id: http://somewhere.org/schemas/nested/bar-4.5.6\n")
with (tmpdir / "schemas" / "baz-7.8.9").open("w") as f:
f.write("id: http://somewhere.org/schemas/baz-7.8.9\n")
mapping = DirectoryResourceMapping(str(tmpdir / "schemas"),
"http://somewhere.org/schemas")
assert isinstance(mapping, Mapping)
assert len(mapping) == 1
assert set(mapping) == {"http://somewhere.org/schemas/foo-1.2.3"}
assert "http://somewhere.org/schemas/foo-1.2.3" in mapping
assert b"http://somewhere.org/schemas/foo-1.2.3" in mapping[
"http://somewhere.org/schemas/foo-1.2.3"]
assert "http://somewhere.org/schemas/baz-7.8.9" not in mapping
assert "http://somewhere.org/schemas/baz-7.8" not in mapping
assert "http://somewhere.org/schemas/foo-1.2.3.yaml" not in mapping
assert "http://somewhere.org/schemas/nested/bar-4.5.6" not in mapping
mapping = DirectoryResourceMapping(str(tmpdir / "schemas"),
"http://somewhere.org/schemas",
recursive=True)
assert len(mapping) == 2
assert set(mapping) == {
"http://somewhere.org/schemas/foo-1.2.3",
"http://somewhere.org/schemas/nested/bar-4.5.6"
}
assert "http://somewhere.org/schemas/foo-1.2.3" in mapping
assert b"http://somewhere.org/schemas/foo-1.2.3" in mapping[
"http://somewhere.org/schemas/foo-1.2.3"]
assert "http://somewhere.org/schemas/baz-7.8.9" not in mapping
assert "http://somewhere.org/schemas/baz-7.8" not in mapping
assert "http://somewhere.org/schemas/nested/bar-4.5.6" in mapping
assert b"http://somewhere.org/schemas/nested/bar-4.5.6" in mapping[
"http://somewhere.org/schemas/nested/bar-4.5.6"]
mapping = DirectoryResourceMapping(str(tmpdir / "schemas"),
"http://somewhere.org/schemas",
recursive=True,
filename_pattern="baz-*",
stem_filename=False)
assert len(mapping) == 1
assert set(mapping) == {"http://somewhere.org/schemas/baz-7.8.9"}
assert "http://somewhere.org/schemas/foo-1.2.3" not in mapping
assert "http://somewhere.org/schemas/baz-7.8.9" in mapping
assert b"http://somewhere.org/schemas/baz-7.8.9" in mapping[
"http://somewhere.org/schemas/baz-7.8.9"]
assert "http://somewhere.org/schemas/nested/bar-4.5.6" not in mapping
# Check that the repr is reasonable
# Need to be careful checking the path string because
# pathlib normalizes Windows paths.
assert repr(Path(str(tmpdir / "schemas"))) in repr(mapping)
assert "http://somewhere.org/schemas" in repr(mapping)
assert "recursive=True" in repr(mapping)
assert "filename_pattern='baz-*'" in repr(mapping)
assert "stem_filename=False" in repr(mapping)
# Make sure trailing slash is handled correctly
mapping = DirectoryResourceMapping(str(tmpdir / "schemas"),
"http://somewhere.org/schemas/")
assert len(mapping) == 1
assert set(mapping) == {"http://somewhere.org/schemas/foo-1.2.3"}
assert "http://somewhere.org/schemas/foo-1.2.3" in mapping
assert b"http://somewhere.org/schemas/foo-1.2.3" in mapping[
"http://somewhere.org/schemas/foo-1.2.3"]
