import omni.ext
import omni.kit.livestream.bind


class Extension(omni.ext.IExt):
    def __init__(self):
        pass

    def on_startup(self):
        self._kit_livestream = omni.kit.livestream.bind.acquire_livestream_interface()
        self._kit_livestream.startup()

    def on_shutdown(self):
        self._kit_livestream.shutdown()
        self._kit_livestream = None
