src/ovtk_audiencekit/core/MainProcess.py

@@ -219,9 +219,9 @@ class MainProcess:
                     plugin_module = import_or_reload_mod(module_name,
                                                          default_package='ovtk_audiencekit.plugins',
                                                          external=False)
-                    plugin = plugin_module.Plugin(self.chat_processes, self.event_queue, plugin_name, global_ctx,
+                    plugin = plugin_module.Plugin(self.chat_processes, self.event_queue, plugin_name, global_ctx)
-                                                  **node.props, **secrets_for_mod, _children=node.nodes)
                     self.plugins[plugin_name] = plugin
+                    await plugin._setup(*node.args[1:], **node.props, **secrets_for_mod)
                     # Register UI with webserver
                     self.webserver.register_blueprint(plugin.blueprint)
                 except Exception as e:

src/ovtk_audiencekit/core/PluginBase.py

@@ -86,6 +86,18 @@ class PluginBase(ABC):
                 raise e
             raise PluginError(self._name, str(e)) from e
 
+    async def _setup(self, *args, **kwargs):
+        try:
+            res = self.setup(*args, **kwargs)
+            if asyncio.iscoroutinefunction(self.setup):
+                return await res
+            else:
+                return res
+        except Exception as e:
+            if isinstance(e, KeyboardInterrupt):
+                raise e
+            raise PluginError(self._name, str(e)) from e
+
     # Base class helpers
     def broadcast(self, event):
         """Send event to every active chat"""
@@ -123,6 +135,10 @@ class PluginBase(ABC):
         self._event_queue.put_nowait(event)
 
     # User-defined
+    async def setup(self, *args, **kwargs):
+        """Called when plugin is being loaded."""
+        pass
+
     def close(self):
         """Called when plugin is about to be unloaded. Use this to safely close any resouces if needed"""
         pass
@@ -143,7 +159,7 @@ class PluginBase(ABC):
         pass
 
     @abstractmethod
-    async def run(self, _children=None, _ctx={}, **kwargs):
+    async def run(self, *args, _children=None, _ctx={}, **kwargs):
         """
         Run plugin action, either due to a definition in the config, or due to another plugin
         """

src/ovtk_audiencekit/plugins/AudioAlert/AudioAlert.py

@@ -1,11 +1,11 @@
 import asyncio
+from collections import deque
 
 from ovtk_audiencekit.plugins import PluginBase
 from ovtk_audiencekit.core import Clip
 
 class AudioAlert(PluginBase):
-    def __init__(self, *args, output=None, buffer_length=2048, cutoff_prevention_buffers=None, **kwargs):
+    def setup(self, output=None, buffer_length=2048, cutoff_prevention_buffers=None):
-        super().__init__(*args, **kwargs)
         if cutoff_prevention_buffers:
             self.logger.info('`cutoff_prevention_buffers` are depricated')
 
@@ -13,13 +13,33 @@ class AudioAlert(PluginBase):
         self._buffer_length = int(buffer_length)
         self._output_index = Clip.find_output_index(output)
 
-    def run(self, path, speed=1, immediate=True, **kwargs):
+    def run(self, path, speed=1, immediate=True, poly=1, **kwargs):
-        if self.sounds.get(path) is None:
+        sound = None
-            self.sounds[path] = Clip(path,
+
-                                     self._output_index,
+        if poly != 1:
-                                     buffer_length=self._buffer_length,
+            poly = int(poly)
-                                     speed=speed)
+            sound_dq = self.sounds.get(path)
-        sound = self.sounds.get(path)
+            if sound_dq is None or type(sound_dq) != deque or sound_dq.maxlen != poly:
+                sound_dq = deque(maxlen=poly)
+                self.sounds[path] = sound_dq
+            if len(sound_dq) != poly:
+                self.logger.debug("filling", len(sound_dq), poly, sound_dq)
+                sound = Clip(path,
+                            self._output_index,
+                            buffer_length=self._buffer_length,
+                            speed=speed)
+                sound_dq.append(sound)
+            else:
+                self.logger.debug("rotate", len(sound_dq), poly, sound_dq)
+                sound_dq.rotate(1)
+                sound = sound_dq[0]
+        else:
+            if self.sounds.get(path) is None:
+                self.sounds[path] = Clip(path,
+                                         self._output_index,
+                                         buffer_length=self._buffer_length,
+                                         speed=speed)
+            sound = self.sounds.get(path)
 
         if immediate:
             asyncio.create_task(sound.aplay())

