1

#!/usr/bin/env python

     2

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

     3

    24

    25

import wx

    26

import cPickle

    27

    28

MAX_ITEM_COUNT = 10

    29

MAX_ITEM_SHOWN = 6

    30

if wx.Platform == '__WXMSW__':

    31

    LISTBOX_BORDER_HEIGHT = 2

    32

    LISTBOX_INTERVAL_HEIGHT = 0

    33

else:

    34

    LISTBOX_BORDER_HEIGHT = 4

    35

    LISTBOX_INTERVAL_HEIGHT = 6

    36

    37

class PopupWithListbox(wx.PopupWindow):

    38

    39

    def __init__(self, parent, choices=[]):

    40

        wx.PopupWindow.__init__(self, parent, wx.BORDER_SIMPLE)

    41

    42

        self.ListBox = wx.ListBox(self, -1, style=wx.LB_HSCROLL|wx.LB_SINGLE|wx.LB_SORT)

    43

    44

        self.SetChoices(choices)

    45

    46

        self.ListBox.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)

    47

        self.ListBox.Bind(wx.EVT_MOTION, self.OnMotion)

    48

    49

    def SetChoices(self, choices):

    50

        max_text_width = 0

    51

        max_text_height = 0

    52

    53

        self.ListBox.Clear()

    54

        for choice in choices:

    55

            self.ListBox.Append(choice)

    56

            w, h = self.ListBox.GetTextExtent(choice)

    57

            max_text_width = max(max_text_width, w)

    58

            max_text_height = max(max_text_height, h)

    59

    60

        itemcount = min(len(choices), MAX_ITEM_SHOWN)

    61

        width = self.Parent.GetSize()[0]

    62

        height = max_text_height * itemcount + \

    63

                 LISTBOX_INTERVAL_HEIGHT * max(0, itemcount - 1) + \

    64

                 2 * LISTBOX_BORDER_HEIGHT

    65

        if max_text_width + 10 > width:

    66

            height += 15

    67

        size = wx.Size(width, height)

    68

        if wx.Platform == '__WXMSW__':

    69

            size.width -= 2

    70

        self.ListBox.SetSize(size)

    71

        self.SetClientSize(size)

    72

    73

    def MoveSelection(self, direction):

    74

        selected = self.ListBox.GetSelection()

    75

        if selected == wx.NOT_FOUND:

    76

            if direction >= 0:

    77

                selected = 0

    78

            else:

    79

                selected = self.ListBox.GetCount() - 1

    80

        else:

    81

            selected = (selected + direction) % (self.ListBox.GetCount() + 1)

    82

        if selected == self.ListBox.GetCount():

    83

            selected = wx.NOT_FOUND

    84

        self.ListBox.SetSelection(selected)

    85

    86

    def GetSelection(self):

    87

        return self.ListBox.GetStringSelection()

    88

    89

    def OnLeftDown(self, event):

    90

        selected = self.ListBox.HitTest(wx.Point(event.GetX(), event.GetY()))

    91

        parent_size = self.Parent.GetSize()

    92

        parent_rect = wx.Rect(0, -parent_size[1], parent_size[0], parent_size[1])

    93

        if selected != wx.NOT_FOUND:

    94

            wx.CallAfter(self.Parent.SetValueFromSelected, self.ListBox.GetString(selected))

    95

        elif parent_rect.InsideXY(event.GetX(), event.GetY()):

    96

            result, x, y = self.Parent.HitTest(wx.Point(event.GetX(), event.GetY() + parent_size[1]))

    97

            if result != wx.TE_HT_UNKNOWN:

    98

                self.Parent.SetInsertionPoint(self.Parent.XYToPosition(x, y))

    99

        else:

   100

            wx.CallAfter(self.Parent.DismissListBox)

   101

        event.Skip()

   102

   103

    def OnMotion(self, event):

   104

        self.ListBox.SetSelection(

   105

            self.ListBox.HitTest(wx.Point(event.GetX(), event.GetY())))

   106

        event.Skip()

   107

   108

