1

     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 Destroy(self):

    45

        self.scheme_editor.Destroy()

    46

        wx.Dialog.Destroy(self)

    47

    48

    def OnTypeChoice(self, event):

    49

        index = event.GetSelection()

    50

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

    51

    52

    def SetURI(self, uri):

    53

        try:

    54

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

    55

            scheme = scheme.upper()

    56

        except Exception:

    57

            scheme = None

    58

    59

        if scheme in ConnectorSchemes():

    60

            self.UriTypeChoice.SetStringSelection(scheme)

    61

        else:

    62

            self.UriTypeChoice.SetSelection(0)

    63

            scheme = None

    64

    65

        self._replaceSchemeEditor(scheme)

    66

    67

        if scheme is not None:

    68

            self.scheme_editor.SetLoc(loc)

    69

    70

    def GetURI(self):

    71

        if self.scheme is None:

    72

            return self.scheme_editor.GetURI()

    73

        else:

    74

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

    75

    76

    def _replaceSchemeEditor(self, scheme):

    77

        self.scheme = scheme

    78

    79

        if self.scheme_editor is not None:

    80

            self.editor_sizer.Detach(self.scheme_editor)

    81

            self.scheme_editor.Destroy()

    82

            self.scheme_editor = None

    83

    84

        if scheme is not None:

    85

            EditorClass = EditorClassFromScheme(scheme)

    86

            self.scheme_editor = EditorClass(scheme, self)

    87

        else:

    88

            # None is for searching local network

    89

            self.scheme_editor = DiscoveryPanel(self)

    90

    91

        self.editor_sizer.Add(self.scheme_editor)

    92

        self.scheme_editor.Refresh()

    93

    94

        self.editor_sizer.Layout()

    95

        self.mainSizer.Layout()

    96

        self.Fit()