initial commit 2

4 files changed, 122 insertions, 0 deletions

diff --git a/addons/hw_drivers/iot_handlers/interfaces/DisplayInterface.py b/addons/hw_drivers/iot_handlers/interfaces/DisplayInterface.py
new file mode 100644
index 00000000..57e7a4af
--- /dev/null
+++ b/addons/hw_drivers/iot_handlers/interfaces/DisplayInterface.py
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from re import sub, finditer
+import subprocess
+
+from odoo.addons.hw_drivers.interface import Interface
+
+
+class DisplayInterface(Interface):
+    _loop_delay = 0
+    connection_type = 'display'
+
+    def get_devices(self):
+        display_devices = {}
+        displays = subprocess.check_output(['tvservice', '-l']).decode()
+        x_screen = 0
+        for match in finditer('Display Number (\d), type HDMI (\d)', displays):
+            display_id, hdmi_id = match.groups()
+            tvservice_output = subprocess.check_output(['tvservice', '-nv', display_id]).decode().strip()
+            if tvservice_output:
+                display_name = tvservice_output.split('=')[1]
+                display_identifier = sub('[^a-zA-Z0-9 ]+', '', display_name).replace(' ', '_') + "_" + str(hdmi_id)
+                iot_device = {
+                    'identifier': display_identifier,
+                    'name': display_name,
+                    'x_screen': str(x_screen),
+                }
+                display_devices[display_identifier] = iot_device
+                x_screen += 1
+
+        if not len(display_devices):
+            # No display connected, create "fake" device to be accessed from another computer
+            display_devices['distant_display'] = {
+                'name': "Distant Display",
+            }
+
+        return display_devices
diff --git a/addons/hw_drivers/iot_handlers/interfaces/PrinterInterface.py b/addons/hw_drivers/iot_handlers/interfaces/PrinterInterface.py
new file mode 100644
index 00000000..ca601dbc
--- /dev/null
+++ b/addons/hw_drivers/iot_handlers/interfaces/PrinterInterface.py
@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from cups import Connection as cups_connection
+from re import sub
+from threading import Lock
+
+from odoo.addons.hw_drivers.interface import Interface
+
+conn = cups_connection()
+PPDs = conn.getPPDs()
+cups_lock = Lock()  # We can only make one call to Cups at a time
+
+class PrinterInterface(Interface):
+    _loop_delay = 120
+    connection_type = 'printer'
+
+    def get_devices(self):
+        printer_devices = {}
+        with cups_lock:
+            printers = conn.getPrinters()
+            devices = conn.getDevices()
+            for printer in printers:
+                path = printers.get(printer).get('device-uri', False)
+                if path and path in devices:
+                    devices.get(path).update({'supported': True}) # these printers are automatically supported
+        for path in devices:
+            if 'uuid=' in path:
+                identifier = sub('[^a-zA-Z0-9_]', '', path.split('uuid=')[1])
+            elif 'serial=' in path:
+                identifier = sub('[^a-zA-Z0-9_]', '', path.split('serial=')[1])
+            else:
+                identifier = sub('[^a-zA-Z0-9_]', '', path)
+            devices[path]['identifier'] = identifier
+            devices[path]['url'] = path
+            printer_devices[identifier] = devices[path]
+        return printer_devices
diff --git a/addons/hw_drivers/iot_handlers/interfaces/SerialInterface.py b/addons/hw_drivers/iot_handlers/interfaces/SerialInterface.py
new file mode 100644
index 00000000..ae639697
--- /dev/null
+++ b/addons/hw_drivers/iot_handlers/interfaces/SerialInterface.py
@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from glob import glob
+
+from odoo.addons.hw_drivers.interface import Interface
+
+
+class SerialInterface(Interface):
+    connection_type = 'serial'
+
+    def get_devices(self):
+        serial_devices = {}
+        for identifier in glob('/dev/serial/by-path/*'):
+            serial_devices[identifier] = {
+                'identifier': identifier
+            }
+        return serial_devices
diff --git a/addons/hw_drivers/iot_handlers/interfaces/USBInterface.py b/addons/hw_drivers/iot_handlers/interfaces/USBInterface.py
new file mode 100644
index 00000000..a40e82d6
--- /dev/null
+++ b/addons/hw_drivers/iot_handlers/interfaces/USBInterface.py
@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from usb import core
+
+from odoo.addons.hw_drivers.interface import Interface
+
+
+class USBInterface(Interface):
+    connection_type = 'usb'
+
+    def get_devices(self):
+        """
+        USB devices are identified by a combination of their `idVendor` and
+        `idProduct`. We can't be sure this combination in unique per equipment.
+        To still allow connecting multiple similar equipments, we complete the
+        identifier by a counter. The drawbacks are we can't be sure the equipments
+        will get the same identifiers after a reboot or a disconnect/reconnect.
+        """
+        usb_devices = {}
+        devs = core.find(find_all=True)
+        cpt = 2
+        for dev in devs:
+            identifier = "usb_%04x:%04x" % (dev.idVendor, dev.idProduct)
+            if identifier in usb_devices:
+                identifier += '_%s' % cpt
+                cpt += 1
+            usb_devices[identifier] = dev
+        return usb_devices