1 files changed, 185 insertions, 0 deletions
diff --git a/invest-applet/invest/applet.py b/invest-applet/invest/applet.py
new file mode 100644
index 00000000..c1689586
--- /dev/null
+++ b/invest-applet/invest/applet.py
@@ -0,0 +1,185 @@
+import os, time
+from os.path import *
+import mateapplet, gtk, gtk.gdk, mateconf, gobject
+gobject.threads_init()
+from gettext import gettext as _
+import mateconf
+
+import invest, invest.about, invest.chart, invest.preferences, invest.defs
+from invest.quotes import QuoteUpdater
+from invest.widgets import *
+
+gtk.window_set_default_icon_from_file(join(invest.ART_DATA_DIR, "invest_neutral.svg"))
+
+class InvestApplet:
+ def __init__(self, applet):
+ self.applet = applet
+ self.applet.setup_menu_from_file (
+ None, "Invest_Applet.xml",
+ None, [("About", self.on_about), 
+ ("Help", self.on_help),
+ ("Prefs", self.on_preferences),
+ ("Refresh", self.on_refresh)
+ ])
+
+ evbox = gtk.HBox()
+ self.applet_icon = gtk.Image()
+ self.set_applet_icon(0)
+ self.applet_icon.show()
+ evbox.add(self.applet_icon)
+ self.applet.add(evbox)
+ self.applet.connect("button-press-event",self.button_clicked)
+ self.applet.show_all()
+ self.new_ilw()
+
+ def new_ilw(self):
+ self.quotes_updater = QuoteUpdater(self.set_applet_icon,
+ self.set_applet_tooltip)
+ self.investwidget = InvestWidget(self.quotes_updater)
+ self.ilw = InvestmentsListWindow(self.applet, self.investwidget)
+
+ def reload_ilw(self):
+ self.ilw.destroy()
+ self.new_ilw()
+
+ def button_clicked(self, widget,event):
+ if event.type == gtk.gdk.BUTTON_PRESS and event.button == 1:
+ # Three cases...
+ if len (invest.STOCKS) == 0:
+ # a) We aren't configured yet
+ invest.preferences.show_preferences(self, _("<b>You have not entered any stock information yet</b>"))
+ self.reload_ilw()
+ elif not self.quotes_updater.quotes_valid:
+ # b) We can't get the data (e.g. offline)
+ alert = gtk.MessageDialog(buttons=gtk.BUTTONS_CLOSE)
+ alert.set_markup(_("<b>No stock quotes are currently available</b>"))
+ alert.format_secondary_text(_("The server could not be contacted. The computer is either offline or the servers are down. Try again later."))
+ alert.run()
+ alert.destroy()
+ else:
+ # c) Everything is normal: pop-up the window
+ self.ilw.toggle_show()
+ 
+ def on_about(self, component, verb):
+ invest.about.show_about()
+ 
+ def on_help(self, component, verb):
+ invest.help.show_help()
+
+ def on_preferences(self, component, verb):
+ invest.preferences.show_preferences(self)
+ self.reload_ilw()
+ 
+ def on_refresh(self, component, verb):
+ self.quotes_updater.refresh()
+
+ def set_applet_icon(self, change):
+ if change == 1:
+ pixbuf = gtk.gdk.pixbuf_new_from_file_at_size(join(invest.ART_DATA_DIR, "invest-22_up.png"), -1,-1)
+ elif change == 0:
+ pixbuf = gtk.gdk.pixbuf_new_from_file_at_size(join(invest.ART_DATA_DIR, "invest-22_neutral.png"), -1,-1)
+ else:
+ pixbuf = gtk.gdk.pixbuf_new_from_file_at_size(join(invest.ART_DATA_DIR, "invest-22_down.png"), -1,-1)
+ self.applet_icon.set_from_pixbuf(pixbuf)
+ 
+ def set_applet_tooltip(self, text):
+ self.applet_icon.set_tooltip_text(text)
+
+class InvestmentsListWindow(gtk.Window):
+ def __init__(self, applet, list):
+ gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
+ self.set_type_hint(gtk.gdk.WINDOW_TYPE_HINT_DOCK)
+ self.stick()
+ self.set_resizable(False)
+ self.set_border_width(6)
+ 
+ self.applet = applet # this is the widget we want to align with
+ self.alignment = self.applet.get_orient ()
+ 
+ self.add(list)
+ list.show()
+
+ # boolean variable that identifies if the window is visible
+ # show/hide is triggered by left-clicking on the applet
+ self.hidden = True
+
+ def toggle_show(self):
+ if self.hidden == True:
+ self.update_position()
+ self.show()
+ self.hidden = False
+ elif self.hidden == False:
+ self.hide()
+ self.hidden = True
+
+ def update_position (self):
+ """
+ Calculates the position and moves the window to it.
+ """
+ self.realize()
+
+ # Get our own dimensions & position
+ #(wx, wy) = self.get_origin()
+ (ax, ay) = self.applet.window.get_origin()
+
+ (ww, wh) = self.get_size ()
+ (aw, ah) = self.applet.window.get_size ()
+
+ screen = self.applet.window.get_screen()
+ monitor = screen.get_monitor_geometry (screen.get_monitor_at_window (self.applet.window))
+
+ if self.alignment == mateapplet.ORIENT_LEFT:
+ x = ax - ww
+ y = ay
+
+ if (y + wh > monitor.y + monitor.height):
+ y = monitor.y + monitor.height - wh
+
+ if (y < 0):
+ y = 0
+ 
+ if (y + wh > monitor.height / 2):
+ gravity = gtk.gdk.GRAVITY_SOUTH_WEST
+ else:
+ gravity = gtk.gdk.GRAVITY_NORTH_WEST
+ 
+ elif self.alignment == mateapplet.ORIENT_RIGHT:
+ x = ax + aw
+ y = ay
+
+ if (y + wh > monitor.y + monitor.height):
+ y = monitor.y + monitor.height - wh
+ 
+ if (y < 0):
+ y = 0
+ 
+ if (y + wh > monitor.height / 2):
+ gravity = gtk.gdk.GRAVITY_SOUTH_EAST
+ else:
+ gravity = gtk.gdk.GRAVITY_NORTH_EAST
+
+ elif self.alignment == mateapplet.ORIENT_DOWN:
+ x = ax
+ y = ay + ah
+
+ if (x + ww > monitor.x + monitor.width):
+ x = monitor.x + monitor.width - ww
+
+ if (x < 0):
+ x = 0
+
+ gravity = gtk.gdk.GRAVITY_NORTH_WEST
+ elif self.alignment == mateapplet.ORIENT_UP:
+ x = ax
+ y = ay - wh
+
+ if (x + ww > monitor.x + monitor.width):
+ x = monitor.x + monitor.width - ww
+
+ if (x < 0):
+ x = 0
+
+ gravity = gtk.gdk.GRAVITY_SOUTH_WEST
+
+ self.move(x, y)
+ self.set_gravity(gravity)