[42] | 1 | # The flight handling "wizard"
|
---|
| 2 |
|
---|
| 3 | from mlx.gui.common import *
|
---|
| 4 |
|
---|
| 5 | #-----------------------------------------------------------------------------
|
---|
| 6 |
|
---|
[44] | 7 | class Page(gtk.Alignment):
|
---|
[42] | 8 | """A page in the flight wizard."""
|
---|
[44] | 9 | def __init__(self, wizard, title, help):
|
---|
[42] | 10 | """Construct the page."""
|
---|
[44] | 11 | super(Page, self).__init__(xalign = 0.0, yalign = 0.0,
|
---|
| 12 | xscale = 1.0, yscale = 1.0)
|
---|
| 13 | self.set_padding(padding_top = 4, padding_bottom = 4,
|
---|
| 14 | padding_left = 12, padding_right = 12)
|
---|
| 15 |
|
---|
| 16 | frame = gtk.Frame()
|
---|
| 17 | self.add(frame)
|
---|
| 18 |
|
---|
| 19 | style = self.get_style() if pygobject else self.rc_get_style()
|
---|
| 20 |
|
---|
| 21 | self._vbox = gtk.VBox()
|
---|
[48] | 22 | self._vbox.set_homogeneous(False)
|
---|
[44] | 23 | frame.add(self._vbox)
|
---|
| 24 |
|
---|
| 25 | eventBox = gtk.EventBox()
|
---|
| 26 | eventBox.modify_bg(0, style.bg[3])
|
---|
| 27 |
|
---|
| 28 | alignment = gtk.Alignment(xalign = 0.0, xscale = 0.0)
|
---|
| 29 |
|
---|
| 30 | label = gtk.Label(title)
|
---|
| 31 | label.modify_fg(0, style.fg[3])
|
---|
| 32 | label.modify_font(pango.FontDescription("bold 24"))
|
---|
| 33 | alignment.set_padding(padding_top = 4, padding_bottom = 4,
|
---|
| 34 | padding_left = 6, padding_right = 0)
|
---|
| 35 |
|
---|
| 36 | alignment.add(label)
|
---|
| 37 | eventBox.add(alignment)
|
---|
| 38 |
|
---|
| 39 | self._vbox.pack_start(eventBox, False, False, 0)
|
---|
| 40 |
|
---|
[48] | 41 | table = gtk.Table(3, 1)
|
---|
| 42 | table.set_homogeneous(True)
|
---|
| 43 |
|
---|
| 44 | alignment = gtk.Alignment(xalign = 0.0, yalign = 0.0,
|
---|
| 45 | xscale = 1.0, yscale = 1.0)
|
---|
| 46 | alignment.set_padding(padding_top = 16, padding_bottom = 16,
|
---|
| 47 | padding_left = 16, padding_right = 16)
|
---|
| 48 | alignment.add(table)
|
---|
| 49 | self._vbox.pack_start(alignment, True, True, 0)
|
---|
| 50 |
|
---|
| 51 | alignment = gtk.Alignment(xalign = 0.5, yalign = 0.0,
|
---|
| 52 | xscale = 0, yscale = 0.0)
|
---|
| 53 |
|
---|
[44] | 54 | label = gtk.Label(help)
|
---|
| 55 | label.set_justify(gtk.Justification.CENTER if pygobject
|
---|
| 56 | else gtk.JUSTIFY_CENTER)
|
---|
| 57 | alignment.add(label)
|
---|
[48] | 58 | table.attach(alignment, 0, 1, 0, 1)
|
---|
[44] | 59 |
|
---|
| 60 | self._mainAlignment = gtk.Alignment(xalign = 0.5, yalign = 0.5,
|
---|
[48] | 61 | xscale = 1.0, yscale = 1.0)
|
---|
| 62 | table.attach(self._mainAlignment, 0, 1, 1, 3)
|
---|
| 63 |
|
---|
| 64 | buttonAlignment = gtk.Alignment(xalign = 1.0, xscale=0.0, yscale = 0.0)
|
---|
[44] | 65 | buttonAlignment.set_padding(padding_top = 4, padding_bottom = 10,
|
---|
| 66 | padding_left = 16, padding_right = 16)
|
---|
| 67 |
|
---|
| 68 | self._buttonBox = gtk.HButtonBox()
|
---|
[46] | 69 | self._defaultButton = None
|
---|
[44] | 70 | buttonAlignment.add(self._buttonBox)
|
---|
| 71 |
|
---|
| 72 | self._vbox.pack_start(buttonAlignment, False, False, 0)
|
---|
| 73 |
|
---|
[42] | 74 | self._wizard = wizard
|
---|
| 75 |
|
---|
[44] | 76 | def setMainWidget(self, widget):
|
---|
| 77 | """Set the given widget as the main one."""
|
---|
| 78 | self._mainAlignment.add(widget)
|
---|
| 79 |
|
---|
[46] | 80 | def addButton(self, label, default = False):
|
---|
[44] | 81 | """Add a button with the given label.
|
---|
| 82 |
|
---|
| 83 | Return the button object created."""
|
---|
| 84 | button = gtk.Button(label)
|
---|
| 85 | self._buttonBox.add(button)
|
---|
[48] | 86 | button.set_use_underline(True)
|
---|
[46] | 87 | if default:
|
---|
| 88 | button.set_can_default(True)
|
---|
| 89 | self._defaultButton = button
|
---|
[44] | 90 | return button
|
---|
| 91 |
|
---|
[48] | 92 | def activate(self):
|
---|
| 93 | """Called when this page becomes active.
|
---|
| 94 |
|
---|
| 95 | This default implementation does nothing."""
|
---|
| 96 | pass
|
---|
| 97 |
|
---|
[46] | 98 | def grabDefault(self):
|
---|
| 99 | """If the page has a default button, make it the default one."""
|
---|
| 100 | if self._defaultButton is not None:
|
---|
| 101 | self._defaultButton.grab_default()
|
---|
| 102 |
|
---|
[42] | 103 | #-----------------------------------------------------------------------------
|
---|
| 104 |
|
---|
| 105 | class LoginPage(Page):
|
---|
| 106 | """The login page."""
|
---|
| 107 | def __init__(self, wizard):
|
---|
| 108 | """Construct the login page."""
|
---|
[44] | 109 | help = "Enter your MAVA pilot's ID and password to\n" \
|
---|
| 110 | "log in to the MAVA website and download\n" \
|
---|
| 111 | "your booked flights."
|
---|
| 112 | super(LoginPage, self).__init__(wizard, "Login", help)
|
---|
[42] | 113 |
|
---|
[48] | 114 | alignment = gtk.Alignment(xalign = 0.5, yalign = 0.5,
|
---|
| 115 | xscale = 0.0, yscale = 0.0)
|
---|
| 116 |
|
---|
[45] | 117 | table = gtk.Table(2, 3)
|
---|
[42] | 118 | table.set_row_spacings(4)
|
---|
| 119 | table.set_col_spacings(32)
|
---|
[48] | 120 | alignment.add(table)
|
---|
| 121 | self.setMainWidget(alignment)
|
---|
[42] | 122 |
|
---|
| 123 | labelAlignment = gtk.Alignment(xalign=1.0, xscale=0.0)
|
---|
[45] | 124 | label = gtk.Label("Pilot _ID:")
|
---|
| 125 | label.set_use_underline(True)
|
---|
| 126 | labelAlignment.add(label)
|
---|
[42] | 127 | table.attach(labelAlignment, 0, 1, 0, 1)
|
---|
| 128 |
|
---|
| 129 | self._pilotID = gtk.Entry()
|
---|
| 130 | self._pilotID.connect("changed", self._setLoginButton)
|
---|
[45] | 131 | self._pilotID.set_tooltip_text("Enter your MAVA pilot's ID. This "
|
---|
| 132 | "usually starts with a "
|
---|
| 133 | "'P' followed by 3 digits.")
|
---|
[42] | 134 | table.attach(self._pilotID, 1, 2, 0, 1)
|
---|
[45] | 135 | label.set_mnemonic_widget(self._pilotID)
|
---|
[42] | 136 |
|
---|
| 137 | labelAlignment = gtk.Alignment(xalign=1.0, xscale=0.0)
|
---|
[45] | 138 | label = gtk.Label("_Password:")
|
---|
| 139 | label.set_use_underline(True)
|
---|
| 140 | labelAlignment.add(label)
|
---|
[42] | 141 | table.attach(labelAlignment, 0, 1, 1, 2)
|
---|
| 142 |
|
---|
| 143 | self._password = gtk.Entry()
|
---|
| 144 | self._password.set_visibility(False)
|
---|
| 145 | self._password.connect("changed", self._setLoginButton)
|
---|
[45] | 146 | self._password.set_tooltip_text("Enter the password for your pilot's ID")
|
---|
[42] | 147 | table.attach(self._password, 1, 2, 1, 2)
|
---|
[45] | 148 | label.set_mnemonic_widget(self._password)
|
---|
[42] | 149 |
|
---|
[45] | 150 | self._rememberButton = gtk.CheckButton("_Remember password")
|
---|
| 151 | self._rememberButton.set_use_underline(True)
|
---|
| 152 | self._rememberButton.set_tooltip_text("If checked, your password will "
|
---|
| 153 | "be stored, so that you should "
|
---|
| 154 | "not have to enter it every time. "
|
---|
| 155 | "Note, however, that the password "
|
---|
| 156 | "is stored as text, and anybody "
|
---|
| 157 | "who can access your files will "
|
---|
| 158 | "be able to read it.")
|
---|
| 159 | table.attach(self._rememberButton, 1, 2, 2, 3, ypadding = 8)
|
---|
| 160 |
|
---|
[46] | 161 | self._loginButton = self.addButton("_Login", default = True)
|
---|
[42] | 162 | self._loginButton.set_sensitive(False)
|
---|
| 163 | self._loginButton.connect("clicked", self._loginClicked)
|
---|
[45] | 164 | self._loginButton.set_tooltip_text("Click to log in.")
|
---|
[42] | 165 |
|
---|
| 166 | config = self._wizard.gui.config
|
---|
| 167 | self._pilotID.set_text(config.pilotID)
|
---|
| 168 | self._password.set_text(config.password)
|
---|
[45] | 169 | self._rememberButton.set_active(config.rememberPassword)
|
---|
[42] | 170 |
|
---|
| 171 | def _setLoginButton(self, entry):
|
---|
| 172 | """Set the login button's sensitivity.
|
---|
| 173 |
|
---|
| 174 | The button is sensitive only if both the pilot ID and the password
|
---|
| 175 | fields contain values."""
|
---|
| 176 | self._loginButton.set_sensitive(self._pilotID.get_text()!="" and
|
---|
| 177 | self._password.get_text()!="")
|
---|
| 178 |
|
---|
| 179 | def _loginClicked(self, button):
|
---|
| 180 | """Called when the login button was clicked."""
|
---|
[49] | 181 | self._wizard.gui.beginBusy("Logging in...")
|
---|
[42] | 182 | self._wizard.gui.webHandler.login(self._pilotID.get_text(),
|
---|
| 183 | self._password.get_text(),
|
---|
| 184 | self._loginResultCallback)
|
---|
| 185 |
|
---|
| 186 | def _loginResultCallback(self, returned, result):
|
---|
| 187 | """The login result callback, called in the web handler's thread."""
|
---|
| 188 | gobject.idle_add(self._handleLoginResult, returned, result)
|
---|
| 189 |
|
---|
| 190 | def _handleLoginResult(self, returned, result):
|
---|
| 191 | """Handle the login result."""
|
---|
[49] | 192 | self._wizard.gui.endBusy()
|
---|
[42] | 193 | if returned:
|
---|
| 194 | if result.loggedIn:
|
---|
| 195 | config = self._wizard.gui.config
|
---|
[45] | 196 |
|
---|
[42] | 197 | config.pilotID = self._pilotID.get_text()
|
---|
[45] | 198 |
|
---|
| 199 | rememberPassword = self._rememberButton.get_active()
|
---|
| 200 | config.password = self._password.get_text() if rememberPassword \
|
---|
| 201 | else ""
|
---|
| 202 |
|
---|
| 203 | config.rememberPassword = rememberPassword
|
---|
| 204 |
|
---|
[43] | 205 | config.save()
|
---|
[48] | 206 | self._wizard._loginResult = result
|
---|
[42] | 207 | self._wizard.nextPage()
|
---|
| 208 | else:
|
---|
| 209 | dialog = gtk.MessageDialog(type = MESSAGETYPE_ERROR,
|
---|
| 210 | buttons = BUTTONSTYPE_OK,
|
---|
| 211 | message_format =
|
---|
| 212 | "Invalid pilot's ID or password.")
|
---|
| 213 | dialog.format_secondary_markup("Check the ID and try to reenter"
|
---|
| 214 | " the password.")
|
---|
| 215 | dialog.run()
|
---|
| 216 | dialog.hide()
|
---|
| 217 | else:
|
---|
| 218 | dialog = gtk.MessageDialog(type = MESSAGETYPE_ERROR,
|
---|
| 219 | buttons = BUTTONSTYPE_OK,
|
---|
| 220 | message_format =
|
---|
| 221 | "Failed to connect to the MAVA website.")
|
---|
| 222 | dialog.format_secondary_markup("Try again in a few minutes.")
|
---|
| 223 | dialog.run()
|
---|
| 224 | dialog.hide()
|
---|
| 225 |
|
---|
| 226 | #-----------------------------------------------------------------------------
|
---|
| 227 |
|
---|
| 228 | class FlightSelectionPage(Page):
|
---|
| 229 | """The page to select the flight."""
|
---|
| 230 | def __init__(self, wizard):
|
---|
| 231 | """Construct the flight selection page."""
|
---|
[44] | 232 | super(FlightSelectionPage, self).__init__(wizard, "Flight selection",
|
---|
[48] | 233 | "Select the flight you want "
|
---|
| 234 | "to perform.")
|
---|
| 235 |
|
---|
| 236 |
|
---|
| 237 | self._listStore = gtk.ListStore(str, str, str, str)
|
---|
| 238 | self._flightList = gtk.TreeView(self._listStore)
|
---|
| 239 | column = gtk.TreeViewColumn("Flight no.", gtk.CellRendererText(),
|
---|
| 240 | text = 1)
|
---|
| 241 | column.set_expand(True)
|
---|
| 242 | self._flightList.append_column(column)
|
---|
| 243 | column = gtk.TreeViewColumn("Departure time [UTC]", gtk.CellRendererText(),
|
---|
| 244 | text = 0)
|
---|
| 245 | column.set_expand(True)
|
---|
| 246 | self._flightList.append_column(column)
|
---|
| 247 | column = gtk.TreeViewColumn("From", gtk.CellRendererText(),
|
---|
| 248 | text = 2)
|
---|
| 249 | column.set_expand(True)
|
---|
| 250 | self._flightList.append_column(column)
|
---|
| 251 | column = gtk.TreeViewColumn("To", gtk.CellRendererText(),
|
---|
| 252 | text = 3)
|
---|
| 253 | column.set_expand(True)
|
---|
| 254 | self._flightList.append_column(column)
|
---|
| 255 |
|
---|
| 256 | flightSelection = self._flightList.get_selection()
|
---|
| 257 | flightSelection.connect("changed", self._selectionChanged)
|
---|
| 258 |
|
---|
| 259 | scrolledWindow = gtk.ScrolledWindow()
|
---|
| 260 | scrolledWindow.add(self._flightList)
|
---|
| 261 | scrolledWindow.set_size_request(400, -1)
|
---|
| 262 | scrolledWindow.set_policy(gtk.PolicyType.AUTOMATIC if pygobject
|
---|
| 263 | else gtk.POLICY_AUTOMATIC,
|
---|
| 264 | gtk.PolicyType.ALWAYS if pygobject
|
---|
| 265 | else gtk.POLICY_ALWAYS)
|
---|
| 266 |
|
---|
| 267 | alignment = gtk.Alignment(xalign = 0.5, yalign = 0.0, xscale = 0.0, yscale = 1.0)
|
---|
| 268 | alignment.add(scrolledWindow)
|
---|
| 269 |
|
---|
| 270 | self.setMainWidget(alignment)
|
---|
| 271 |
|
---|
| 272 | self._button = self.addButton(gtk.STOCK_GO_FORWARD, default = True)
|
---|
| 273 | self._button.set_use_stock(True)
|
---|
| 274 | self._button.set_sensitive(False)
|
---|
| 275 |
|
---|
| 276 | self._activated = False
|
---|
| 277 |
|
---|
| 278 | def activate(self):
|
---|
| 279 | """Fill the flight list."""
|
---|
| 280 | if not self._activated:
|
---|
| 281 | for flight in self._wizard.loginResult.flights:
|
---|
| 282 | self._listStore.append([str(flight.departureTime),
|
---|
| 283 | flight.callsign,
|
---|
| 284 | flight.departureICAO,
|
---|
| 285 | flight.arrivalICAO])
|
---|
| 286 | self._activated = True
|
---|
| 287 |
|
---|
| 288 | def _selectionChanged(self, selection):
|
---|
| 289 | """Called when the selection is changed."""
|
---|
| 290 | self._button.set_sensitive(selection.count_selected_rows()==1)
|
---|
[42] | 291 |
|
---|
| 292 | #-----------------------------------------------------------------------------
|
---|
| 293 |
|
---|
| 294 | class Wizard(gtk.VBox):
|
---|
| 295 | """The flight wizard."""
|
---|
| 296 | def __init__(self, gui):
|
---|
| 297 | """Construct the wizard."""
|
---|
| 298 | super(Wizard, self).__init__()
|
---|
| 299 |
|
---|
| 300 | self.gui = gui
|
---|
| 301 |
|
---|
| 302 | self._pages = []
|
---|
| 303 | self._currentPage = None
|
---|
| 304 |
|
---|
| 305 | self._pages.append(LoginPage(self))
|
---|
| 306 | self._pages.append(FlightSelectionPage(self))
|
---|
| 307 |
|
---|
[48] | 308 | self._loginResult = None
|
---|
| 309 |
|
---|
[42] | 310 | self.setCurrentPage(0)
|
---|
| 311 |
|
---|
[48] | 312 | @property
|
---|
| 313 | def loginResult(self):
|
---|
| 314 | """Get the login result."""
|
---|
| 315 | return self._loginResult
|
---|
| 316 |
|
---|
[42] | 317 | def setCurrentPage(self, index):
|
---|
| 318 | """Set the current page to the one with the given index."""
|
---|
| 319 | assert index < len(self._pages)
|
---|
| 320 |
|
---|
| 321 | if self._currentPage is not None:
|
---|
| 322 | self.remove(self._pages[self._currentPage])
|
---|
| 323 |
|
---|
| 324 | self._currentPage = index
|
---|
| 325 | self.add(self._pages[index])
|
---|
[48] | 326 | self._pages[index].activate()
|
---|
[42] | 327 | self.show_all()
|
---|
| 328 |
|
---|
| 329 | def nextPage(self):
|
---|
| 330 | """Go to the next page."""
|
---|
| 331 | self.setCurrentPage(self._currentPage + 1)
|
---|
[48] | 332 | self.grabDefault()
|
---|
[46] | 333 |
|
---|
| 334 | def grabDefault(self):
|
---|
| 335 | """Make the default button of the current page the default."""
|
---|
| 336 | self._pages[self._currentPage].grabDefault()
|
---|
[42] | 337 |
|
---|
| 338 | #-----------------------------------------------------------------------------
|
---|
| 339 |
|
---|