Source code for policyengine_core.entities.helpers

from typing import List, Union

from policyengine_core import entities
from policyengine_core.entities.entity import Entity


def build_entity(
    key: str,
    plural: str,
    label: str,
    doc: str = "",
    roles: List = None,
    is_person: bool = False,
    containing_entities: List[str] = (),
) -> Entity:
    if is_person:
        return entities.Entity(key, plural, label, doc)
    else:
        return entities.GroupEntity(
            key,
            plural,
            label,
            doc,
            roles,
            containing_entities=containing_entities,
        )