python codegen shortname() plus test

Author: mr-c

schema_salad/metaschema.py

@@ -22,7 +22,7 @@
     Type,
     Union,
 )
-from urllib.parse import quote, urlsplit, urlunsplit
+from urllib.parse import quote, urlsplit, urlunsplit, urlparse
 from urllib.request import pathname2url
 
 from ruamel.yaml.comments import CommentedMap
@@ -674,6 +674,18 @@ def save_relative_uri(
         return save(uri, top=False, base_url=base_url)
 
 
+def shortname(inputid: str) -> str:
+    """
+    Compute the shortname of a fully qualified identifer.
+
+    See https://w3id.org/cwl/v1.2/SchemaSalad.html#Short_names.
+    """
+    parsed_id = urlparse(inputid)
+    if parsed_id.fragment:
+        return parsed_id.fragment.split("/")[-1]
+    return parsed_id.path.split("/")[-1]
+
+
 def parser_info() -> str:
     return "org.w3id.cwl.salad"
 

schema_salad/python_codegen_support.py

@@ -19,7 +19,7 @@
     Type,
     Union,
 )
-from urllib.parse import quote, urlsplit, urlunsplit
+from urllib.parse import quote, urlsplit, urlunsplit, urlparse
 from urllib.request import pathname2url
 
 from ruamel.yaml.comments import CommentedMap
@@ -669,3 +669,15 @@ def save_relative_uri(
         return uri
     else:
         return save(uri, top=False, base_url=base_url)
+
+
+def shortname(inputid: str) -> str:
+    """
+    Compute the shortname of a fully qualified identifer.
+
+    See https://w3id.org/cwl/v1.2/SchemaSalad.html#Short_names.
+    """
+    parsed_id = urlparse(inputid)
+    if parsed_id.fragment:
+        return parsed_id.fragment.split("/")[-1]
+    return parsed_id.path.split("/")[-1]

schema_salad/tests/test_cg.py

@@ -177,6 +177,16 @@ def test_load_pt() -> None:
     ] == doc.symbols
 
 
+def test_shortname() -> None:
+    """Test shortname() function."""
+    assert cg_metaschema.shortname("http://example.com/foo") == "foo"
+    assert cg_metaschema.shortname("http://example.com/#bar") == "bar"
+    assert cg_metaschema.shortname("http://example.com/foo/bar") == "bar"
+    assert cg_metaschema.shortname("http://example.com/foo#bar") == "bar"
+    assert cg_metaschema.shortname("http://example.com/#foo/bar") == "bar"
+    assert cg_metaschema.shortname("http://example.com/foo#bar/baz") == "baz"
+
+
 @pytest.fixture
 def metaschema_pre() -> Any:
     """Prep-parsed schema for testing."""