update

Author: bierdok

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='talkops',
-    version='1.3.4',
+    version='1.4.0',
     author='PicoUX',
     description="TalkOps SDK",
     long_description=open('README.md',encoding='utf-8').read(),

talkops/event_bus.py

@@ -5,9 +5,10 @@
 import socket
 
 class EventBus:
-    def __init__(self, use_state, use_config):
+    def __init__(self, use_state, use_config, set_enabled):
         self._use_state = use_state
         self._use_config = use_config
+        self._set_enabled = set_enabled
         self._last_event_state: str | None = None
         self._client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
         self._client.setblocking(False)
@@ -48,6 +49,30 @@ async def _publish_state_periodically(self):
 
     async def _on_event(self, event):
         config = self._use_config()
+        if event['type'] == 'boot':
+            self._set_enabled(event['enabled'])
+            for name, value in event['parameters'].items():
+                for parameter in config['parameters']:
+                    if parameter.name != name:
+                        continue
+                    if isinstance(value, str):
+                        parameter.set_value(value)
+                    else:
+                        parameter.set_value("")
+            asyncio.create_task(self._publish_state())
+        if event['type'] == 'enable':
+            self._set_enabled(True)
+        if event['type'] == 'disable':
+            self._set_enabled(False)
+        ready = True
+        for parameter in config['parameters']:
+            if parameter.is_optional:
+                continue
+            if parameter.has_value:
+                continue
+            ready = False
+        if not ready:
+            return
         if event['type'] == 'function_call':
             for function in config['functions']:
                 if function.__name__ != event['name']:
@@ -64,25 +89,6 @@ async def _on_event(self, event):
                 return
             print(f'Function {function.__name__} not found.', file=sys.stderr)
             return
-        if event['type'] == 'boot':
-            for name, value in event['parameters'].items():
-                for parameter in config['parameters']:
-                    if parameter.name != name:
-                        continue
-                    if isinstance(value, str):
-                        parameter.set_value(value)
-                    else:
-                        parameter.set_value("")
-            ready = True
-            for parameter in config['parameters']:
-                if parameter.is_optional:
-                    continue
-                if parameter.has_value:
-                    continue
-                ready = False
-            asyncio.create_task(self._publish_state())
-            if not ready:
-                return
         if event['type'] in config['callbacks']:
             callback = config['callbacks'][event['type']]
             arguments_list = [event['args'].get(name) for name in callback.__code__.co_varnames]

talkops/extension.py

@@ -15,6 +15,7 @@ def __init__(self):
         self._callbacks = {}
         self._category = None
         self._demo = False
+        self._enabled = False
         self._event_bus = None
         self._features = []
         self._functions = []
@@ -24,7 +25,6 @@ def __init__(self):
         self._instructions = None
         self._name = None
         self._parameters = []
-        self._software_version = None
         self._started = False
         self._website = None
 
@@ -47,7 +47,6 @@ async def _setup(self):
                     'name': 'python',
                     'version': version('talkops'),
                 },
-                'softwareVersion': self._software_version,
                 'website': self._website,
             }
         )
@@ -64,17 +63,20 @@ async def _setup(self):
                     'name': 'python',
                     'version': version('talkops'),
                 },
-                'softwareVersion': self._software_version,
                 'functionSchemas': self._function_schemas,
             },
             lambda: {
                 'callbacks': self._callbacks,
                 'functions': self._functions,
                 'parameters': self._parameters,
-            }
+            },
+            self._set_enabled
         )
         await asyncio.gather(self._event_bus.start())
 
+    def _set_enabled(self, enabled):
+        self._enabled = enabled
+
     def start(self):
         if self._started:
             return
@@ -123,10 +125,6 @@ def set_website(self, website):
         self._website = website
         return self
 
-    def set_software_version(self, software_version):
-        self._software_version = software_version
-        return self
-
     def set_category(self, category):
         if category not in self._get_categories():
             raise ValueError(f'category must be one of the following strings: {", ".join(self._get_categories())}')