1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# See COPYING file for copyrights details.

     5

     6

from itertools import repeat, islice, chain

     7

     8

## URI parsing functions

     9

    10

def split_as_dict(s, sep, labels):

    11

    return dict(zip(labels, islice(chain(s.split(sep), repeat("")), len(labels))))

    12

    13

def parse_tcp(loc):

    14

    return split_as_dict(loc, ":", ["host", "port"])

    15

    16

def parse_sslpsk(loc):

    17

    locals().update(**split_as_dict(loc, "#", ["hostport", "ID"]))

    18

    return dict(**parse_tcp(hostport), ID=ID) # type: ignore

    19

    20

def parse_serial(loc):

    21

    return split_as_dict(loc, "@", ["device", "baudrate"])

    22

    23

def parse_usb(loc):

    24

    return split_as_dict(loc, ":", ["VID", "PID", "serialnumber"])

    25

    26

## URI building functions

    27

    28

def build_tcp(fields):

    29

    if fields['port']:

    30

        return "{host}:{port}".format(**fields)

    31

    return fields['host']

    32

    33

def build_sslpsk(fields):

    34

    return "{hostport}#{ID}".format(hostport=build_tcp(fields), **fields)

    35

    36

def build_serial(fields):

    37

    if fields['baudrate']:

    38

        return "{device}@{baudrate}".format(**fields)

    39

    return fields['device']

    40

    41

def build_usb(fields):

    42

    if fields['serialnumber']:

    43

        return "{VID}:{PID}:{serialnumber}".format(**fields)

    44

    if fields['PID']:

    45

        return "{VID}:{PID}".format(**fields)

    46

    return fields['VID']

    47

    48

## Dialog fields definition

    49

    50

model_tcp = [('host', _("Host:")),

    51

             ('port', _("Port:"))]

    52

    53

model_serial = [('device', _("Device:")),

    54

                ('baudrate', _("Baud rate:"))]

    55

    56

model_usb = [('VID', _("Vendor ID:")),

    57

             ('PID', _("Product ID:")),

    58

             ('serialnumber', _("Serial number:"))]

    59

    60

    61

## Schemes description

    62

    63

schemes_desc = [

    64

#   ( scheme name ,  data model , use ID,   parser   ,    builder  )

    65

    ("LOCAL",       [],           False, lambda x: {}, lambda x: ""),

    66

    ("ERPC",        model_tcp,    False, parse_tcp,    build_tcp   ),

    67

    ("ERPCS",       model_tcp,    True,  parse_sslpsk, build_sslpsk),

    68

    ("ERPC-SERIAL", model_serial, False, parse_serial, build_serial),

    69

    ("ERPC-USB",    model_usb,    False, parse_usb,    build_usb   )]

    70

    71

per_scheme_model = {sch: desc for sch, *desc in schemes_desc}

    72