connectors/ERPC_URI.py
changeset 4038 4b47b4ce0f12
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/connectors/ERPC_URI.py	Mon Nov 18 13:37:08 2024 +0100
@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# See COPYING file for copyrights details.
+
+from itertools import repeat, islice, chain
+
+## URI parsing functions
+
+def split_as_dict(s, sep, labels):
+    return dict(zip(labels, islice(chain(s.split(sep), repeat("")), len(labels))))
+
+def parse_tcp(loc):
+    return split_as_dict(loc, ":", ["host", "port"])
+
+def parse_sslpsk(loc):
+    locals().update(**split_as_dict(loc, "#", ["hostport", "ID"]))
+    return dict(**parse_tcp(hostport), ID=ID) # type: ignore
+
+def parse_serial(loc):
+    return split_as_dict(loc, "@", ["device", "baudrate"])
+
+def parse_usb(loc):
+    return split_as_dict(loc, ":", ["VID", "PID", "serialnumber"])
+
+## URI building functions
+
+def build_tcp(fields):
+    if fields['port']:
+        return "{host}:{port}".format(**fields)
+    return fields['host']
+
+def build_sslpsk(fields):
+    return "{hostport}#{ID}".format(hostport=build_tcp(fields), **fields)
+
+def build_serial(fields):
+    if fields['baudrate']:
+        return "{device}@{baudrate}".format(**fields)
+    return fields['device']
+
+def build_usb(fields):
+    if fields['serialnumber']:
+        return "{VID}:{PID}:{serialnumber}".format(**fields)
+    if fields['PID']:
+        return "{VID}:{PID}".format(**fields)
+    return fields['VID']
+    
+## Dialog fields definition
+
+model_tcp = [('host', _("Host:")),
+             ('port', _("Port:"))]
+
+model_serial = [('device', _("Device:")),
+                ('baudrate', _("Baud rate:"))]
+
+model_usb = [('VID', _("Vendor ID:")),
+             ('PID', _("Product ID:")),
+             ('serialnumber', _("Serial number:"))]
+
+
+## Schemes description
+
+schemes_desc = [
+#   ( scheme name ,  data model , use ID,   parser   ,    builder  )
+    ("LOCAL",       [],           False, lambda x: {}, lambda x: ""),
+    ("ERPC",        model_tcp,    False, parse_tcp,    build_tcp   ),
+    ("ERPCS",       model_tcp,    True,  parse_sslpsk, build_sslpsk),
+    ("ERPC-SERIAL", model_serial, False, parse_serial, build_serial),
+    ("ERPC-USB",    model_usb,    False, parse_usb,    build_usb   )]
+
+per_scheme_model = {sch: desc for sch, *desc in schemes_desc}
+