2.1.2 - Some TypeSystem methods

.github/workflows/python-package.yml

@@ -70,7 +70,7 @@ jobs:
 
       - run: |
           mk python-release owner=vkottler \
-                            repo=runtimepy version=2.1.1
+                            repo=runtimepy version=2.1.2
         if: |
           matrix.python-version == '3.11'
           && matrix.system == 'ubuntu-latest'

README.md

@@ -2,11 +2,11 @@
     =====================================
     generator=datazen
     version=3.1.2
-    hash=3fefcc8e6720ba32defa794a99a02d02
+    hash=64a758df8095e218dce93935454221a7
     =====================================
 -->
 
-# runtimepy ([2.1.1](https://pypi.org/project/runtimepy/))
+# runtimepy ([2.1.2](https://pypi.org/project/runtimepy/))
 
 [![python](https://img.shields.io/pypi/pyversions/runtimepy.svg)](https://pypi.org/project/runtimepy/)
 ![Build Status](https://github.com/vkottler/runtimepy/workflows/Python%20Package/badge.svg)

local/variables/package.yaml

@@ -1,5 +1,5 @@
 ---
 major: 2
 minor: 1
-patch: 1
+patch: 2
 entry: runtimepy

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta:__legacy__"
 
 [project]
 name = "runtimepy"
-version = "2.1.1"
+version = "2.1.2"
 description = "A framework for implementing Python services."
 readme = "README.md"
 requires-python = ">=3.8"

runtimepy/__init__.py

@@ -1,7 +1,7 @@
 # =====================================
 # generator=datazen
 # version=3.1.2
-# hash=48f84a8c64666c2d207321d75e066401
+# hash=f83181278a252952c4fdf15493634816
 # =====================================
 
 """
@@ -10,4 +10,4 @@
 
 DESCRIPTION = "A framework for implementing Python services."
 PKG_NAME = "runtimepy"
-VERSION = "2.1.1"
+VERSION = "2.1.2"

runtimepy/codec/system/__init__.py

@@ -11,6 +11,7 @@
 # internal
 from runtimepy import PKG_NAME
 from runtimepy.codec.protocol import Protocol
+from runtimepy.enum import RuntimeEnum
 from runtimepy.enum.registry import (
     DEFAULT_ENUM_PRIMITIVE,
     EnumRegistry,
@@ -45,6 +46,13 @@ def __init__(self, *namespace: str) -> None:
 
         self.root_namespace = global_namespace.child(*namespace)
 
+    def get_enum(self, name: str, *namespace: str) -> RuntimeEnum:
+        """Lookup an enum type at runtime."""
+
+        found = self._find_name(name, *namespace, strict=True)
+        assert found is not None
+        return self._enums[found]
+
     def runtime_int_enum(self, enum: Type[RuntimeIntEnum]) -> None:
         """Register an enumeration class."""
 
@@ -76,6 +84,13 @@ def register(self, name: str, *namespace: str) -> Protocol:
         ] = new_type
         return new_type
 
+    def get_protocol(self, name: str, *namespace: str) -> Protocol:
+        """Get a custom protocol by name."""
+
+        found = self._find_name(name, *namespace, strict=True)
+        assert found is not None
+        return self.custom[found]
+
     def add(self, custom_type: str, field_name: str, field_type: str) -> None:
         """Add a field to a custom type."""
 

tests/codec/system/test_system.py

@@ -21,6 +21,8 @@ def get_test_system() -> TypeSystem:
 
     assert system.size("SampleStruct") == 7
 
+    assert system.get_enum("TestEnum2")
+
     return system
 
 
@@ -42,6 +44,8 @@ def test_type_system_compound_types():
 
     assert system.size("SomeStruct") == 16
 
+    assert system.get_protocol("SomeStruct")
+
 
 def test_type_system_basic():
     """Test basic interactions with a custom type system."""