stbkit-core 0.1.0a9.dev1__py3-none-any.whl

stbkit/core/_internal/stb_logger.py

@@ -0,0 +1,86 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+import io
+import logging
+import sys
+import time
+from logging import Logger
+from typing import Any, TextIO
+
+
+def get_logger(
+    *,
+    logger: Logger | None = None,
+    file: str | TextIO | None = None,
+    silent: bool = False,
+    show_time: bool = False,
+    show_elapsed: bool = False,
+    name: str = "stb",
+    show_name: bool = False,
+    separator: str = ":",
+    level: int = logging.WARNING,
+) -> Logger:
+    """
+    ロガー、ファイル、標準出力、出力無しのいずれかに対応するLoggerを返す。
+    logger: Loggerオブジェクトを指定（優先）
+    file: ファイルパス(str)またはファイルオブジェクト(io.TextIOBase)を指定
+    silent: Trueなら出力無し（NullHandlerを返す）
+    show_time: Trueで時刻を表示
+    show_elapsed: Trueで経過時間を表示
+    name: ロガー名を指定
+    すべて未指定なら標準出力
+    """
+    log: Logger = logging.getLogger(name)
+    log.handlers.clear()
+    log.propagate = False
+
+    if silent:
+        log.addHandler(logging.NullHandler())
+        return log
+    if logger is not None:
+        return logger
+
+    if file is not None:
+        if isinstance(file, str):
+            handler: logging.Handler = logging.FileHandler(file)
+        elif isinstance(file, io.TextIOBase):
+            handler = logging.StreamHandler(file)
+        else:
+            raise TypeError("fileはstrまたはio.TextIOBaseで指定してください")
+    else:
+        handler = logging.StreamHandler(sys.stderr)
+
+    # フォーマットの組み立て
+    fmt_parts: list[str] = []
+    if show_time:
+        fmt_parts.append("%(asctime)s")
+    if show_elapsed:
+        fmt_parts.append("%(elapsed)s")
+    fmt_parts.append("%(levelname).1s")
+    if show_name:
+        fmt_parts.append("%(name)s")
+
+    fmt_parts.append("%(message)s")
+    fmt: str = separator.join(fmt_parts)
+
+    class ElapsedFormatter(logging.Formatter):
+        def __init__(self, fmt: str):
+            super().__init__(fmt)
+            self.start_time = time.time()
+
+        def format(self, record: Any) -> str:
+            if show_elapsed:
+                elapsed = time.time() - self.start_time
+                record.elapsed = f"{elapsed:.2f}s"
+            else:
+                record.elapsed = ""
+            return super().format(record)
+
+    handler.setFormatter(ElapsedFormatter(fmt))
+    log.addHandler(handler)
+    log.setLevel(level=level)
+    return log

stbkit/core/_internal/transform/__init__.py

@@ -0,0 +1,5 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.

stbkit/core/_internal/transform/guid_tools.py

@@ -0,0 +1,73 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+import uuid
+from collections.abc import Callable
+from enum import Enum
+from uuid import UUID
+
+from ...data_model.common import StBridgeElement, _FieldKind
+from ..name_converter import private_field_name
+
+
+class UuidGenerateType(Enum):
+    AUTO = "auto"
+    UUID4 = "uuid4"
+    UUID7 = "uuid7"
+
+
+_uuid7: Callable[[], UUID] | None = getattr(uuid, "uuid7", None)
+
+
+def generate_uuid(generate_type: UuidGenerateType = UuidGenerateType.AUTO) -> UUID:
+    match generate_type:
+        case UuidGenerateType.AUTO:
+            if callable(_uuid7):
+                return _uuid7()
+            return uuid.uuid4()
+        case UuidGenerateType.UUID7:
+            if callable(_uuid7):
+                return _uuid7()
+            raise RuntimeError("UUID7の利用はPython 3.14以降である必要があります")
+        case UuidGenerateType.UUID4:
+            return uuid.uuid4()
+
+
+def assign_guid_all(stb_element: StBridgeElement) -> None:
+    """子要素も含めたすべての要素にguidを割り当てる。"""
+    for field_key, field_info in stb_element._fields.items():
+        p_name: str = private_field_name(field_key)
+        if field_info.py_type == UUID or field_info.py_type == "UUID":
+            current_guid = getattr(stb_element, p_name)
+            if current_guid is None:
+                setattr(stb_element, p_name, uuid.uuid4())
+        elif field_info.kind == _FieldKind.ELEMENT:
+            child = getattr(stb_element, p_name)
+            if isinstance(child, list):
+                for c in child:
+                    if isinstance(c, StBridgeElement):
+                        assign_guid_all(c)
+            elif isinstance(child, StBridgeElement):
+                assign_guid_all(child)
+
+
+def is_guid_assigned_all(stb_element: StBridgeElement) -> bool:
+    """全要素にguidが割り当てられているかを確認する。"""
+    for field_key, field_info in stb_element._fields.items():
+        p_name: str = private_field_name(field_key)
+        if field_info.py_type == UUID or field_info.py_type == "UUID":
+            current_guid = getattr(stb_element, p_name)
+            if current_guid is None:
+                return False
+        elif field_info.kind == _FieldKind.ELEMENT:
+            child = getattr(stb_element, p_name)
+            if isinstance(child, list):
+                for c in child:
+                    if isinstance(c, StBridgeElement) and not is_guid_assigned_all(c):
+                        return False
+            elif isinstance(child, StBridgeElement) and not is_guid_assigned_all(child):
+                return False
+    return True

