8b8d1f75e2543efb4105732a239dd1a70316e250
[mevemon] / package / src / ui / fremantle / gui.py
1 #
2 # mEveMon - A character monitor for EVE Online
3 # Copyright (c) 2010  Ryan and Danny Campbell, and the mEveMon Team
4 #
5 # mEveMon is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # mEveMon is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 #
18
19 import sys, time
20
21 import gtk
22 import hildon
23 import gobject
24
25 from ui import models
26 import validation
27 import util
28
29 class BaseUI():
30     menu_items = ("Settings", "About", "Refresh")
31
32     def create_menu(self, window):
33         menu = hildon.AppMenu()
34
35         for command in self.menu_items:
36             # Create menu entries
37             button = hildon.GtkButton(gtk.HILDON_SIZE_AUTO)
38             button.set_label(command)
39
40             if command == "About":
41                 button.connect("clicked", self.about_clicked)
42             elif command == "Settings":
43                 button.connect("clicked", self.settings_clicked, window)
44             elif command == "Refresh":
45                 button.connect("clicked", self.refresh_clicked)
46             else:
47                 assert False, command
48
49             # Add entry to the view menu
50             menu.append(button)
51
52         menu.show_all()
53
54         return menu
55
56     def settings_clicked(self, button, window):
57
58         RESPONSE_NEW, RESPONSE_EDIT, RESPONSE_DELETE = range(3)
59
60         dialog = gtk.Dialog()
61         dialog.set_transient_for(window)
62         dialog.set_title("Settings")
63
64         pannable_area = hildon.PannableArea()
65
66         dialog_vbox = dialog.vbox
67
68         vbox = gtk.VBox(False, 1)
69
70         self.accounts_model = models.AccountsModel(self.controller)
71         
72         accounts_treeview = hildon.GtkTreeView(gtk.HILDON_UI_MODE_NORMAL)
73         accounts_treeview.set_model(self.accounts_model)
74         accounts_treeview.set_headers_visible(True)
75         self.add_columns_to_accounts(accounts_treeview)
76         vbox.pack_start(accounts_treeview, False, False, 1)
77
78         # all stock responses are negative, so we can use any positive value
79         new_button = dialog.add_button("New", RESPONSE_NEW)
80         #TODO: get edit button working
81         #edit_button = dialog.add_button("Edit", RESPONSE_EDIT)
82         delete_button = dialog.add_button("Delete", RESPONSE_DELETE)
83         ok_button = dialog.add_button(gtk.STOCK_OK, gtk.RESPONSE_OK)
84         
85         pannable_area.add_with_viewport(vbox)
86         
87         dialog_vbox.pack_start(pannable_area, True, True, 1)
88       
89       
90         dialog.show_all()
91
92         result = dialog.run()
93
94         while(result != gtk.RESPONSE_DELETE_EVENT):
95             if result == RESPONSE_NEW:
96                 self.new_account_clicked(window)
97             #elif result == RESPONSE_EDIT:
98             #    # get the selected treeview item and pop up the account_box
99             #    self.edit_account(accounts_treeview)
100             elif result == RESPONSE_DELETE:
101                 # get the selected treeview item, and delete the gconf keys
102                 self.delete_account(accounts_treeview)
103             elif result == gtk.RESPONSE_OK:
104                 self.char_model.get_characters()
105                 break
106         
107             result = dialog.run()
108
109         dialog.destroy()
110
111
112
113     def get_selected_item(self, treeview, column):
114         selection = treeview.get_selection()
115         model, miter = selection.get_selected()
116
117         value = model.get_value(miter, column)
118
119         return value
120
121     def edit_account(self, treeview):
122         uid = self.get_selected_item(treeview, 0)
123         # pop up the account dialog
124
125         self.accounts_model.get_accounts()
126
127     def delete_account(self, treeview):
128         uid = self.get_selected_item(treeview, 0)
129         self.controller.remove_account(uid)
130         # refresh model
131         self.accounts_model.get_accounts()
132
133
134     def add_columns_to_accounts(self, treeview):
135         #Column 0 for the treeview
136         renderer = gtk.CellRendererText()
137         column = gtk.TreeViewColumn('User ID', renderer, 
138                 text=models.AccountsModel.C_UID)
139         column.set_property("expand", True)
140         treeview.append_column(column)
141
142         #Column 2 (characters) for the treeview
143         column = gtk.TreeViewColumn('Characters', renderer, 
144                 markup=models.AccountsModel.C_CHARS)
145         column.set_property("expand", True)
146         treeview.append_column(column)
147
148
149     def new_account_clicked(self, window):
150         dialog = gtk.Dialog()
151     
152         #get the vbox to pack all the settings into
153         vbox = dialog.vbox
154     
155         dialog.set_transient_for(window)
156         dialog.set_title("New Account")
157
158         uidLabel = gtk.Label("User ID:")
159         uidLabel.set_justify(gtk.JUSTIFY_LEFT)
160         vbox.add(uidLabel)
161         
162         uidEntry = hildon.Entry(gtk.HILDON_SIZE_FINGER_HEIGHT)
163         uidEntry.set_placeholder("User ID")
164         uidEntry.set_property('is_focus', False)
165         
166         vbox.add(uidEntry)
167
168         apiLabel = gtk.Label("API key:")
169         apiLabel.set_justify(gtk.JUSTIFY_LEFT)
170         vbox.add(apiLabel)
171         
172         apiEntry = hildon.Entry(gtk.HILDON_SIZE_FINGER_HEIGHT)
173         apiEntry.set_placeholder("API Key")
174         apiEntry.set_property('is_focus', False)
175
176         vbox.add(apiEntry)
177        
178         ok_button = dialog.add_button(gtk.STOCK_OK, gtk.RESPONSE_OK)
179
180         dialog.show_all()
181         result = dialog.run()
182         
183         valid_credentials = False
184
185         while not valid_credentials:
186             if result == gtk.RESPONSE_OK:
187                 uid = uidEntry.get_text()
188                 # auth() fails if api_key has lower-case characters
189                 api_key = apiEntry.get_text().upper()
190             
191                 try:
192                     validation.uid(uid)
193                     validation.api_key(api_key)
194                 except validation.ValidationError, e:
195                     self.report_error(e.message)
196                     result = dialog.run()
197                 else:
198                     valid_credentials = True
199                     self.controller.add_account(uid, api_key)
200                     self.accounts_model.get_accounts()
201             else:
202                 break
203
204         dialog.destroy()
205
206
207     def report_error(self, error):
208         hildon.hildon_banner_show_information(self.win, '', error)
209     
210     def about_clicked(self, button):
211         dialog = gtk.AboutDialog()
212         dialog.set_website(self.controller.about_website)
213         dialog.set_website_label(self.controller.about_website)
214         dialog.set_name(self.controller.about_name)
215         dialog.set_authors(self.controller.about_authors)
216         dialog.set_comments(self.controller.about_text)
217         dialog.set_version(self.controller.app_version)
218         dialog.run()
219         dialog.destroy()
220
221     def add_label(self, text, box, markup=True, align="left", padding=1):
222         label = gtk.Label(text)
223         if markup:
224             label.set_use_markup(True)
225         if align == "left":
226             label.set_alignment(0, 0.5)
227
228         box.pack_start(label, False, False, padding)
229
230         return label
231
232 class mEveMonUI(BaseUI):
233
234     def __init__(self, controller):
235         self.controller = controller
236         gtk.set_application_name("mEveMon")
237
238         # create the main window
239         self.win = hildon.StackableWindow()
240         self.win.connect("destroy", self.controller.quit)
241         self.win.show_all()
242         hildon.hildon_gtk_window_set_progress_indicator(self.win, 1)
243
244         # Create menu
245         menu = self.create_menu(self.win)
246         # Attach menu to the window
247         self.win.set_app_menu(menu)
248
249         pannable_area = hildon.PannableArea()
250
251         character_win = CharacterSheetUI(self.controller)
252
253         # gtk.HILDON_UI_MODE_NORMAL -> not selection in the treeview
254         # gtk.HILDON_UI_MODE_EDIT -> selection in the treeview
255         treeview = hildon.GtkTreeView(gtk.HILDON_UI_MODE_NORMAL)
256         treeview.connect('row-activated', character_win.build_window)
257
258         self.char_model = models.CharacterListModel(self.controller)
259         treeview.set_model(self.char_model)
260         self.add_columns_to_treeview(treeview)
261
262         pannable_area.add(treeview)
263
264         self.win.add(pannable_area);
265         
266         self.win.show_all()
267
268         hildon.hildon_gtk_window_set_progress_indicator(self.win, 0)
269
270     def add_columns_to_treeview(self, treeview):
271         #Column 0 for the treeview
272         renderer = gtk.CellRendererPixbuf()
273         column = gtk.TreeViewColumn()
274         column.pack_start(renderer, True)
275         column.add_attribute(renderer, "pixbuf", 
276                 models.CharacterListModel.C_PORTRAIT)
277         treeview.append_column(column)
278
279         #Column 1 for the treeview
280         renderer = gtk.CellRendererText()
281         column = gtk.TreeViewColumn('Character Name', renderer, 
282                 text=models.CharacterListModel.C_NAME)
283         column.set_property("expand", True)
284         treeview.append_column(column)
285  
286     def refresh_clicked(self, button):
287         hildon.hildon_gtk_window_set_progress_indicator(self.win, 1)
288         self.char_model.get_characters()
289         hildon.hildon_gtk_window_set_progress_indicator(self.win, 0)
290     
291
292 class CharacterSheetUI(BaseUI):
293     UPDATE_INTERVAL = 1
294
295     def __init__(self, controller):
296         self.controller = controller
297         self.sheet = None
298         self.char_id = None
299         self.skills_model = None
300
301
302     def build_window(self, treeview, path, view_column):
303         # TODO: this is a really long and ugly function, split it up somehow
304
305         self.win = hildon.StackableWindow()
306         hildon.hildon_gtk_window_set_progress_indicator(self.win, 1)
307         self.win.show_all() 
308
309         # Create menu
310         # NOTE: we probably want a window-specific menu for this page, but the
311         # main appmenu works for now
312         menu = self.create_menu(self.win)
313         # Attach menu to the window
314         self.win.set_app_menu(menu)
315
316         pannable_area = hildon.PannableArea()
317
318         model = treeview.get_model()
319         miter = model.get_iter(path)
320         
321         # column 0 is the portrait, column 1 is name
322         char_name = model.get_value(miter, 1)
323         self.uid = model.get_value(miter, 2)
324         self.char_id = self.controller.char_name2id(char_name)
325
326         self.sheet = self.controller.get_char_sheet(self.uid, self.char_id)
327
328         self.win.set_title(char_name)
329
330
331         hbox = gtk.HBox(False, 0)
332         info_vbox = gtk.VBox(False, 0)
333
334         portrait = gtk.Image()
335         portrait.set_from_file(self.controller.get_portrait(char_name, 256))
336         portrait.show()
337
338         hbox.pack_start(portrait, False, False, 10)
339         hbox.pack_start(info_vbox, False, False, 5)
340
341         vbox = gtk.VBox(False, 0)
342         pannable_area.add_with_viewport(vbox)
343
344         vbox.pack_start(hbox, False, False, 0)
345
346         self.fill_info(info_vbox)
347         self.fill_stats(info_vbox)
348
349         separator = gtk.HSeparator()
350         vbox.pack_start(separator, False, False, 5)
351         separator.show()
352
353         
354         self.add_label("<big>Skill in Training:</big>", vbox, align="normal")
355
356         self.display_skill_in_training(vbox)
357
358         separator = gtk.HSeparator()
359         vbox.pack_start(separator, False, False, 0)
360         separator.show()
361         
362         self.add_label("<big>Skills:</big>", vbox, align="normal")
363
364         skills_treeview = hildon.GtkTreeView(gtk.HILDON_UI_MODE_NORMAL)
365         self.skills_model = models.CharacterSkillsModel(self.controller, self.char_id)
366         skills_treeview.set_model(self.skills_model)
367         self.add_columns_to_skills_view(skills_treeview)
368         vbox.pack_start(skills_treeview, False, False, 1)
369
370         self.win.add(pannable_area)
371         self.win.show_all()
372
373         hildon.hildon_gtk_window_set_progress_indicator(self.win, 0)
374         
375         # if we start the timer too early, get_is_topmost() returns False
376         self.timer = gobject.timeout_add_seconds(self.UPDATE_INTERVAL, self.update_live_sp)
377
378         self.win.connect("destroy", self.back)
379
380     def back(self, widget):
381         gobject.source_remove(self.timer)
382         gtk.Window.destroy(self.win)
383
384
385     def display_skill_in_training(self, vbox):
386         skill = self.controller.get_skill_in_training(self.uid, self.char_id)
387         
388         if skill.skillInTraining:
389
390             skilltree = self.controller.get_skill_tree()
391             
392             # I'm assuming that we will always find a skill with the skill ID
393             for group in skilltree.skillGroups:
394                 found_skill = group.skills.Get(skill.trainingTypeID, False)
395                 if found_skill:
396                     skill_name = found_skill.typeName
397                     break
398                 
399             self.add_label("%s <small>(Level %d)</small>" % (skill_name, skill.trainingToLevel),
400                     vbox, align="normal")
401             self.add_label("<small>start time: %s\t\tend time: %s</small>" 
402                     %(time.ctime(skill.trainingStartTime),
403                 time.ctime(skill.trainingEndTime)), vbox, align="normal")
404
405             progressbar = gtk.ProgressBar()
406             fraction_completed = (time.time() - skill.trainingStartTime) / \
407                     (skill.trainingEndTime - skill.trainingStartTime)
408
409             progressbar.set_fraction(fraction_completed)
410             align = gtk.Alignment(0.5, 0.5, 0.5, 0)
411             vbox.pack_start(align, False, False, 5)
412             align.show()
413             align.add(progressbar)
414             progressbar.show()
415         else:
416             self.add_label("<small>No skills are currently being trained</small>",
417                     vbox, align="normal")
418
419
420
421     def fill_info(self, box):
422         self.add_label("<big><big>%s</big></big>" % self.sheet.name, box)
423         self.add_label("<small>%s %s %s</small>" % (self.sheet.gender, 
424             self.sheet.race, self.sheet.bloodLine), box)
425         self.add_label("", box, markup=False)
426         self.add_label("<small><b>Corp:</b> %s</small>" % self.sheet.corporationName, box)
427         self.add_label("<small><b>Balance:</b> %s ISK</small>" % 
428                 util.comma(self.sheet.balance), box)
429
430         self.live_sp_val = self.controller.get_sp(self.uid, self.char_id)
431         self.live_sp = self.add_label("<small><b>Total SP:</b> %s</small>" %
432                 util.comma(int(self.live_sp_val)), box)
433         
434         self.spps = self.controller.get_spps(self.uid, self.char_id)[0]
435
436
437     def fill_stats(self, box):
438
439         atr = self.sheet.attributes
440
441         self.add_label("<small><b>I: </b>%d  <b>M: </b>%d  <b>C: </b>%d  " \
442                 "<b>P: </b>%d  <b>W: </b>%d</small>" % (atr.intelligence,
443                     atr.memory, atr.charisma, atr.perception, atr.willpower), box)
444
445
446     def add_columns_to_skills_view(self, treeview):
447         #Column 0 for the treeview
448         renderer = gtk.CellRendererText()
449         column = gtk.TreeViewColumn('Skill Name', renderer, 
450                 markup=models.CharacterSkillsModel.C_NAME)
451         column.set_property("expand", True)
452         treeview.append_column(column)
453         
454         #Column 1 for the treeview
455         column = gtk.TreeViewColumn('Rank', renderer, 
456                 markup=models.CharacterSkillsModel.C_RANK)
457         column.set_property("expand", True)
458         treeview.append_column(column)
459
460         #Column 2
461         column = gtk.TreeViewColumn('Points', renderer,
462                 markup=models.CharacterSkillsModel.C_SKILLPOINTS)
463         column.set_property("expand", True)
464         treeview.append_column(column)
465
466         #Column 3
467         column = gtk.TreeViewColumn('Level', renderer, 
468                 markup=models.CharacterSkillsModel.C_LEVEL)
469         column.set_property("expand", True)
470         treeview.append_column(column)
471
472
473     def refresh_clicked(self, button):
474         hildon.hildon_gtk_window_set_progress_indicator(self.win, 1)
475         self.skills_model.get_skills()
476         hildon.hildon_gtk_window_set_progress_indicator(self.win, 0)
477
478
479     def update_live_sp(self):
480         self.live_sp_val = self.live_sp_val + self.spps * self.UPDATE_INTERVAL
481         self.live_sp.set_label("<small><b>Total SP:</b> %s</small>" %
482                                 util.comma(int(self.live_sp_val)))
483
484         return True
485
486
487 if __name__ == "__main__":
488     main()
489