1

#!/usr/bin/env python

     2

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

     3

     4

# See COPYING file for copyrights details.

     5

     6

from __future__ import absolute_import

     7

     8

from itertools import repeat, islice, chain

     9

import wx

    10

    11

from connectors.SchemeEditor import SchemeEditor

    12

    13

    14

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

    15

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

    16

    17

# (scheme, model, secure)

    18

models = [("LOCAL", [], False), ("PYRO", model, False), ("PYROS", model, True)]

    19

    20

Schemes = list(zip(*models)[0])

    21

    22

_PerSchemeConf = {sch: (mod, sec) for sch, mod, sec in models}

    23

    24

    25

class PYRO_dialog(SchemeEditor):

    26

    def __init__(self, scheme, *args, **kwargs):

    27

        # ID selector is enabled only on PYROS (secure)

    28

        self.model, self.EnableIDSelector = _PerSchemeConf[scheme]

    29

    30

        SchemeEditor.__init__(self, scheme, *args, **kwargs)

    31

    32

    # pylint: disable=unused-variable

    33

    def SetLoc(self, loc):

    34

        hostport, ID = list(islice(chain(loc.split("#"), repeat("")), 2))

    35

        host, port = list(islice(chain(hostport.split(":"), repeat("")), 2))

    36

        self.SetFields(locals())

    37

    38

    def GetLoc(self):

    39

        if self.model:

    40

            fields = self.GetFields()

    41

            template = "{host}"

    42

            if fields['port']:

    43

                template += ":{port}"

    44

            if fields['ID']:

    45

                template += "#{ID}"

    46

    47

            return template.format(**fields)

    48

        return ''