stbkit/core/_internal/transform/select.py

@@ -0,0 +1,114 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+"""要素の抜粋を行う処理。
+
+TODO:モジュール名や関数名を検討中。filterやextract等。
+    また、inputを編集するか、コピーしてreturnするか
+"""
+
+from ...data_model._internal.filter_option import FilterOption, NameFilter
+from ...data_model.common import (
+    StBridgeElement,
+    StBridgeRoot,
+    _FieldKind,
+    _StBridgeElementList,
+)
+from ...stb_reporting import Reporter
+from ..name_converter import private_field_name
+
+
+def select_elements(
+    stb: StBridgeRoot,
+    option: FilterOption,
+    *,
+    reporter: Reporter,
+) -> None:
+    """SelectOptionで指定した要素のみが残るようにStBridgeRootを編集する。"""
+
+    def _traverse(element: StBridgeElement) -> bool:
+        """再帰的に要素をたどり、選択オプションに基づいて要素や属性を削除する。
+        RETURN: 要素がinclude指定で残る場合True、そうでない場合False"""
+        has_include_element: bool = False
+        for key, field in element._fields.items():
+            child_has_include_element: bool = False
+            # 要素の場合
+            if field.kind == _FieldKind.ELEMENT:
+                value = getattr(element, private_field_name(key), None)
+                if not value:
+                    continue
+                if isinstance(value, list) and len(value) > 0:
+                    xml_name: str = value[0]._xml_name()
+                    for item in value:
+                        if isinstance(item, StBridgeElement):
+                            child_has_include_element = (
+                                child_has_include_element or _traverse(item)
+                            )
+                elif isinstance(value, StBridgeElement):
+                    xml_name = value._xml_name()
+                    child_has_include_element = child_has_include_element or _traverse(
+                        value
+                    )
+                else:
+                    continue
+                # include指定がある場合はそれ以外は除外
+                if child_has_include_element or (
+                    option.global_element_filter.include
+                    and xml_name in option.global_element_filter.include
+                ):
+                    has_include_element = True
+                elif (
+                    option.global_element_filter.include
+                    and xml_name not in option.global_element_filter.include
+                ):
+                    if field.max_occurs == 1:
+                        setattr(element, private_field_name(key), None)
+                    else:
+                        setattr(
+                            element,
+                            private_field_name(key),
+                            _StBridgeElementList(parent=element),
+                        )
+                    continue
+                # exclude指定がある場合は除外
+                if xml_name in option.global_element_filter.exclude:
+                    if field.max_occurs == 1:
+                        setattr(element, private_field_name(key), None)
+                    else:
+                        setattr(
+                            element,
+                            private_field_name(key),
+                            _StBridgeElementList(parent=element),
+                        )
+                    continue
+
+            # 属性の場合
+            elif field.kind == _FieldKind.ATTRIBUTE:
+                xml_name = element._xml_name(key)
+                # 属性抽出オプション
+                attrs_opt: NameFilter = option.element_attribute_filters.get(
+                    element._xml_name(), NameFilter()
+                )
+                if attrs_opt and xml_name in attrs_opt.exclude:
+                    setattr(element, private_field_name(key), None)
+                    continue
+                if (
+                    attrs_opt
+                    and attrs_opt.include
+                    and xml_name not in attrs_opt.include
+                ):
+                    setattr(element, private_field_name(key), None)
+                    continue
+                # グローバル除外
+                if (
+                    option.global_attribute_filter
+                    and xml_name in option.global_attribute_filter.exclude
+                ):
+                    setattr(element, private_field_name(key), None)
+                    continue
+        return has_include_element
+
+    _traverse(stb)

