import re
from typing import Callable, List, Match, Optional, Text, Tuple, Union
from linkml_runtime.linkml_model.meta import ClassDefinition, SlotDefinition
from linkml_runtime.utils.formatutils import sfx
from linkml.utils.schemaloader import SchemaLoader
[docs]def strval(txt: str) -> str:
txt = str(txt).replace('"', '\\"')
return f'"{txt}"'
[docs]def default_uri_for(loader: SchemaLoader) -> str:
dflt = loader.schema.default_prefix if loader.schema.default_prefix else sfx(loader.schema.id)
return sfx(loader.namespaces.uri_for(dflt))
[docs]def default_curie_or_uri(loader: SchemaLoader) -> str:
dflt = loader.schema.default_prefix if loader.schema.default_prefix else sfx(loader.schema.id)
if ":/" in dflt:
prefix = loader.namespaces.prefix_for(loader.schema.default_prefix)
if prefix:
dflt = prefix
return dflt
[docs]def curie_for(loader: SchemaLoader, is_class: bool) -> Optional[str]:
"""Return the Curie for the schema in loader. Return None if there is no curie form"""
prefix = default_curie_or_uri(loader)
suffix = "camelcase(self.name)" if is_class else "underscore(self.alias if self.alias else self.name)"
if ":/" not in prefix:
return '"' + prefix + ":" + '" + ' + suffix
else:
pn = loader.namespaces.curie_for(prefix, default_ok=False)
return ('"' + pn + '" + ' + suffix) if pn else None
[docs]def uri_for(s: str, loader: SchemaLoader) -> str:
uri = str(loader.namespaces.uri_for(s))
return loader.namespaces.curie_for(uri, True, True) or strval(uri)
[docs]def default_ns_for(loader: SchemaLoader, cls: ClassDefinition) -> str:
"""Return code to produce the default namespace for the supplied class"""
# TODO: figure out how to mark a slot as a namespace
return "sfx(str(self.id))" if "id" in cls.slots else "None"
# cls_id = None
# for slotname in cls.slots:
# slot = loader.schema.slots[slotname]
# if slot.identifier:
# cls_id = slotname
# return f"sfx(str(self.{cls_id}))" if cls_id else "None"
# Library of named default values -- this is here to prevent code injection
# Contents: Match text (as re),
# flag that indicates whether we're generating a default value expression or postinig code
# Function that takes the match string, SchemaLoader, ClassDefinition, and SlotDefinition and returns the
# appropriate string
default_library: List[
Tuple[
Text,
bool,
Callable[[Match[str], SchemaLoader, ClassDefinition, SlotDefinition], str],
]
] = [
(r"[Tt]rue", False, lambda _, __, ___, ____: "True"),
(r"[Ff]alse", False, lambda _, __, ___, ____: "False"),
(r"int\(([-+]?[0-9]+)\)", False, lambda m, __, ___, ____: int(m[1])),
(
r"float\(([-+]?[0-9]*\.?[0-9]+([eE][-+]?[0-9]+)?)\)",
False,
lambda m, __, ___, ____: float(m[1]),
),
(
r"date\((\d{4})-(\d{2})-(\d{2})\)",
False,
lambda m, __, ___, ____: f"date({int(m[1])}, {int(m[2])}, {int(m[3])})",
),
(
r"datetime\((\d{4})-(\d{2})-(\d{2})T(\d{2}):(\d{2}):(\d{2})Z\)",
False,
lambda m, __, ___, ____: f"datetime({int(m[1])}, {int(m[2])}, {int(m[3])}, "
f"{int(m[4])}, {int(m[5])}, {int(m[6])})",
),
# TODO: We have to make the real URI available before any of these can work
# ("class_uri", True,
# lambda _, loader, ___, ____: f'"{default_uri_for(loader)}" + camelcase(self.name)'),
# ("slot_uri", True,
# lambda _, loader, ___, ____: f'"{default_uri_for(loader)}" +
# underscore(self.alias if self.alias else self.name)'),
# ("class_curie", True, lambda _, loader, ___, ____: curie_for(loader, True)),
# ("slot_curie", True, lambda _, loader, ___, ____: curie_for(loader, False)),
("class_uri", True, lambda _, loader, ___, ____: "None"),
("slot_uri", True, lambda _, loader, ___, ____: "None"),
("class_curie", True, lambda _, loader, ___, ____: "None"),
("slot_curie", True, lambda _, loader, ___, ____: "None"),
# See: https://github.com/linkml/linkml/issues/1333
# ("class_uri", False, lambda _, __, class_definition, ____: 'class_class_uri'),
# ("class_curie", False, lambda _, __, class_definition, ____: 'class_class_curie'),
# ("slot_uri", True, lambda _, loader, ___, slot_definition: f'slots.{slot_definition.name}.uri'),
# ("slot_curie", True, lambda _, loader, ___, slot_definition: f'slots.{slot_definition.name}.curie'),
# ("default_range", False, lambda _, loader, __, ____: f"{strval(loader.schema.default_range)}"),
("default_range", False, lambda _, __, ___, ____: "None"),
("bnode", False, lambda _, __, ___, ____: "bnode()"),
(r"string\((.*)\)", False, lambda m, __, ___, ____: strval(m[1])),
(r"uri\((.*)\)", False, lambda m, loader, _, __: uri_for(m[1], loader)),
("default_ns", True, lambda _, loader, cls, ____: default_ns_for(loader, cls)),
# ("default_ns", False, lambda _, loader, __, ____: f"{strval(loader.schema.default_prefix)}"),
]
[docs]def isabsent_match(
txt: Text,
) -> Union[
Tuple[Match[str], bool, Callable[[Match[str], SchemaLoader, ClassDefinition, SlotDefinition], str]],
Tuple[None, None, None],
]:
txt = str(txt)
for pattern, postinit, f in default_library:
m = re.match(pattern + "$", txt)
if m:
return m, postinit, f
raise ValueError(f"Incompatible default value: {txt}")
[docs]def ifabsent_value_declaration(txt: Text, loader, cls, slot) -> Optional[str]:
m, postinit, f = isabsent_match(txt)
if m and not postinit:
return f(m, loader, cls, slot)
[docs]def ifabsent_postinit_declaration(txt: Text, loader, cls, slot) -> Optional[str]:
m, postinit, f = isabsent_match(txt)
if m and postinit:
return f(m, loader, cls, slot)