source: src/mlx/gui/checklist.py@ 241:dea155dd3ac0

Last change on this file since 241:dea155dd3ac0 was 233:c0ab38ca1546, checked in by István Váradi <ivaradi@…>, 12 years ago

Fixed the handling of filenames acquired via file choosers
File size: 12.0 KB
Line 
1# Module for editing checklists
2
3#------------------------------------------------------------------------------
4
5from common import *
6
7from mlx.i18n import xstr
8import mlx.const as const
9import mlx.config as config
10
11import os
12
13#------------------------------------------------------------------------------
14
15class ChecklistEditor(gtk.Dialog):
16 """The dialog to edit the checklists."""
17 def __init__(self, gui):
18 super(ChecklistEditor, self).__init__(WINDOW_TITLE_BASE + " - " +
19 xstr("chklst_title"),
20 gui.mainWindow,
21 DIALOG_MODAL)
22
23 self.add_button(xstr("button_cancel"), RESPONSETYPE_REJECT)
24 self.add_button(xstr("button_ok"), RESPONSETYPE_ACCEPT)
25
26 self._gui = gui
27 self._checklists = {}
28 self._currentAircraftType = const.aircraftTypes[0]
29
30 contentArea = self.get_content_area()
31
32 typeBox = gtk.HBox()
33
34 label = gtk.Label(xstr("chklst_aircraftType"))
35 label.set_use_underline(True)
36
37 typeBox.pack_start(label, False, False, 4)
38
39 self._aircraftTypeModel = gtk.ListStore(str, int)
40 for type in const.aircraftTypes:
41 name = aircraftNames[type] if type in aircraftNames \
42 else "Aircraft type #%d" % (type,)
43 self._aircraftTypeModel.append([name, type])
44 self._aircraftType = gtk.ComboBox(model = self._aircraftTypeModel)
45 renderer = gtk.CellRendererText()
46 self._aircraftType.pack_start(renderer, True)
47 self._aircraftType.add_attribute(renderer, "text", 0)
48 self._aircraftType.set_tooltip_text(xstr("chklst_aircraftType_tooltip"))
49 self._aircraftType.set_active(0)
50 self._aircraftType.connect("changed", self._aircraftTypeChanged)
51 label.set_mnemonic_widget(self._aircraftType)
52
53 typeBox.pack_start(self._aircraftType, True, True, 4)
54
55 typeBoxAlignment = gtk.Alignment(xalign = 0.5, yalign = 0.5,
56 xscale = 0.0, yscale = 0.0)
57 typeBoxAlignment.set_size_request(400, -1)
58 typeBoxAlignment.add(typeBox)
59
60 contentArea.pack_start(typeBoxAlignment, False, False, 12)
61
62 fileBox = gtk.HBox()
63
64 self._fileChooser = gtk.FileChooserWidget()
65 self._fileChooser.set_select_multiple(True)
66
67 filter = gtk.FileFilter()
68 filter.set_name(xstr("chklst_filter_audio"))
69 filter.add_pattern("*.wav")
70 filter.add_pattern("*.mp3")
71 self._fileChooser.add_filter(filter)
72
73 filter = gtk.FileFilter()
74 filter.set_name(xstr("file_filter_all"))
75 filter.add_pattern("*.*")
76 self._fileChooser.add_filter(filter)
77
78 self._fileChooser.connect("selection-changed",
79 self._fileChooserSelectionChanged)
80
81 fileBox.pack_start(self._fileChooser, True, True, 4)
82
83 controlBox = gtk.VBox()
84 controlAlignment = gtk.Alignment(xalign = 0.0, yalign = 0.0,
85 xscale = 0.0, yscale = 0.0)
86 controlAlignment.set_padding(padding_top = 0, padding_bottom = 0,
87 padding_left = 32, padding_right = 32)
88 controlAlignment.add(controlBox)
89 fileBox.pack_start(controlAlignment, False, False, 0)
90
91 self._addButton = gtk.Button(xstr("chklst_add"))
92 self._addButton.set_use_underline(True)
93 self._addButton.set_tooltip_text(xstr("chklst_add_tooltip"))
94 self._addButton.connect("clicked", self._addButtonClicked)
95 addAlignment = gtk.Alignment(xalign = 0.5, yalign = 0.0,
96 xscale = 0.0, yscale = 0.0)
97 addAlignment.set_padding(padding_top = 64, padding_bottom = 0,
98 padding_left = 0, padding_right = 0)
99 addAlignment.add(self._addButton)
100 controlBox.pack_start(addAlignment, False, False, 0)
101
102 self._removeButton = gtk.Button(xstr("chklst_remove"))
103 self._removeButton.set_use_underline(True)
104 self._removeButton.set_tooltip_text(xstr("chklst_remove_tooltip"))
105 self._removeButton.set_sensitive(False)
106 self._removeButton.connect("clicked", self._removeButtonClicked)
107
108 removeAlignment = gtk.Alignment(xalign = 0.5, yalign = 0.0,
109 xscale = 0.0, yscale = 0.0)
110 removeAlignment.set_padding(padding_top = 64, padding_bottom = 0,
111 padding_left = 0, padding_right = 0)
112 removeAlignment.add(self._removeButton)
113 controlBox.pack_start(removeAlignment, False, False, 0)
114
115 self._moveUpButton = gtk.Button(xstr("chklst_moveUp"))
116 self._moveUpButton.set_use_underline(True)
117 self._moveUpButton.set_tooltip_text(xstr("chklst_moveUp_tooltip"))
118 self._moveUpButton.set_sensitive(False)
119 self._moveUpButton.connect("clicked", self._moveUpButtonClicked)
120
121 moveUpAlignment = gtk.Alignment(xalign = 0.5, yalign = 0.0,
122 xscale = 0.0, yscale = 0.0)
123 moveUpAlignment.set_padding(padding_top = 16, padding_bottom = 0,
124 padding_left = 0, padding_right = 0)
125 moveUpAlignment.add(self._moveUpButton)
126 controlBox.pack_start(moveUpAlignment, False, False, 0)
127
128 self._moveDownButton = gtk.Button(xstr("chklst_moveDown"))
129 self._moveDownButton.set_use_underline(True)
130 self._moveDownButton.set_tooltip_text(xstr("chklst_moveDown_tooltip"))
131 self._moveDownButton.set_sensitive(False)
132 self._moveDownButton.connect("clicked", self._moveDownButtonClicked)
133
134 moveDownAlignment = gtk.Alignment(xalign = 0.5, yalign = 0.0,
135 xscale = 0.0, yscale = 0.0)
136 moveDownAlignment.set_padding(padding_top = 4, padding_bottom = 0,
137 padding_left = 0, padding_right = 0)
138 moveDownAlignment.add(self._moveDownButton)
139 controlBox.pack_start(moveDownAlignment, False, False, 0)
140
141 self._fileListModel = gtk.ListStore(str, str)
142 self._fileList = gtk.TreeView(model = self._fileListModel)
143 column = gtk.TreeViewColumn(xstr("chklst_header"),
144 gtk.CellRendererText(), text = 0)
145 column.set_expand(True)
146 column.set_clickable(False)
147 column.set_reorderable(False)
148 self._fileList.append_column(column)
149 self._fileList.set_tooltip_column(1)
150 self._fileList.set_reorderable(True)
151 self._fileList.set_size_request(200, -1)
152 selection = self._fileList.get_selection()
153 selection.set_mode(SELECTION_MULTIPLE)
154 selection.connect("changed", self._fileListSelectionChanged)
155
156 fileBox.pack_start(self._fileList, False, False, 4)
157
158 contentArea.pack_start(fileBox, True, True, 4)
159
160 self.set_size_request(900, 500)
161
162 def run(self):
163 """Run the checklist editor dialog."""
164 self._checklists = {}
165 self._displayCurrentChecklist()
166 self.show_all()
167 response = super(ChecklistEditor, self).run()
168 self.hide()
169
170 if response==RESPONSETYPE_ACCEPT:
171 self._saveChecklist()
172 config = self._gui.config
173 for (aircraftType, checklist) in self._checklists.iteritems():
174 config.setChecklist(aircraftType, checklist)
175 config.save()
176
177 def _aircraftTypeChanged(self, comboBox):
178 """Called when the aircraft's type has changed."""
179 self._saveChecklist()
180 self._displayCurrentChecklist()
181
182 def _fileChooserSelectionChanged(self, fileChooser):
183 """Called when the selection of the given file chooser is changed."""
184 numFiles = 0
185 numSelected = 0
186 for path in fileChooser.get_filenames():
187 path = text2unicode(path)
188 numSelected += 1
189 if os.path.isfile(path): numFiles += 1
190
191 self._addButton.set_sensitive(numFiles>0 and numFiles==numSelected)
192
193 def _addButtonClicked(self, button):
194 """Called when the Add button is clicked."""
195 for path in self._fileChooser.get_filenames():
196 path = text2unicode(path)
197 self._fileListModel.append([os.path.basename(path),
198 path])
199 self._fileChooser.unselect_all()
200
201 def _removeButtonClicked(self, button):
202 """Called when the Remove button is clicked."""
203 selection = self._fileList.get_selection()
204 (model, paths) = selection.get_selected_rows()
205 while paths:
206 model.remove(model.get_iter(paths[0]))
207 (model, paths) = selection.get_selected_rows()
208
209 def _moveUpButtonClicked(self, button):
210 """Called when the move up button is clicked."""
211 self._moveSelected(True)
212
213 def _moveDownButtonClicked(self, button):
214 """Called when the move down button is clicked."""
215 self._moveSelected(False)
216
217 def _moveSelected(self, up):
218 """Move the selected files up or down."""
219 selection = self._fileList.get_selection()
220 (model, paths) = selection.get_selected_rows()
221 indexes = [(path.get_indices() if pygobject else path)[0]
222 for path in paths]
223 indexes.sort()
224 if not up:
225 indexes.reverse()
226
227 for index in indexes:
228 fromIter = model.iter_nth_child(None, index)
229 toIter = model.iter_nth_child(None, index-1 if up else index + 1)
230 if up:
231 model.move_before(fromIter, toIter)
232 else:
233 model.move_after(fromIter, toIter)
234
235 self._moveUpButton.set_sensitive(indexes[0]>1 if up else True)
236 numRows = model.iter_n_children(None)
237 self._moveDownButton.set_sensitive(True if up else
238 indexes[-1]<(numRows-2))
239
240 def _fileListSelectionChanged(self, selection):
241 """Called when the selection in the file list changes."""
242 anySelected = selection.count_selected_rows()>0
243 self._removeButton.set_sensitive(anySelected)
244
245 if anySelected:
246 (model, paths) = selection.get_selected_rows()
247 minIndex = None
248 maxIndex = None
249 for path in paths:
250 [index] = path.get_indices() if pygobject else path
251 if minIndex is None or index<minIndex: minIndex = index
252 if maxIndex is None or index>maxIndex: maxIndex = index
253
254 self._moveUpButton.set_sensitive(minIndex>0)
255
256 numRows = model.iter_n_children(None)
257 self._moveDownButton.set_sensitive(maxIndex<(numRows-1))
258 else:
259 self._moveUpButton.set_sensitive(False)
260 self._moveDownButton.set_sensitive(False)
261
262 def _getAircraftType(self):
263 """Get the currently selected aircraft type."""
264 index = self._aircraftType.get_active()
265 return self._aircraftTypeModel[index][1]
266
267 def _saveChecklist(self):
268 """Save the currently displayed checklist for the previously displayed
269 aircraft type."""
270 fileList = []
271 model = self._fileListModel
272 iter = model.get_iter_first()
273 while iter is not None:
274 path = model.get(iter, 1)[0]
275 fileList.append(path)
276 iter = model.iter_next(iter)
277
278 self._checklists[self._currentAircraftType] = config.Checklist(fileList)
279
280 def _displayCurrentChecklist(self):
281 """Display the checklist for the currently selected aircraft type."""
282 aircraftType = self._getAircraftType()
283 self._currentAircraftType = aircraftType
284 if aircraftType not in self._checklists:
285 self._checklists[aircraftType] = \
286 self._gui.config.getChecklist(aircraftType).clone()
287 checklist = self._checklists[aircraftType]
288
289 self._fileListModel.clear()
290 for path in checklist:
291 self._fileListModel.append([os.path.basename(path), path])
292
293
294#------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.