stbkit/core/_internal/xml_value_converter.py

@@ -0,0 +1,16 @@
+# Copyright 2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from typing import Any
+from uuid import UUID
+
+
+def any_to_xml_str(value: Any) -> str:
+    if isinstance(value, UUID):
+        return value.hex
+    if isinstance(value, bool):
+        return "true" if value else "false"
+    return str(value)

stbkit/core/data_model/__init__.py

@@ -0,0 +1,5 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.

stbkit/core/data_model/_internal/filter_option.py

@@ -0,0 +1,25 @@
+# Copyright 2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from dataclasses import dataclass, field
+
+
+@dataclass
+class NameFilter:
+    """
+    include, excludeのフィルター。同時指定時はexclude優先
+    """
+
+    include: list[str] = field(default_factory=list)
+    exclude: list[str] = field(default_factory=list)
+
+
+@dataclass
+class FilterOption:
+    global_attribute_filter: NameFilter = field(default_factory=NameFilter)
+    global_element_filter: NameFilter = field(default_factory=NameFilter)
+    element_attribute_filters: dict[str, NameFilter] = field(default_factory=dict)
+    element_child_filters: dict[str, NameFilter] = field(default_factory=dict)

stbkit/core/data_model/_internal/stb_types.py

@@ -0,0 +1,98 @@
+# Copyright 2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+import uuid
+from enum import Enum, IntEnum, StrEnum
+
+from ...stb_typing import (
+    Angle,
+    Length,
+    Monolist,
+    MonolistId,
+    MonolistLength,
+    NonNegativeInteger,
+    NonNegativeLength,
+    PositiveInteger,
+    PositiveIntegerList,
+    Ratio,
+)
+
+
+class DataType(Enum):
+    ANGLE = "Angle"
+    BOOL = "bool"
+    FLOAT = "float"
+    INT = "int"
+    INT_ENUM = "IntEnum"
+    LENGTH = "Length"
+    MONOLIST = "Monolist"
+    MONOLIST_LENGTH = "MonolistLength"
+    MONOLIST_ID = "MonolistId"
+    NON_NEGATIVE_INTEGER = "NonNegativeInteger"
+    NON_NEGATIVE_LENGTH = "NonNegativeLength"
+    POSITIVE_INTEGER = "PositiveInteger"
+    POSITIVE_INTEGER_LIST = "PositiveIntegerList"
+    RATIO = "Ratio"
+    STR = "str"
+    STR_ENUM = "StrEnum"
+    UUID = "UUID"
+
+    def to_type(
+        self,
+    ) -> type[
+        Angle
+        | bool
+        | float
+        | int
+        | IntEnum
+        | Length
+        | Monolist
+        | MonolistLength
+        | MonolistId
+        | NonNegativeInteger
+        | NonNegativeLength
+        | PositiveInteger
+        | PositiveIntegerList
+        | Ratio
+        | str
+        | StrEnum
+        | uuid.UUID
+    ]:
+        match self:
+            case DataType.ANGLE:
+                return Angle
+            case DataType.BOOL:
+                return bool
+            case DataType.FLOAT:
+                return float
+            case DataType.INT:
+                return int
+            case DataType.INT_ENUM:
+                return IntEnum
+            case DataType.LENGTH:
+                return Length
+            case DataType.MONOLIST:
+                return Monolist
+            case DataType.MONOLIST_LENGTH:
+                return MonolistLength
+            case DataType.MONOLIST_ID:
+                return MonolistId
+            case DataType.NON_NEGATIVE_INTEGER:
+                return NonNegativeInteger
+            case DataType.NON_NEGATIVE_LENGTH:
+                return NonNegativeLength
+            case DataType.POSITIVE_INTEGER:
+                return PositiveInteger
+            case DataType.POSITIVE_INTEGER_LIST:
+                return PositiveIntegerList
+            case DataType.RATIO:
+                return Ratio
+            case DataType.STR:
+                return str
+            case DataType.STR_ENUM:
+                return StrEnum
+            case DataType.UUID:
+                return uuid.UUID