Initial implementation

Author: brunojuca

setup.cfg

@@ -0,0 +1,10 @@
+[options]
+package_dir=
+    =src
+packages=find:
+
+[options.packages.find]
+where=src
+
+[metadata]
+description-file = README.md

setup.py

@@ -0,0 +1,20 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='client_python_m2p',
+    version='1.0.0',
+    url='https://github.com/easytopic-project/client-python-m2p',
+    license='MIT License',
+    author='Bruno Jucá',
+    author_email='brunoolijuca@gmail.com',
+    keywords='python client for m2p',
+    description='Python client for M2P',
+    packages=find_packages('src'),
+    package_dir={'': 'src'},
+    install_requires=['pika'],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+)

src/client_python_m2p/__init__.py

@@ -0,0 +1,56 @@
+import pika
+import time
+import json
+
+class M2P:
+
+    def __init__(self, queue_host, host_port, specs, user_callback):
+        self.queue_host = queue_host
+        self.host_port = host_port
+        self.specs = specs
+        self.RECONNECT_TIMEOUT = 3
+        self.RECONNECT_MAX = 30
+        self.ch = None
+        self.user_callback = user_callback
+
+
+    def __getConnection(self):
+        retries = 0
+        try:
+            print("Connection succeed")
+            return pika.BlockingConnection(
+                pika.ConnectionParameters(
+                    host=self.queue_host, port=self.host_port, client_properties={"module_specs": self.specs})
+            )
+        except:
+            retries += 1
+            if (retries > self.RECONNECT_MAX):
+                raise Exception("Failed to connect to server")
+            print("Connection failed, attempt " + str(retries) +
+                ". Retrying in " + str(self.RECONNECT_TIMEOUT) + " seconds...", flush=True)
+            time.sleep(self.RECONNECT_TIMEOUT)
+            return self.__getConnection(self.queue_host, self.host_port, self.specs)
+
+    def callback(self, ch, method, properties, body, user_callback, id):
+        print("Running python test", flush=True)
+
+        msg = json.loads(body)
+
+        msg = user_callback(msg)
+
+        ch.basic_publish(exchange="", routing_key=id + "-out", body=json.dumps(msg))
+
+        print("Results sent.", flush=True)
+
+    def connect(self):
+        connection = self.__getConnection()
+        ch = connection.channel()
+        ch.queue_declare(self.specs["id"] + "-in", durable=True)
+        ch.queue_declare(self.specs["id"] + "-out", durable=True)
+        self.ch = ch
+
+    def run(self):
+        self.ch.basic_consume(queue=self.specs["id"] + "-in", on_message_callback=self.callback, auto_ack=True)
+        print("Waiting for messages in " + self.specs["id"] + "-in" +
+            ". Output will be sended to " + self.specs["id"] + "-out" + ". To exit press CTRL+C", flush=True)
+        self.ch.start_consuming()

test/example.py

@@ -0,0 +1,30 @@
+import sys, os.path
+modules_dir = (os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))+ '/src/')
+sys.path.append(modules_dir)
+
+from client_python_m2p import M2P
+
+import os
+import json
+
+f = open(os.path.dirname(__file__) + '/module-name-specs.json')
+specs = json.load(f)
+
+QUEUE_SERVER_HOST, QUEUE_SERVER_PORT = os.environ.get(
+    "QUEUE_SERVER", "localhost:3002").split(":")
+
+def myCode(msg):
+
+    print(msg)
+    
+    msg["echo-image"] = msg["input-image"]
+    msg["echo-video"] = msg["input-video"]
+    msg["backwords-text"] = msg["input-text"][::-1]
+
+    return msg
+
+MyM2P = M2P(QUEUE_SERVER_HOST, QUEUE_SERVER_PORT, specs, myCode)
+
+MyM2P.connect()
+
+MyM2P.run()

test/module-name-specs.json

@@ -0,0 +1,50 @@
+{
+  "id": "template-module",
+  "name": "Template Module",
+  "description": "Template module that receives an image, a video and a text, and give everything back as output, but with the text backwords",
+  "author": "Bruno Jucá",
+  "email": "brunojuca@ice.ufjf.br",
+  "input_queue": "template-module-python-in",
+  "output_queue": "template-module-python-out",
+  "input": [
+    {
+      "id": "input-image",
+      "link": true,
+      "type": "file",
+      "required": true,
+      "accept": ["video/mp4"]
+    },
+    {
+      "id": "input-video",
+      "link": true,
+      "type": "file",
+      "required": true,
+      "accept": ["video/mp4"]
+    },
+    {
+      "id": "input-text",
+      "link": false,
+      "type": "text",
+      "required": true
+    }
+  ],
+  "output": [
+    {
+      "id": "echo-image",
+      "link": true,
+      "type": "file",
+      "accept": ["video/mp4"]
+    },
+    {
+      "id": "echo-video",
+      "link": true,
+      "type": "file",
+      "accept": ["video/mp4"]
+    },
+    {
+      "id": "echo-text",
+      "link": false,
+      "type": "text"
+    }
+  ]
+}