X-Git-Url: http://git.maemo.org/git/?p=watersofshiloah;a=blobdiff_plain;f=src%2Fimagestore.py;h=0cb481347d12fd491195aa5bf6f073dc78f44c13;hp=e5e351ce50d51e836a7c8846f2cb0bf9dbeb54d5;hb=bf983084fcb46a045684cb62014a0693527907e3;hpb=a97a3a7ab405b56fdce94015e555a469d58b012b

diff --git a/src/imagestore.py b/src/imagestore.py
index e5e351c..0cb4813 100644
--- a/src/imagestore.py
+++ b/src/imagestore.py
@@ -1,42 +1,138 @@
+from __future__ import with_statement
+
 import os
+import logging
 
 import cairo
 import gtk
 
+import browser_emu
+from util import go_utils
+import util.misc as misc_utils
+
+
+_moduleLogger = logging.getLogger(__name__)
+
 
 class ImageStore(object):
 
 	STORE_LOOKUP = {
 		"next": "next.png",
 		"prev": "prev.png",
+		"home": "home.png",
 		"pause": "pause.png",
 		"play": "play.png",
 		"stop": "stop.png",
-		"generic_background": "radiobackground_01.png",
-		"night_temple_background": "radiobackground_02.png",
-		"day_temple_background": "radiobackground_03.png",
-		"presidency_background": "radiobackground_04.png",
-		"scriptures_background": "radiobackground_05.png",
-		"conference": "conference.png",
+		"pause_pressed": "pausepressed.png",
+		"play_pressed": "playpressed.png",
+		"stop_pressed": "stoppressed.png",
+
+		"small_next": "small_next.png",
+		"small_prev": "small_prev.png",
+		"small_home": "small_home.png",
+		"small_pause": "small_pause.png",
+		"small_play": "small_play.png",
+		"small_stop": "small_stop.png",
+		"small_pause_pressed": "small_pausepressed.png",
+		"small_play_pressed": "small_playpressed.png",
+		"small_stop_pressed": "small_stoppressed.png",
+
+		"loading": "loading.gif",
+
+		"radio_header": "radio_header.png",
+		"conference_background": "conference_bg.png",
+		"magazine_background": "magazine_bg.png",
+		"scripture_background": "scripture_bg.png",
+
+		"conferences": "conference.png",
 		"magazines": "magazines.png",
-		"more": "more.png",
 		"mormonmessages": "mormonmessages.png",
 		"radio": "radio.png",
 		"scriptures": "scriptures.png",
+
+		"more": "more.png",
 		"icon": "icon.png",
+		"nomagazineimage": "nomagazineimage.png",
 	}
 
 	def __init__(self, storePath, cachePath):
 		self._storePath = storePath
 		self._cachePath = cachePath
 
-	def get_surface_from_store(self, image):
-		path = os.path.join(self._storePath, image)
+		self._browser = browser_emu.MozillaEmulator()
+		self._downloader = go_utils.AsyncPool()
+
+	def start(self):
+		self._downloader.start()
+
+	def stop(self):
+		self._downloader.stop()
+
+	def get_surface_from_store(self, imageName):
+		path = os.path.join(self._storePath, imageName)
 		image = cairo.ImageSurface.create_from_png(path)
 		return image
 
-	def get_image_from_store(self, image):
-		path = os.path.join(self._storePath, image)
+	def get_image_from_store(self, imageName):
+		path = os.path.join(self._storePath, imageName)
 		image = gtk.Image()
 		image.set_from_file(path)
 		return image
+
+	def set_image_from_store(self, image, imageName):
+		path = os.path.join(self._storePath, imageName)
+		image.set_from_file(path)
+		return image
+
+	def get_pixbuf_from_store(self, imageName):
+		path = os.path.join(self._storePath, imageName)
+		return gtk.gdk.pixbuf_new_from_file(path)
+
+	def get_pixbuf_from_url(self, url, on_success, on_error):
+		# @ todo Test bad image for both paths
+		filepath = self._url_to_cache(url)
+		if os.path.exists(filepath):
+			pix = gtk.gdk.pixbuf_new_from_file(filepath)
+			try:
+				on_success(pix)
+			except Exception:
+				pass
+			doDownload = False
+		else:
+			doDownload = True
+
+		if doDownload:
+			self._get_image(
+				url,
+				lambda filepath: on_success(gtk.gdk.pixbuf_new_from_file(filepath)),
+				on_error,
+			)
+
+	def get_pixbuf_animation_from_store(self, imageName):
+		path = os.path.join(self._storePath, imageName)
+		return gtk.gdk.PixbufAnimation(path)
+
+	def _get_image(self, url, on_success, on_error):
+		self._downloader.add_task(
+			self._browser.download,
+			(url, ),
+			{},
+			lambda image: self._on_get_image(url, image, on_success, on_error),
+			on_error,
+		)
+
+	@misc_utils.log_exception(_moduleLogger)
+	def _on_get_image(self, url, image, on_success, on_error):
+		try:
+			filepath = self._url_to_cache(url)
+			_moduleLogger.info("Saved %s" % filepath)
+			with open(filepath, "wb") as f:
+				f.write(image)
+			on_success(filepath)
+		except Exception, e:
+			on_error(e)
+
+	def _url_to_cache(self, url):
+		filename = url.rsplit("/", 1)[-1]
+		filepath = os.path.join(self._cachePath, filename)
+		return filepath