src/ovtk_audiencekit/plugins/Jail/JailPlugin.py

@@ -35,8 +35,7 @@ owomap = {
 }
 
 class JailPlugin(PluginBase):
-    def __init__(self, *args, min_level='vip', persist=True, **kwargs):
+    def setup(self, min_level='vip', persist=True):
-        super().__init__(*args, **kwargs)
         self.persist = persist
         self._cache = os.path.join(CACHE_DIR, 'Jail', 'sentences')
         os.makedirs(os.path.dirname(self._cache), exist_ok=True)

src/ovtk_audiencekit/plugins/OBS/obs.py

@@ -6,19 +6,15 @@ from ovtk_audiencekit.plugins import PluginBase
 
 
 class OBSWSPlugin(PluginBase):
-    def __init__(self, *args, password=None, uri='ws://localhost:4455', **kwargs):
+    async def setup(self, password=None, uri='ws://localhost:4455'):
-        super().__init__(*args, **kwargs)
         self.uri = uri
 
         self.obsws = simpleobsws.WebSocketClient(url=uri, password=password)
-        asyncio.get_event_loop().run_until_complete(self.setup())
+                await self.obsws.connect()
-
+                success = await self.obsws.wait_until_identified()
-    async def setup(self):
+                if not success:
-        await self.obsws.connect()
+                    await self.obsws.disconnect()
-        success = await self.obsws.wait_until_identified()
+                    raise RuntimeError(f'Could not connect to OBS websocket at {self.uri}')
-        if not success:
-            await self.obsws.disconnect()
-            raise RuntimeError(f'Could not connect to OBS websocket at {self.uri}')
 
     async def run(self, type, _children=None, _ctx={}, **kwargs):
         req = simpleobsws.Request(type, requestData=kwargs)

src/ovtk_audiencekit/plugins/OSC/osc.py

@@ -4,8 +4,7 @@ from ovtk_audiencekit.plugins import PluginBase
 
 
 class OSCPlugin(PluginBase):
-    def __init__(self, *args, ip='localhost', port=None, **kwargs):
+    def setup(self, ip='localhost', port=None):
-        super().__init__(*args, **kwargs)
         if port is None:
             raise RuntimeError('A unique port must be specified')
         self.client = SimpleUDPClient(ip, int(port))

src/ovtk_audiencekit/plugins/PhraseCounter/PhraseCounter.py

@@ -64,8 +64,7 @@ class PhraseCounter:
 
 
 class PhraseCounterPlugin(PluginBase):
-    def __init__(self, *args, debounce_time=1, persist=False, **kwargs):
+    def setup(self, debounce_time=1, persist=False):
-        super().__init__(*args, **kwargs)
         self.debounce_time = debounce_time
         self.persist = persist
 

src/ovtk_audiencekit/plugins/Shoutout/ShoutoutPlugin.py

@@ -9,10 +9,8 @@ from ovtk_audiencekit.chats.Twitch import Process as Twitch
 
 
 class ShoutoutPlugin(PluginBase):
-    def __init__(self, *args, command='so', min_level='vip',
+    def setup(self, command='so', min_level='vip',
-                 text='Check out {link}!~ They were last streaming {last_game}',
+                 text='Check out {link}!~ They were last streaming {last_game}'):
-                 **kwargs):
-        super().__init__(*args, **kwargs)
         self.text = text
         if command:
             self.command = Command(name=command, help='Shoutout another user', required_level=min_level)

src/ovtk_audiencekit/plugins/TTS/TTS.py

@@ -13,10 +13,8 @@ from ovtk_audiencekit.core.Data import CACHE_DIR
 
 
 class TextToSpeechPlugin(PluginBase):
-    def __init__(self, *args, output=None, cuda=None,
+    def setup(self, output=None, cuda=None,
-                 engine="tts_models/en/ljspeech/tacotron2-DDC", speaker_wav=None,
+                 engine="tts_models/en/ljspeech/tacotron2-DDC", speaker_wav=None, **kwargs):
-                 _children=None, **kwargs):
-        super().__init__(*args, _children=_children)
 
         self.speaker_wav = speaker_wav