utils.py

from datetime import datetime, timezone
from enum import Enum

STRING_SEPARATOR = " | "


class InlineList(list):
    """Class wrapper to indicate that the list should be written as one line in YAML.
    Used for 'bbox' property."""

    pass


def is_valid_string(text):
    if len(str(text).replace(" ", "")) >= 1:
        return True
    return False


def get_enum_value_from_string(enum_type: Enum, text: str):
    if isinstance(enum_type, type) and issubclass(enum_type, Enum):
        for member in enum_type:
            if text == member.value:
                return member
        # handle case with non-string enums
        for member in enum_type:
            if text == str(member.value) or (text == "" and member.value is None):
                return member

    raise ValueError(f"Unexpected value '{text}', expected type: '{enum_type}'")


def bbox_from_list(raw_bbox_list: list):

    # this loop is to not add empty decimals unnecessarily
    list_bbox_val = []
    for part in raw_bbox_list:
        if isinstance(part, str):  # if the list is read from UI widgets
            part = part.strip()
            if "." in part:
                list_bbox_val.append(float(part))
            else:
                list_bbox_val.append(int(part))
        else:  # if the list is already taken from actual data (int or float)
            list_bbox_val.append(part)

    if len(list_bbox_val) != 4 and len(list_bbox_val) != 6:
        raise ValueError(
            f"Wrong number of values: {len(list_bbox_val)}. Expected: 4 or 6"
        )

    return InlineList(list_bbox_val)