Coverage for src/robotide/widgets/list.py: 35%

1# Copyright 2008-2015 Nokia Networks 

2# Copyright 2016- Robot Framework Foundation 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); 

5# you may not use this file except in compliance with the License. 

6# You may obtain a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, 

12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

13# See the License for the specific language governing permissions and 

14# limitations under the License. 

15 

16import os 2 ctx1ab

17 

18import wx 2 ctx1ab

19from wx.lib.mixins.listctrl import ListCtrlAutoWidthMixin 2 ctx1ab

20 

21IS_WINDOWS = os.sep == '\\' 2 ctx1ab

22 

23 

24class VirtualList(wx.ListCtrl, ListCtrlAutoWidthMixin): 2 ctx1ab

25 _style = wx.LC_REPORT | wx.LC_SINGLE_SEL | wx.LC_HRULES | wx.LC_VIRTUAL 2 ctx1ab

26 

27 def __init__(self, parent, headers, model): 2 ctx1ab

28 wx.ListCtrl.__init__(self, parent, style=self._style) 

29 ListCtrlAutoWidthMixin.__init__(self) 

30 """ 

31 self.SetBackgroundColour(Colour(200, 222, 40)) 

32 self.SetOwnBackgroundColour(Colour(200, 222, 40)) 

33 self.SetForegroundColour(Colour(7, 0, 70)) 

34 self.SetOwnForegroundColour(Colour(7, 0, 70)) 

35 """ 

36 self._model = model 

37 self._selection_listeners = [] 

38 self._create_headers(headers) 

39 self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_list_item_selected) 

40 if IS_WINDOWS: 

41 self.Bind(wx.EVT_LEFT_DOWN, self.on_left_down) 

42 self.SetItemCount(model.count) 

43 self.SetImageList(model.images, wx.IMAGE_LIST_SMALL) 

44 

45 def on_left_down(self, event): 2 ctx1ab

46 item, flags = self.HitTest(event.Position) 

47 if flags | wx.LIST_HITTEST_ONITEM: 

48 wx.CallAfter(self.inform_listeners, item) 

49 event.Skip() 

50 

51 def _create_headers(self, headers): 2 ctx1ab

52 for idx, name in enumerate(headers): 

53 self.InsertColumn(idx, name) 

54 self.SetColumnWidth(0, 200) 

55 self.SetColumnWidth(1, 160) 

56 

57 def refresh_items(self): 2 ctx1ab

58 self.SetItemCount(self._model.count) 

59 self.SetImageList(self._model.images, wx.IMAGE_LIST_SMALL) 

60 

61 def add_selection_listener(self, listener): 2 ctx1ab

62 self._selection_listeners.append(listener) 

63 

64 def on_list_item_selected(self, event): 2 ctx1ab

65 self.inform_listeners(event.Index) 

66 

67 def inform_listeners(self, selected_index): 2 ctx1ab

68 for listener in self._selection_listeners: 

69 listener(selected_index) 

70 

71 def OnGetItemText(self, row, col): # Overrides wx method 2 ctx1ab

72 return self._model.item_text(row, col) 

73 

74 def OnGetItemImage(self, item): # Overrides wx method 2 ctx1ab

75 return self._model.image(item) 

76 

77 def OnGetItemAttr(self, item): # Overrides wx method 2 ctx1ab

78 return self._model.item_attributes(item) 

79 

80 

81class ListModel(object): 2 ctx1ab

82 

83 @property 2 ctx1ab

84 def count(self): 2 ctx1ab

85 return 0 

86 

87 @property 2 ctx1ab

88 def images(self): 2 ctx1ab

89 return None 

90 

91 def item_text(self, row, col): 2 ctx1ab

92 _ = row 

93 _ = col 

94 return '' 

95 

96 def image(self, row): 2 ctx1ab

97 _ = row 

98 return -1 

99 

100 def item_attributes(self, row): 2 ctx1ab

101 _ = row 

102 return None