class TextCtrlAutoComplete(wx.TextCtrl):

   109

   110

    def __init__ (self, parent, choices=None, dropDownClick=True,

   111

                  element_path=None, **therest):

   112

        """

   113

        Constructor works just like wx.TextCtrl except you can pass in a

   114

        list of choices.  You can also change the choice list at any time

   115

        by calling setChoices.

   116

        """

   117

   118

        therest['style'] = wx.TE_PROCESS_ENTER | therest.get('style', 0)

   119

   120

        wx.TextCtrl.__init__(self, parent, **therest)

   121

   122

        #Some variables

   123

        self._dropDownClick = dropDownClick

   124

        self._lastinsertionpoint = None

   125

        self._hasfocus = False

   126

   127

        self._screenheight = wx.SystemSettings.GetMetric(wx.SYS_SCREEN_Y)

   128

        self.element_path = element_path

   129

   130

        self.listbox = None

   131

   132

        self.SetChoices(choices)

   133

   134

        #gp = self

   135

        #while ( gp != None ) :

   136

        #    gp.Bind ( wx.EVT_MOVE , self.onControlChanged, gp )

   137

        #    gp.Bind ( wx.EVT_SIZE , self.onControlChanged, gp )

   138

        #    gp = gp.GetParent()

   139

   140

        self.Bind(wx.EVT_KILL_FOCUS, self.OnControlChanged)

   141

        self.Bind(wx.EVT_TEXT_ENTER, self.OnControlChanged)

   142

        self.Bind(wx.EVT_TEXT, self.OnEnteredText)

   143

        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)

   144

   145

        #If need drop down on left click

   146

        if dropDownClick:

   147

            self.Bind(wx.EVT_LEFT_DOWN, self.OnClickToggleDown)

   148

            self.Bind(wx.EVT_LEFT_UP, self.OnClickToggleUp)

   149

   150

    def ChangeValue(self, value):

   151

        wx.TextCtrl.ChangeValue(self, value)

   152

        self.RefreshListBoxChoices()

   153

   154

    def OnEnteredText(self, event):

   155

        wx.CallAfter(self.RefreshListBoxChoices)

   156

        event.Skip()

   157

   158

    def OnKeyDown(self, event):

   159

        """ Do some work when the user press on the keys:

   160

            up and down: move the cursor

   161

        """

   162

        keycode = event.GetKeyCode()

   163

        if keycode in [wx.WXK_DOWN, wx.WXK_UP]:

   164

            self.PopupListBox()

   165

            if keycode == wx.WXK_DOWN:

   166

                self.listbox.MoveSelection(1)

   167

            else:

   168

                self.listbox.MoveSelection(-1)

   169

        elif keycode in [wx.WXK_LEFT, wx.WXK_RIGHT, wx.WXK_RETURN] and self.listbox is not None:

   170

            selected = self.listbox.GetSelection()

   171

            if selected != "":

   172

                self.SetValueFromSelected(selected)

   173

            else:

   174

                event.Skip()

   175

        elif event.GetKeyCode() == wx.WXK_ESCAPE:

   176

            self.DismissListBox()

   177

        else:

   178

            event.Skip()

   179

   180

    def OnClickToggleDown(self, event):

   181

        self._lastinsertionpoint = self.GetInsertionPoint()

   182

        event.Skip()

   183

   184

    def OnClickToggleUp(self, event):

   185

        if not self._hasfocus:

   186

            self._hasfocus = True

   187

        elif self.GetInsertionPoint() == self._lastinsertionpoint:

   188

            wx.CallAfter(self.PopupListBox)

   189

        self._lastinsertionpoint = None

   190

        event.Skip()

   191

   192

    def OnControlChanged(self, event):

   193

        res = self.GetValue()

   194

        config = wx.ConfigBase.Get()

   195

        listentries = cPickle.loads(str(config.Read(self.element_path, cPickle.dumps([]))))

   196

        if res and res not in listentries:

   197

            listentries = (listentries + [res])[-MAX_ITEM_COUNT:]

   198

            config.Write(self.element_path, cPickle.dumps(listentries))

   199

            config.Flush()

   200

            self.SetChoices(listentries)

   201

        self.DismissListBox()

   202

        self._hasfocus = False

   203

        event.Skip()

   204

   205

    def SetChoices(self, choices):

   206

        self._choices = choices

   207

        self.RefreshListBoxChoices()

   208

   209

    def GetChoices(self):

   210

        return self._choices

   211

   212

    def SetValueFromSelected(self, selected):

   213

        """

   214

        Sets the wx.TextCtrl value from the selected wx.ListCtrl item.

   215

        Will do nothing if no item is selected in the wx.ListCtrl.

   216

        """

   217

        if selected != "":

   218

            self.SetValue(selected)

   219

        self.DismissListBox()

   220

   221

    def RefreshListBoxChoices(self):

   222

        if self.listbox is not None:

   223

            text = self.GetValue()

   224

            choices = [choice for choice in self._choices if choice.startswith(text)]

   225

            self.listbox.SetChoices(choices)

   226

   227

    def PopupListBox(self):

   228

        if self.listbox is None:

   229

            self.listbox = PopupWithListbox(self)

   230

   231

            # Show the popup right below or above the button

   232

            # depending on available screen space...

   233

            pos = self.ClientToScreen((0, 0))

   234

            sz = self.GetSize()

   235

            if wx.Platform == '__WXMSW__':

   236

                pos.x -= 2

   237

                pos.y -= 2

   238

            self.listbox.Position(pos, (0, sz[1]))

   239

   240

            self.RefreshListBoxChoices()

   241

   242

            self.listbox.Show()

   243

   244

    def DismissListBox(self):

   245

        if self.listbox is not None:

   246

            if self.listbox.ListBox.HasCapture():

   247

                self.listbox.ListBox.ReleaseMouse()

   248

            self.listbox.Destroy()

   249

            self.listbox = None