ironflock 1.0.3__tar.gz → 1.0.4__tar.gz

{ironflock-1.0.3/ironflock.egg-info → ironflock-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: ironflock
-Version: 1.0.3
+Version: 1.0.4
 Summary: SDK to integrate your IronFlock Industry 4 Apps with the IronFlock Data Infrastructure
 Home-page: https://github.com/RecordEvolution/ironflock-py
 Author: Record Evolution GmbH
@@ -10,6 +10,16 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: autobahn[asyncio,serialization]==22.3.2
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # ironflock
 

{ironflock-1.0.3 → ironflock-1.0.4}/ironflock/__init__.py

@@ -7,4 +7,4 @@ from ironflock.ironflock import IronFlock
 
 __all__ = ["IronFlock", "create_application_component", "create_application_session"]
 
-__version__ = "0.0.19"
+__version__ = "1.0.4"

{ironflock-1.0.3 → ironflock-1.0.4}/ironflock/ironflock.py

@@ -1,8 +1,9 @@
 import os
+import asyncio
 from typing import Optional
 from autobahn.asyncio.component import Component, run
 from autobahn.wamp.interfaces import ISession
-from autobahn.wamp.types import PublishOptions
+from autobahn.wamp.types import PublishOptions, RegisterOptions
 from autobahn.wamp.request import Publication
 
 from ironflock.AutobahnConnection import getSerialNumber, create_application_component
@@ -13,8 +14,6 @@ class IronFlock:
 
     Example:
 
-        ironFlock = IronFlock()
-
         async def main():
             while True:
                 publication = await ironFlock.publish("test.publish.pw", 1, "two", 3, foo="bar")
@@ -23,7 +22,7 @@ class IronFlock:
 
 
         if __name__ == "__main__":
-            asyncio.get_event_loop().create_task(main())
+            ironflock = IronFlock(mainFunc=main)
             ironFlock.run()
     """
 
@@ -45,17 +44,22 @@ class IronFlock:
         async def onJoin(session, details):
             print("component joined")
             self._session = session
+            self._main_task = asyncio.create_task(mainFunc())
+
             if self.mainFunc: 
                 await self.mainFunc()
 
         @self._component.on_disconnect
-        def onDisconnect(*args, **kwargs):
-            print("component disconnected")
-            self._session = None
-
         @self._component.on_leave
-        def onLeave(*args, **kwargs):
+        async def onLeave(*args, **kwargs):
             print("component left")
+            if self._main_task:
+                self._main_task.cancel()
+                try:
+                    await self._main_task
+                except asyncio.CancelledError:
+                    pass
+                self._main_task = None
             self._session = None
 
     @property
@@ -122,6 +126,24 @@ class IronFlock:
         if hasattr(self, "_session") and hasattr(self._session, "call"):
             res = await self._session.call('ironflock.location_service.update', payload, **extra)
         return res
+    
+    async def register_function(self, topic: str, func: function):
+        """Registers a function to be called when a message is received on the given topic.
+        
+        Args:
+            topic (str): The URI of the topic to register the function for, e.g. "example.mytopic1".
+            func (callable): The function to call when a message is received on the topic.
+        """
+        swarm_key = os.environ.get("SWARM_KEY")
+        app_key = os.environ.get("APP_KEY")
+        env_value = os.environ.get("ENV")
+        
+        topic = f"{swarm_key}.{self._device_key}.{app_key}.{env_value}.{topic}"
+        
+        if self._session is not None:
+            await self._session.register(topic, func, options=RegisterOptions(force_reregister=True))
+        else:
+            print("cannot register function, not connected")
 
     async def publish_to_table(
         self, tablename: str, *args, **kwargs

{ironflock-1.0.3 → ironflock-1.0.4/ironflock.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: ironflock
-Version: 1.0.3
+Version: 1.0.4
 Summary: SDK to integrate your IronFlock Industry 4 Apps with the IronFlock Data Infrastructure
 Home-page: https://github.com/RecordEvolution/ironflock-py
 Author: Record Evolution GmbH
@@ -10,6 +10,16 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: autobahn[asyncio,serialization]==22.3.2
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # ironflock
 

{ironflock-1.0.3 → ironflock-1.0.4}/setup.py

@@ -10,7 +10,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="ironflock",
-    version="1.0.3",
+    version="1.0.4",
     description="SDK to integrate your IronFlock Industry 4 Apps with the IronFlock Data Infrastructure",
     long_description=long_description,
     long_description_content_type="text/markdown",