Source code for geoh5py.objects.object_type

#  Copyright (c) 2023 Mira Geoscience Ltd.
#  This file is part of geoh5py.
#  geoh5py is free software: you can redistribute it and/or modify
#  it under the terms of the GNU Lesser General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#  geoh5py is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  GNU Lesser General Public License for more details.
#  You should have received a copy of the GNU Lesser General Public License
#  along with geoh5py.  If not, see <>.

from __future__ import annotations

import uuid
from typing import TYPE_CHECKING

from ..shared import EntityType

    from .. import workspace

[docs]class ObjectType(EntityType): """ Object type class """ def __init__(self, workspace: workspace.Workspace, **kwargs): assert workspace is not None super().__init__(workspace, **kwargs) workspace._register_type(self) @staticmethod def _is_abstract() -> bool: return False
[docs] @staticmethod def create_custom(workspace: workspace.Workspace) -> ObjectType: """Creates a new instance of ObjectType for an unlisted custom Object type with a new auto-generated UUID. :param workspace: An active Workspace class """ return ObjectType(workspace)
[docs] @classmethod def find_or_create( cls, workspace: workspace.Workspace, entity_class, **kwargs ) -> ObjectType: """Find or creates an EntityType with given :obj:`uuid.UUID` that matches the given Group implementation class. It is expected to have a single instance of EntityType in the Workspace for each concrete Entity class. :param workspace: An active Workspace class :param entity_class: An Group implementation class. :return: A new instance of GroupType. """ uid = uuid.uuid4() if getattr(entity_class, "default_type_uid", None) is not None: uid = entity_class.default_type_uid() if "ID" in kwargs: kwargs["ID"] = uid else: kwargs["uid"] = uid else: for key, val in kwargs.items(): if key.lower() in ["id", "uid"]: uid = uuid.UUID(val) entity_type = cls.find(workspace, uid) if entity_type is not None: return entity_type return cls(workspace, **kwargs)