linkml 1.9.4rc2__py3-none-any.whl → 1.9.5rc1__py3-none-any.whl

linkml/generators/rdfgen.py

@@ -67,12 +67,21 @@ class RDFGenerator(Generator):
             base=str(self.namespaces._base),
             prefix=True,
         )
-        out = self._data(graph)
         if output:
+            # Binary-safe when -o/--output is used:
+            # delegate to RDFLib (Graph.serialize(destination=..., format=...)).
+            # Serializers that produce bytes write directly to the file; stdout stays empty.
+            fmt = "turtle" if self.format == "ttl" else self.format
+            try:
+                out = graph.serialize(format=fmt)
+            except UnicodeDecodeError:
+                graph.serialize(destination=output, format=fmt)
+                return ""
             with open(output, "w", encoding="UTF-8") as outf:
                 outf.write(out)
+            return out
 
-        return out
+        return self._data(graph)
 
 
 @shared_arguments(RDFGenerator)

linkml/generators/rustgen/__init__.py

@@ -0,0 +1,3 @@
+from linkml.generators.rustgen.rustgen import RUST_MODES, RustGenerator
+
+__all__ = ["RUST_MODES", "RustGenerator"]

linkml/generators/rustgen/build.py

@@ -0,0 +1,94 @@
+from linkml.generators.common.build import (
+    BuildResult,
+    SchemaResult,
+)
+from linkml.generators.common.build import (
+    ClassResult as ClassResult_,
+)
+from linkml.generators.common.build import EnumResult as EnumResult_
+from linkml.generators.common.build import (
+    SlotResult as SlotResult_,
+)
+from linkml.generators.common.build import TypeResult as TypeResult_
+from linkml.generators.rustgen.template import (
+    Imports,
+    RustCargo,
+    RustEnum,
+    RustFile,
+    RustProperty,
+    RustPyProject,
+    RustStruct,
+    RustTemplateModel,
+    RustTypeAlias,
+)
+
+
+class RustBuildResult(BuildResult):
+    """
+    BuildResult parent class for rustgen
+    """
+
+    imports: Imports = Imports()
+
+    def merge(self, other: "RustBuildResult") -> "RustBuildResult":
+        self.imports += other.imports
+        return self
+
+
+class TypeResult(RustBuildResult, TypeResult_):
+    """
+    A linkml type as a type alias
+    """
+
+    type_: RustTypeAlias
+
+
+class ClassResult(RustBuildResult, ClassResult_):
+    """
+    A single built rust struct
+    """
+
+    cls: RustStruct
+
+
+class SlotResult(RustBuildResult, SlotResult_):
+    """
+    A type alias
+    """
+
+    slot: RustTypeAlias
+
+
+class AttributeResult(RustBuildResult, SlotResult_):
+    """
+    A field within a rust struct
+    """
+
+    attribute: RustProperty
+
+
+class EnumResult(RustBuildResult, EnumResult_):
+    """
+    A rust enum!
+    """
+
+    enum: RustEnum
+
+
+class CrateResult(RustBuildResult, SchemaResult):
+    """
+    A schema built into a rust crate
+    """
+
+    cargo: RustCargo
+    file: RustFile
+    extra_files: dict[str, RustTemplateModel]
+    pyproject: RustPyProject
+
+
+class FileResult(RustBuildResult, SchemaResult):
+    """
+    A schema built into a single rust file
+    """
+
+    file: RustFile

linkml/generators/rustgen/cli.py

@@ -0,0 +1,65 @@
+from pathlib import Path
+from typing import Optional, get_args
+
+import click
+
+from linkml._version import __version__
+from linkml.generators.rustgen import RUST_MODES, RustGenerator
+from linkml.utils.generator import shared_arguments
+
+
+@shared_arguments(RustGenerator)
+@click.option(
+    "-m",
+    "--mode",
+    type=click.Choice([a for a in get_args(RUST_MODES)]),
+    default="crate",
+    help="Generation mode: 'crate' (Cargo package) or 'file' (single .rs)",
+)
+@click.option(
+    "-f",
+    "--force",
+    is_flag=True,
+    help="Overwrite output if it already exists",
+)
+@click.option(
+    "-p",
+    "--pyo3",
+    is_flag=True,
+    help=(
+        "Add 'pyo3' to Cargo.toml default features and emit Python module glue (cdylib + #[pymodule]). "
+        'Source always includes #[cfg(feature="pyo3")] gates; this flag only enables the crate feature by default.'
+    ),
+)
+@click.option(
+    "-s",
+    "--serde",
+    is_flag=True,
+    help=(
+        "Add 'serde' to Cargo.toml default features. Source always includes #[cfg(feature=\"serde\")] derives/attrs; "
+        "this flag only enables the crate feature by default."
+    ),
+)
+@click.option("-n", "--crate-name", type=str, default=None, help="Name of the generated crate/module")
+@click.option(
+    "-o",
+    "--output",
+    type=click.Path(dir_okay=True),
+    help="Output directory (crate mode) or .rs file (file mode)",
+)
+@click.version_option(__version__, "-V", "--version")
+@click.command(name="rust")
+def cli(
+    yamlfile: Path,
+    mode: RUST_MODES = "crate",
+    force: bool = False,
+    pyo3: bool = False,
+    serde: bool = False,
+    crate_name: Optional[str] = None,
+    output: Optional[Path] = None,
+    **kwargs,
+):
+    gen = RustGenerator(yamlfile, mode=mode, pyo3=pyo3, serde=serde, output=output, crate_name=crate_name, **kwargs)
+    serialized = gen.serialize(force=force)
+    if output is None:
+        print(serialized)