import omni.kit.actions.core

def register_actions(extension_id, cls):

    action_registry = omni.kit.actions.core.get_action_registry()
    actions_tag = "AOV Menu Actions"

    # actions
    action_registry.register_action(
        "omni.kit.menu.aov",
        "create_single_aov",
        cls._on_create_single_aov,
        display_name="Create->AOV",
        description="Create AOV",
        tag=actions_tag,
    )


def deregister_actions(extension_id):
    action_registry = omni.kit.actions.core.get_action_registry()
    action_registry.deregister_all_actions_for_extension(extension_id)
