1

from __future__ import absolute_import

     2

     3

import wx

     4

from connectors import ConnectorSchemes, EditorClassFromScheme

     5

from controls.DiscoveryPanel import DiscoveryPanel

     6

     7

     8

class UriEditor(wx.Dialog):

     9

    def _init_ctrls(self, parent):

    10

        self.UriTypeChoice = wx.Choice(parent=self, choices=self.choices)

    11

        self.UriTypeChoice.SetSelection(0)

    12

        self.Bind(wx.EVT_CHOICE, self.OnTypeChoice, self.UriTypeChoice)

    13

        self.editor_sizer = wx.BoxSizer(wx.HORIZONTAL)

    14

        self.ButtonSizer = self.CreateButtonSizer(wx.OK | wx.CANCEL)

    15

    16

    def _init_sizers(self):

    17

        self.mainSizer = wx.BoxSizer(wx.VERTICAL)

    18

        typeSizer = wx.BoxSizer(wx.HORIZONTAL)

    19

        typeSizer.Add(wx.StaticText(self, wx.ID_ANY, _("Scheme :")), border=5,

    20

                      flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL)

    21

        typeSizer.Add(self.UriTypeChoice, border=5, flag=wx.ALL)

    22

        self.mainSizer.Add(typeSizer)

    23

    24

        self.mainSizer.Add(self.editor_sizer, border=5, flag=wx.ALL)

    25

        self.mainSizer.Add(self.ButtonSizer, border=5,

    26

                           flag=wx.BOTTOM | wx.ALIGN_CENTER_HORIZONTAL)

    27

        self.SetSizer(self.mainSizer)

    28

        self.Layout()

    29

        self.Fit()

    30

    31

    def __init__(self, parent, ctr, uri=''):

    32

        self.ctr = ctr

    33

        wx.Dialog.__init__(self,

    34

                           name='UriEditor', parent=parent,

    35

                           title=_('URI Editor'))

    36

        self.choices = [_("- Search local network -")] + ConnectorSchemes()

    37

        self._init_ctrls(parent)

    38

        self._init_sizers()

    39

        self.scheme = None

    40

        self.scheme_editor = None

    41

        self.SetURI(uri)

    42

        self.CenterOnParent()

    43

    44

    def OnTypeChoice(self, event):

    45

        index = event.GetSelection()

    46

        self._replaceSchemeEditor(event.GetString() if index > 0 else None)

    47

    48

    def SetURI(self, uri):

    49

        try:

    50

            scheme, loc = uri.strip().split("://", 1)

    51

            scheme = scheme.upper()

    52

        except Exception:

    53

            scheme = None

    54

    55

        if scheme in ConnectorSchemes():

    56

            self.UriTypeChoice.SetStringSelection(scheme)

    57

        else:

    58

            self.UriTypeChoice.SetSelection(0)

    59

            scheme = None

    60

    61

        self._replaceSchemeEditor(scheme)

    62

    63

        if scheme is not None:

    64

            self.scheme_editor.SetLoc(loc)

    65

    66

    def GetURI(self):

    67

        if self.scheme is None:

    68

            return self.scheme_editor.GetURI()

    69

        else:

    70

            return self.scheme+"://"+self.scheme_editor.GetLoc()

    71

    72

    def _replaceSchemeEditor(self, scheme):

    73

        self.scheme = scheme

    74

    75

        if self.scheme_editor is not None:

    76

            self.editor_sizer.Detach(self.scheme_editor)

    77

            self.scheme_editor.Destroy()

    78

            self.scheme_editor = None

    79

    80

        if scheme is not None:

    81

            EditorClass = EditorClassFromScheme(scheme)

    82

            self.scheme_editor = EditorClass(scheme, self)

    83

        else:

    84

            # None is for searching local network

    85

            self.scheme_editor = DiscoveryPanel(self)

    86

    87

        self.editor_sizer.Add(self.scheme_editor)

    88

        self.scheme_editor.Refresh()

    89

    90

        self.editor_sizer.Layout()

    91

        self.mainSizer.Layout()

    92

        self.Fit()