vcs.maemo.org Git - stockthis/blob - stockthis.py

   1 #!/usr/bin/env python2.5
   2 # -*- coding: UTF8 -*-
   3 # Copyright (C) 2008 by Daniel Martin Yerga
   4 # <dyerga@gmail.com>
   5 # This program 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 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program 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, write to the Free Software
  17 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18 #
  19 # StocksPy: Application to get stocks data from Yahoo Finance.
  20 # Version 0.1
  21 #
  22
  23 import urllib2
  24 import gtk, gobject
  25 try:
  26     import hildon
  27     HILDON = True
  28 except:
  29     HILDON = False
  30
  31 try:
  32     import osso
  33     OSSO = True
  34     osso_c = osso.Context("net.yerga.stockthis", "0.1", False)
  35 except:
  36     OSSO = False
  37
  38 from marketdata import markets, idmarket, localmarkets, localids
  39
  40 #TODO: detect if running in Fremantle
  41 FREMANTLE=True
  42
  43 #detect if is ran locally or not
  44 import sys
  45 runningpath = sys.path[0]
  46
  47 if '/usr/share' in runningpath:
  48     running_locally = False
  49 else:
  50     running_locally = True
  51
  52 if running_locally:
  53     imgdir = 'pixmaps/'
  54 else:
  55     imgdir = '/usr/share/stockthis/pixmaps/'
  56
  57 loading_img = imgdir + 'loading.gif'
  58
  59 gtk.gdk.threads_init()
  60
  61 class StocksPy:
  62
  63     def __init__(self):
  64         if HILDON:
  65             self.program = hildon.Program()
  66             self.program.__init__()
  67             gtk.set_application_name('')
  68             if FREMANTLE:
  69                 self.window = hildon.StackableWindow()
  70             else:
  71                 self.window = hildon.Window()
  72             self.program.add_window(self.window)
  73         else:
  74             self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
  75
  76         self.window.set_default_size(800, 480)
  77         self.window.set_title('StocksPy')
  78         self.window.connect("destroy", gtk.main_quit)
  79         self.window.connect("key-press-event", self.on_key_press)
  80         self.window.connect("window-state-event", self.on_window_state_change)
  81         self.window_in_fullscreen = False
  82
  83         if HILDON:
  84             if FREMANTLE:
  85                 menu = hildon.AppMenu()
  86                 self.window.set_main_menu(menu)
  87                 about_menu = gtk.Button("About")
  88             else:
  89                 menu = gtk.Menu()
  90                 self.window.set_menu(menu)
  91                 about_menu = gtk.MenuItem("About")
  92                 about_menu.set_size_request(-1, 55)
  93
  94             about_menu.connect("activate", self.on_about)
  95             menu.append(about_menu)
  96             menu.show_all()
  97         else:
  98             #TODO: create a gtk.menubar
  99             pass
 100
 101
 102         vbox = gtk.VBox()
 103
 104         self.toolbar = gtk.HBox()
 105         self.toolbar.set_property("no-show-all", True)
 106         self.toolbar.hide()
 107         self.toolbar.set_homogeneous(True)
 108         self.toolbar.set_size_request(-1, 55)
 109
 110         self.markets_btn = gtk.Button('Markets')
 111         self.markets_btn.connect("clicked", self.create_markets_view)
 112         self.markets_btn.set_property("can_focus", False)
 113
 114         self.components_btn = gtk.Button('Components')
 115         self.components_btn.connect("clicked", self.show_components_view)
 116         self.components_btn.set_property("can_focus", False)
 117
 118         self.graph_btn = gtk.Button('Graphs')
 119         self.graph_btn.connect("clicked", self.create_graphs_view)
 120         self.graph_btn.set_property("can_focus", False)
 121
 122         self.refresh_btn = gtk.Button('Refresh')
 123         self.refresh_btn.connect("clicked", self.refresh_stock_data)
 124         self.refresh_btn.set_property("can_focus", False)
 125
 126         self.quotes_btn = gtk.Button('Quotes')
 127         self.quotes_btn.connect("clicked", self.show_quotes_view)
 128         self.quotes_btn.set_property("can_focus", False)
 129
 130         self.toolbar.pack_start(self.markets_btn)
 131         self.toolbar.pack_start(self.components_btn)
 132         self.toolbar.pack_start(self.graph_btn)
 133         self.toolbar.pack_start(self.quotes_btn)
 134         self.toolbar.pack_start(self.refresh_btn)
 135
 136         self.mainbox = gtk.VBox()
 137
 138
 139         if FREMANTLE:
 140             self.swin = hildon.PannableArea()
 141         else:
 142             self.swin = gtk.ScrolledWindow()
 143             if HILDON:
 144                 hildon.hildon_helper_set_thumb_scrollbar(self.swin, True)
 145             self.swin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
 146         self.swin.add_with_viewport(self.mainbox)
 147
 148         vbox.pack_start(self.swin, True, True, 0)
 149         vbox.pack_start(gtk.HSeparator(), False, False, 5)
 150         vbox.pack_start(self.toolbar, False, False, 0)
 151
 152         self.create_markets_view(self.mainbox)
 153         self.window.add(vbox)
 154
 155         self.window.show_all()
 156
 157     def create_markets_view(self, widget):
 158         names = markets
 159         ids = idmarket
 160         self.toolbar.hide()
 161
 162         actual_view = self.mainbox.get_children()
 163         if len(actual_view) > 0:
 164             for a_widget in actual_view:
 165                 a_widget.destroy()
 166
 167         self.marketsbox = gtk.VBox()
 168         lnames = len(names)
 169
 170         for i in range(lnames):
 171             button = gtk.Button(names[i])
 172             button.connect("clicked", self.create_components_view, ids[i])
 173             button.set_size_request(-1, 65)
 174             button.set_property("can_focus", False)
 175             self.marketsbox.pack_start(button, True, True, 2)
 176
 177         self.mainbox.pack_start(self.marketsbox, True, True, 2)
 178         self.mainbox.show_all()
 179
 180     def show_components_view(self, widget):
 181         kind = self.market_id
 182         self.create_components_view(widget, kind)
 183
 184     def create_components_view(self, widget, kind):
 185         actual_view = self.mainbox.get_children()
 186         for i in actual_view:
 187             i.destroy()
 188
 189         self.market_id = kind
 190         self.toolbar.show()
 191         self.components_btn.hide()
 192         self.markets_btn.show()
 193         self.refresh_btn.hide()
 194         self.graph_btn.hide()
 195         self.quotes_btn.hide()
 196
 197         names = localmarkets[idmarket.index(kind)]
 198         ids = localids[idmarket.index(kind)]
 199
 200         self.compbox = gtk.VBox()
 201
 202         self.components_trv = gtk.TreeView()
 203         selection = self.components_trv.get_selection()
 204         selection.connect("changed", self.changed_selection, ids)
 205         self.components_trv.set_rubber_banding(True)
 206         self.components_trv.set_headers_visible(False)
 207         self.components_model = self.__create_components_model()
 208         self.components_trv.set_model(self.components_model)
 209         self._components_trv_columns(self.components_trv)
 210         self.add_initial_components(names, ids)
 211
 212         self.compbox.pack_start(self.components_trv, True, True, 0)
 213
 214         self.mainbox.pack_start(self.compbox, True, True, 0)
 215         self.mainbox.show_all()
 216
 217     def changed_selection(self, widget, ids):
 218         n, i, m = self.get_selected_from_treeview(self.components_trv, self.components_model, 1)
 219         if n is not None:
 220             self.create_quotes_view(widget, n)
 221
 222     def __create_components_model(self):
 223         lstore = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
 224         return lstore
 225
 226     def _components_trv_columns(self, treeview):
 227         renderer = gtk.CellRendererText()
 228         renderer.set_property('scale', 1.6)
 229         #renderer.set_property("background", 'lightgray')
 230         renderer.set_property("xpad", 15)
 231         column = gtk.TreeViewColumn('Name', renderer, text=0)
 232         column.set_expand(True)
 233         treeview.append_column(column)
 234
 235         renderer = gtk.CellRendererText()
 236         column = gtk.TreeViewColumn('IDS', renderer, text=1)
 237         column.set_visible(False)
 238         treeview.append_column(column)
 239
 240     def add_initial_components(self, kind, ids):
 241         for i in range(len(kind)):
 242             niter = self.components_model.append()
 243             self.components_model.set(niter, 0, kind[i], 1, ids[i])
 244
 245     def get_selected_from_treeview(self, treeview, model, setting):
 246         selection = treeview.get_selection()
 247         selected_model, selected_iter = selection.get_selected()
 248         if selected_iter:
 249             selected_name = model.get_value(selected_iter, setting)
 250         else:
 251             selected_name = None
 252         return selected_name, selected_iter, selected_model
 253
 254     def show_quotes_view(self, widget):
 255         kind = self.stocks_id
 256         self.create_quotes_view(widget, kind)
 257
 258     def create_quotes_view(self, widget, kind):
 259         self.stocks_id = kind
 260         self.toolbar.show()
 261         self.components_btn.show()
 262         self.markets_btn.show()
 263         self.refresh_btn.show()
 264         self.graph_btn.show()
 265         self.quotes_btn.hide()
 266
 267         #actual_view = self.mainbox.get_children()
 268         #actual_view[0].destroy()
 269
 270         try:
 271             self.graphbox.destroy()
 272         except:
 273             pass
 274
 275         for i in range(len(localids)):
 276             if kind in localids[i]:
 277                 ind1, ind2 = i, localids[i].index(kind)
 278
 279         self.quotesbox = gtk.VBox()
 280
 281         label1 = gtk.Label('')
 282         label1.set_markup('<b><big>'+localmarkets[ind1][ind2]+'</big></b>')
 283         color = gtk.gdk.color_parse("#03A5FF")
 284         label1.modify_fg(gtk.STATE_NORMAL, color)
 285
 286         self.databox = gtk.VBox()
 287
 288         self.imagebox = gtk.VBox()
 289         self.dataimg = gtk.Image()
 290         self.imagebox.pack_start(self.dataimg)
 291
 292         self.show_data(kind)
 293
 294         hbox1 = gtk.HBox()
 295
 296         label2 = gtk.Label('')
 297         label2.set_markup('<b><big>Price:</big></b>')
 298         self.lprice = gtk.Label('')
 299
 300         hbox1.pack_start(label2, False, False, 50)
 301         hbox1.pack_start(self.lprice, False, False, 185)
 302
 303         hbox2 = gtk.HBox()
 304
 305         label4 = gtk.Label('')
 306         label4.set_markup('<b><big>Change:</big></b>')
 307         self.lchange = gtk.Label('')
 308         self.lpercent = gtk.Label('')
 309
 310         hbox2.pack_start(label4, False, False, 50)
 311         hbox2.pack_start(self.lchange  , False, False, 145)
 312         hbox2.pack_start(self.lpercent, False, False, 0)
 313
 314         hbox3 = gtk.HBox()
 315
 316         label7 = gtk.Label('')
 317         label7.set_markup('<b><big>Volume:</big></b>')
 318         self.lvolume = gtk.Label('')
 319
 320         hbox3.pack_start(label7, False, False, 50)
 321         hbox3.pack_start(self.lvolume  , False, False, 145)
 322
 323         hbox4 = gtk.HBox()
 324
 325         label9 = gtk.Label('')
 326         label9.set_markup('<b><big>52 week high:</big></b>')
 327         self.l52whigh = gtk.Label('')
 328
 329         hbox4.pack_start(label9, False, False, 50)
 330         hbox4.pack_start(self.l52whigh , False, False, 55)
 331
 332         hbox5 = gtk.HBox()
 333
 334         label11 = gtk.Label('')
 335         label11.set_markup('<b><big>52 week low:</big></b>')
 336         self.l52wlow = gtk.Label('')
 337
 338         hbox5.pack_start(label11, False, False, 50)
 339         hbox5.pack_start(self.l52wlow , False, False, 70)
 340
 341         self.databox.pack_start(hbox1, True, True, 0)
 342         self.databox.pack_start(hbox2, True, True, 0)
 343         self.databox.pack_start(hbox3, True, True, 0)
 344         self.databox.pack_start(hbox4, True, True, 0)
 345         self.databox.pack_start(hbox5, True, True, 0)
 346
 347         self.quotesbox.pack_start(label1, False, False, 0)
 348         self.quotesbox.pack_start(gtk.HSeparator(), False, False, 0)
 349         self.quotesbox.pack_start(self.databox, True, True, 0)
 350         self.quotesbox.pack_start(self.imagebox, True, True, 0)
 351
 352         self.mainbox.pack_start(self.quotesbox, True, True, 0)
 353         self.mainbox.show_all()
 354         self.compbox.hide()
 355         self.databox.hide()
 356
 357     def show_data(self, kind):
 358         import thread
 359         self.dataimg.set_from_file(loading_img)
 360         self.imagebox.show()
 361         self.databox.hide()
 362         thread.start_new_thread(self.get_data, (kind,))
 363
 364     def get_data(self, kind):
 365         self.graph_btn.show()
 366         self.refresh_btn.show()
 367         self.quotes_btn.hide()
 368
 369         from ystockquote import ystockquote as yt
 370         data = yt.get_all(kind)
 371
 372         try:
 373             ch_percent = 100.0 * float(data['change'])/float(data['price'])
 374         except ValueError:
 375             ch_percent = 0.0
 376
 377         self.lprice.set_label(data['price'])
 378         self.lchange.set_label(data['change'])
 379         self.lpercent.set_label('%6.2f %%' % ch_percent)
 380
 381         if '-' in data['change']:
 382             color = gtk.gdk.color_parse("#FF0000")
 383         else:
 384             color = gtk.gdk.color_parse("#16EB78")
 385
 386         self.lpercent.modify_fg(gtk.STATE_NORMAL, color)
 387         self.lchange.modify_fg(gtk.STATE_NORMAL, color)
 388
 389         self.lvolume.set_label(data['volume'])
 390         self.l52whigh.set_label(data['52_week_high'])
 391         self.l52wlow.set_label(data['52_week_low'])
 392
 393         self.databox.show_all()
 394         self.imagebox.hide()
 395
 396     def refresh_stock_data(self, widget):
 397         self.show_data(self.stocks_id)
 398
 399     def create_graphs_view(self, widget):
 400         self.graph_btn.hide()
 401         self.refresh_btn.hide()
 402         self.quotes_btn.show()
 403
 404         self.graph = gtk.Image()
 405
 406         kind = self.stocks_id
 407
 408         self.show_graph(None, '1d', kind)
 409
 410         for i in range(len(localids)):
 411             if kind in localids[i]:
 412                 ind1, ind2 = i, localids[i].index(kind)
 413
 414         #self.mainbox.destroy()
 415         #self.mainbox = gtk.VBox()
 416         #self.swin.add_with_viewport(self.mainbox)
 417
 418         actual_view = self.mainbox.get_children()
 419         for a_widget in actual_view:
 420             a_widget.destroy()
 421
 422         self.graphbox = gtk.VBox()
 423
 424         label1 = gtk.Label(localmarkets[ind1][ind2])
 425         color = gtk.gdk.color_parse("#03A5FF")
 426         label1.modify_fg(gtk.STATE_NORMAL, color)
 427
 428         hbox1 = gtk.HBox()
 429         hbox1.set_size_request(-1, 55)
 430         hbox1.set_homogeneous(True)
 431
 432         self.btn1d = gtk.Button('1d')
 433         self.btn1d.set_property("can_focus", False)
 434         self.btn1d.connect("clicked", self.show_graph, '1d', kind)
 435         self.btn5d = gtk.Button('5d')
 436         self.btn5d.set_property("can_focus", False)
 437         self.btn5d.connect("clicked", self.show_graph, '5d', kind)
 438         self.btn3m = gtk.Button('3m')
 439         self.btn3m.set_property("can_focus", False)
 440         self.btn3m.connect("clicked", self.show_graph, '3m', kind)
 441         self.btn6m = gtk.Button('6m')
 442         self.btn6m.set_property("can_focus", False)
 443         self.btn6m.connect("clicked", self.show_graph, '6m', kind)
 444         self.btn1y = gtk.Button('1y')
 445         self.btn1y.set_property("can_focus", False)
 446         self.btn1y.connect("clicked", self.show_graph, '1y', kind)
 447         self.btn2y = gtk.Button('2y')
 448         self.btn2y.set_property("can_focus", False)
 449         self.btn2y.connect("clicked", self.show_graph, '2y', kind)
 450         self.btn5y = gtk.Button('5y')
 451         self.btn5y.set_property("can_focus", False)
 452         self.btn5y.connect("clicked", self.show_graph, '5y', kind)
 453         self.btnmax = gtk.Button('max')
 454         self.btnmax.set_property("can_focus", False)
 455         self.btnmax.connect("clicked", self.show_graph, 'max', kind)
 456
 457         hbox1.pack_start(self.btn1d)
 458         hbox1.pack_start(self.btn5d)
 459         hbox1.pack_start(self.btn3m)
 460         hbox1.pack_start(self.btn6m)
 461         hbox1.pack_start(self.btn1y)
 462         hbox1.pack_start(self.btn2y)
 463         hbox1.pack_start(self.btn5y)
 464         hbox1.pack_start(self.btnmax)
 465
 466         self.graphbox.pack_start(label1, False, False, 2)
 467         self.graphbox.pack_start(hbox1, False, False, 0)
 468         self.graphbox.pack_start(self.graph, True, True, 5)
 469
 470         self.mainbox.pack_start(self.graphbox, False, False, 2)
 471         self.mainbox.show_all()
 472
 473     def show_graph(self, widget, option, kind):
 474         import thread
 475         self.graph.set_from_file(loading_img)
 476         thread.start_new_thread(self.get_graph_data, (option, kind))
 477
 478     def get_graph_data(self, option, kind):
 479         if option == '1d':
 480             url = 'http://uk.ichart.yahoo.com/b?s=%s' % kind
 481         elif option == '5d':
 482             url = 'http://uk.ichart.yahoo.com/w?s=%s' % kind
 483         elif option == '3m':
 484             url = 'http://chart.finance.yahoo.com/c/3m/s/%s' % kind.lower()
 485         elif option == '6m':
 486             url = 'http://chart.finance.yahoo.com/c/6m/s/%s' % kind.lower()
 487         elif option == '1y':
 488             url = 'http://chart.finance.yahoo.com/c/1y/s/%s' % kind.lower()
 489         elif option == '2y':
 490             url = 'http://chart.finance.yahoo.com/c/2y/s/%s' % kind.lower()
 491         elif option == '5y':
 492             url = 'http://chart.finance.yahoo.com/c/5y/s/%s' % kind.lower()
 493         elif option == 'max':
 494             url = 'http://chart.finance.yahoo.com/c/my/s/%s' % kind.lower()
 495
 496         myimg = urllib2.urlopen(url)
 497         imgdata=myimg.read()
 498
 499         pbl = gtk.gdk.PixbufLoader()
 500         pbl.write(imgdata)
 501
 502         pbuf = pbl.get_pixbuf()
 503         pbl.close()
 504
 505         self.graph.set_from_pixbuf(pbuf)
 506
 507     #Functions for fullscreen
 508     def on_window_state_change(self, widget, event, *args):
 509         if event.new_window_state & gtk.gdk.WINDOW_STATE_FULLSCREEN:
 510             self.window_in_fullscreen = True
 511         else:
 512             self.window_in_fullscreen = False
 513
 514     #F6 fullscreen, F7 bigger font, F8 smaller font
 515     def on_key_press(self, widget, event, *args):
 516         if event.keyval == gtk.keysyms.F6:
 517             if self.window_in_fullscreen:
 518                 self.window.unfullscreen ()
 519             else:
 520                 self.window.fullscreen ()
 521
 522     def on_about(self, widget):
 523         dialog = gtk.AboutDialog()
 524         dialog.set_name("StockThis")
 525         dialog.set_version("0.1")
 526         dialog.set_copyright("Copyright © 2009")
 527         dialog.set_website("http://stockthis.garage.maemo.org")
 528         dialog.set_authors(["Daniel Martin Yerga <dyerga@gmail.com>"])
 529         logo = gtk.gdk.pixbuf_new_from_file(imgdir + "stockthis.png")
 530         dialog.set_logo(logo)
 531         dialog.set_license("This program is released under the GNU\nGeneral Public License. Please visit \nhttp://www.gnu.org/copyleft/gpl.html\nfor details.")
 532         dialog.set_artists(["Logo by Daniel Martin Yerga"])
 533         dialog.run()
 534         dialog.destroy()
 535
 536 if __name__ == "__main__":
 537     stockspy = StocksPy()
 538     gtk.gdk.threads_enter()
 539     gtk.main()
 540     gtk.gdk.threads_leave()
 541