fixed bug, added new debs and tarball
[comic-widget] / src / usr / lib / hildon-desktop / comicwidget.py
index 4b6239d..5838299 100644 (file)
@@ -11,12 +11,18 @@ import osso
 import cairo
 import datetime
 import shutil
+import sys
 
 supports_alpha = False
 
+# LOGGING!
+# sys.stdout = open('/home/user/.comic-widget/output_log.txt', 'a')
+# sys.stderr = open('/home/user/.comic-widget/error_log.txt', 'a')
+# print "Start logging!"
+
 # constants. dbfile is the location of the csv
 # comiccache is the location of the images
-APP_VERSION = "0.3.0"
+APP_VERSION = "0.3.2"
 
 basedbdir = "/opt/comic-widget/db/"
 imagedir = "/opt/comic-widget/images/"
@@ -26,7 +32,13 @@ comiccache = "/home/user/MyDocs/.comics/"
 comics = {"xkcd":{"name":"xkcd","link":"http://xkcd.org/","start":666,"dbfile":dbdir + "comicdb.xkcd.csv"},
                "sinfest":{"name":"Sinfest","link":"http://sinfest.com/","start":3400,"dbfile":dbdir + "comicdb.sinfest.csv"},
                "phd":{"name":"PHD Comics","link":"http://www.phdcomics.com/","start":1240,"dbfile":dbdir + "comicdb.phd.csv"},
-               "dilbert":{"name":"Dilbert","link":"http://dilbert.com.com/","start":"2009-01-01","dbfile":dbdir + "comicdb.dilbert.csv"}}
+               "dilbert":{"name":"Dilbert","link":"http://dilbert.com/","start":"2009-01-01","dbfile":dbdir + "comicdb.dilbert.csv"},
+               "cyanide":{"name":"Cyanide&Happiness","link":"http://explosm.com/","start":"1920","dbfile":dbdir + "comicdb.cyanide.csv"},
+               }
+defaults = {}
+previous = False
+next = False
+
 
 # handling of the comics
 class ComicDb():
@@ -45,6 +57,8 @@ class ComicDb():
                for row in dbr:
                        self.db.insert(0,row)
                dbf.close()
+               if len(self.db) == 0:
+                       self.refresh()
                self.currentcomic = 0
 
 
@@ -74,6 +88,7 @@ class ComicDb():
                                dbf = open(self.dbfile, 'w')
                                dbf.write('comic,id,link,url,filename,title\n')
                                dbf.close()
+                               
                        if os.path.isfile(self.dbfile) == True:
                                dbf = open(self.dbfile, 'r')
                                return dbf
@@ -84,10 +99,20 @@ class ComicDb():
        def get_comic(self):
                print str(self.currentcomic)
                if self.currentcomic < 0:
-                       self.refresh()
                        self.currentcomic = 0
+                       if len(self.db) > 0:
+                               fetchid = self.db[self.currentcomic]['id']
+                       else:
+                               fetchid = comics[self.comic]['start']
+                       self.fetch_newer(self.comic, fetchid)
+                       self.refresh()
                if  len(self.db) < (self.currentcomic + 1):
-                       self.currentcomic -= 1
+                       self.currentcomic = len(self.db) - 1
+                       if len(self.db) > 0:
+                               fetchid = self.db[self.currentcomic]['id']
+                       else:
+                               fetchid = comics[self.comic]['start']
+                       self.fetch_earlier(self.comic, fetchid)
                        self.refresh()
                        self.currentcomic = len(self.db) - 1
                        
@@ -147,12 +172,13 @@ class ComicDb():
                dbf.close()
 
        def refresh(self):
-               if len(self.db) < 1:
-                       self.fetch_latest_std(self.comic, self.start)
-               elif self.currentcomic == 0 or self.currentcomic < 0:
-                       self.fetch_latest_std(self.comic, self.db[0]['id'])
-               elif self.currentcomic == (len(self.db) - 1):
-                       self.fetch_earlier(self.comic, self.db[self.currentcomic]['id'])
+       #       if len(self.db) < 1:
+       #               self.currentcomic = -1
+       #               self.fetch_latest_std(self.comic, self.start)
+       #       elif self.currentcomic == 0 or self.currentcomic < 0:
+       #               self.fetch_latest_std(self.comic, self.db[0]['id'])
+       #       elif self.currentcomic == (len(self.db) - 1):
+       #               self.fetch_earlier(self.comic, self.db[self.currentcomic]['id'])
                dbf = open(self.dbfile, 'r')
                dbr = csv.DictReader(dbf)
                self.db = []
@@ -160,8 +186,14 @@ class ComicDb():
                        self.db.insert(0,row)
                dbf.close()
 
+
+# fetch earlier
        def fetch_earlier(self, comic, earliest):
-               if len(earliest) == 10:
+               print "fetch before, " + comic + " earliest"
+               if comic == "cyanide":
+                       comicid = self.get_prev_id(comic,earliest)
+                       print "got " + comicid + " as the one before current..."
+               elif len(earliest) == 10:
                        # date id.
                        dt = string.split(earliest, "-")
                        d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
@@ -171,49 +203,170 @@ class ComicDb():
                        comicid = int(earliest) - 1
 
                irow = self.get_irow(comic, comicid)
-               if irow:
+               if irow and irow[0] == "skip":
+                               print "problem with this one, fetching " + str(irow[2]) + " instead."
+                               irow = self.get_irow(comic, irow[2])
+
+               if irow and len(irow) > 3:
                        print "got irow: "
                        print irow
-                       print "\ninserting first...\n"
-                       self.insert_row_first(irow)
+                       print "\ninserting...\n"
+                       self.insert_row_first([irow[0],irow[1],irow[2],irow[3],irow[4],irow[5]])
                else:
                        print "No comic found at " + comicid
 
-       def fetch_latest_std(self, comic, latest):
-               print "fetching new after " + str(comic) + " " + str(latest)
-               dateid = False
-               if len(str(latest)) == 10:
+
+
+
+
+       def fetch_newer(self, comic, newest):
+               if comic == "cyanide":
+                       comicid = self.get_next_id(comic,newest)
+               elif len(newest) == 10:
                        # date id.
-                       dateid = True
-                       dt = string.split(latest, "-")
+                       dt = string.split(newest, "-")
                        d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
-                       newer = d + datetime.timedelta( 1 )
-                       comicid = newer.isoformat()
+                       newest = d + datetime.timedelta( 1 )
+                       comicid = newest.isoformat()
                else:
-                       comicid = int(latest) + 1
+                       comicid = int(newest) + 1
 
-               lasturl = self.db[0]['url']
+               irow = self.get_irow(comic, comicid)
+               if irow and irow[0] == "skip":
+                               print "problem with this one, fetching " + str(irow[1]) + " instead."
+                               irow = self.get_irow(comic, irow[1])
+
+               if irow and len(irow) > 3:
+                       print "got irow: "
+                       print irow
+                       print "\ninserting...\n"
+                       self.insert_row([irow[0],irow[1],irow[2],irow[3],irow[4],irow[5]])
+               else:
+                       print "No comic found at " + str(comicid)
+
+
+
+
+       def fetch_latest_std(self, comic, latest):
+               print "fetching new after " + str(comic) + " " + str(latest)
+               next = False
+               dateid = False
+               if comic == 'cyanide':
+                       next = self.get_next_id(comic, latest)
+                       if not next:
+                               return
+                       else:
+                               comicid = next
+               else: 
+                       if len(str(latest)) == 10:
+                               # date id.
+                               dateid = True
+                               dt = string.split(latest, "-")
+                               d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
+                               newer = d + datetime.timedelta( 1 )
+                               comicid = newer.isoformat()
+                       else:
+                               comicid = int(latest) + 1
+
+               if len(self.db) > 0:
+                       lasturl = self.db[0]['url']
+               else:
+                       lasturl = "http"
                while True:
                        irow = self.get_irow(comic, comicid)
                        if irow:
                                print "got irow: " + str(irow)
-                               if irow[3] == lasturl:
-                                       print "Looping the same, break break break!"
-                                       break
-                               lasturl = irow[3]
-                               print "inserting..."
-                               self.insert_row(irow)
+                               if str(irow[0]) == 'skip':
+                                       print "skipping this one..."
+                                       next = int(irow[1])
+                               else:
+                                       if irow[3] == lasturl:
+                                               print "Looping the same, break break break!"
+                                               break
+                                       lasturl = irow[3]
+                                       print "inserting..."
+                                       self.insert_row([irow[0],irow[1],irow[2],irow[3],irow[4],irow[5]])
+                                       if len(irow) > 6:
+                                               next = irow[6]
+                                               if not next:
+                                                       break
                                if dateid:
                                        dt = string.split(comicid, "-")
                                        d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
                                        newer = d + datetime.timedelta( 1 )
                                        comicid = newer.isoformat()
+                               elif next:
+                                       comicid = next
                                else:
                                        comicid += 1
                        else:
                                break
 
 
+
+
+       def get_next_id(self, comic, number):
+               if comic == 'cyanide':
+                       link = "http://www.explosm.net/comics/" + str(number) + "/"
+                       print "link: " + link
+                       try:
+                               f = urllib2.urlopen(link)
+                               hcode = f.code
+                       except:
+                               hcode = 404
+                       print "got hcode = " + str(hcode) + "\n"
+                       if (hcode != 200):
+                               return False
+                       else:
+                               # print "Cyanide & Happiness is unreliable, so we need to track next and prev"
+                               s = f.read()
+                               f.close()
+                               # title:
+                               splt = string.split(s, 'Previous</a> | <a href="/comics/', 1)
+                               if len(splt) < 2:
+                                       print "no 'next' found"
+                                       return False
+                               splt = string.split(splt[1], '/">Next >', 1)
+                               if len(splt) < 2:
+                                       print "no 'next' found"
+                                       return False
+                               else:
+                                       print "got next: " + splt[0]
+                                       return splt[0]  
+
+       def get_prev_id(self, comic, number):
+               if comic == 'cyanide':
+                       link = "http://www.explosm.net/comics/" + str(number) + "/"
+                       print "link: " + link
+                       try:
+                               f = urllib2.urlopen(link)
+                               hcode = f.code
+                       except:
+                               hcode = 404
+                       print "got hcode = " + str(hcode) + "\n"
+                       if (hcode != 200):
+                               return False
+                       else:
+                               print "Cyanide & Happiness is unreliable, so we need to track next and prev"
+                               s = f.read()
+                               f.close()
+                               # title:
+                               splt = string.split(s, 'First</a> | <a href="/comics/', 1)
+                               if len(splt) < 2:
+                                       print "no 'prev' found"
+                                       return False
+                               splt = string.split(splt[1], '/">< Previous', 1)
+                               if len(splt) < 2 or len(splt[0]) > 5:
+                                       print "no 'next' found"
+                                       return False
+                               else:
+                                       print "got previous: " + splt[0]
+                                       return splt[0]                          
+
+
+
+
+
        def get_irow(self, comic, number):
                dateid = False
                if len(str(number)) == 10:
@@ -230,6 +383,8 @@ class ComicDb():
                        link = "http://dilbert.com/" + str(number) + "/"
                elif comic == 'phd':
                        link = "http://www.phdcomics.com/comics/archive.php?comicid=" + str(number)
+               elif comic == 'cyanide':
+                       link = "http://www.explosm.net/comics/" + str(number) + "/"
                else:
                        return False
 
@@ -296,10 +451,57 @@ class ComicDb():
                                if len(url) < 49:
                                        print "Fake 404! Break break break!"
                                        return False
-                               
 
+                       elif comic == 'cyanide':
+                               s = f.read()
+                               f.close()
+                               # title:
+                               splt = string.split(s, ' First</a> | <a href="/comics/', 1)
+                               if len(splt) < 2:
+                                       print "first comic?"
+                                       prev = False
+                               else:
+                                       splt = string.split(splt[1], '/">< ', 1)
+                                       prev = splt[0];
+                                       print "got prev: " + prev
+                               splt = string.split(s, 'Previous</a> | <a href="/comics/', 1)
+                               if len(splt) < 2:
+                                       print "last comic?"
+                                       next = False
+                               else:
+                                       splt = string.split(splt[1], '/">Next ></a>', 1)
+                                       if len(splt[0]) > 10:
+                                               next = False
+                                       else:
+                                               next = splt[0]
+                                               print "got next: " + next
+
+                               splt = string.split(s, '</tr><tr><td colspan=2>', 1)
+                               if len(splt) < 2:
+                                       print "no comic?"
+                                       return False
+                               splt = string.split(splt[1], ' <b>by <a href="', 1)
+                               title = splt[0];
+                               if len(title) > 10:
+                                       print "title wrong..."
+                                       return False
+                               splt = string.split(splt[1], 'a daily webcomic" src="http://www.explosm', 1)
+                               if len(splt) < 2:
+                                       print "a video? Try skipping"
+                                       return ['skip',next,prev]
+                                       
+                               splt = string.split(splt[1], '">', 1)
+                               url = "http://www.explosm" + splt[0]
+                               splt2 = string.rsplit(url, "/", 1)
+                               filename = splt2[1]
+                               irow = [comic,number,link,url,filename,title,next,prev]
+                               return irow
+       
                        splt2 = string.rsplit(url, "/", 1)
                        filename = splt2[1]
+                       if filename == self.db[0]['filename']:
+                               print "already exists! Break break break!"
+                               return False
                        irow = [comic,number,link,url,filename,title]
                        return irow
 
@@ -307,7 +509,7 @@ class ComicDb():
 # ------------UI
 
 class ComicHomePlugin(hildondesktop.HomePluginItem):
-       __gtype_name__ = 'ComicHomePlugin'
+       # __gtype_name__ = 'ComicHomePlugin'
        def __init__(self):
                hildondesktop.HomePluginItem.__init__(self)
                global supports_alpha
@@ -544,6 +746,11 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
 
                        elif func == 'switch':
                                self.active_comics = self.get_active_comics()
+                               print "closing log"
+                               #sys.stdout.close()
+                               #sys.stdout = sys.__stdout__
+                               #sys.stderr.close()
+                               #sys.stderr = sys.__stderr__
                                print "active comics: " + str(self.active_comics)
                                self.keypointer = (self.keypointer + 1) % len(self.active_comics)
                                self.imgvpos = 0
@@ -609,6 +816,7 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
                        if scaled_buf.get_height() - (vpos + maxheight) < 0:
                                vpos = scaled_buf.get_height() - maxheight
                        scaled_buf.copy_area(0,vpos,478,maxheight,parent_buf,1,0)
+                       # parent_buf.composite(scaled_buf, 0, 0, 478, maxheight, 0, 0, 478, maxheight, gtk.gdk.INTERP_BILINEAR, 0)
                        self.imgvpos = vpos
                else:
                        tmpy = maxheight - scaled_buf.get_height()
@@ -617,7 +825,10 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
 
                        scaled_buf.copy_area(0,0,478,scaled_buf.get_height(),parent_buf,1,tmpy)
                        self.imgvpos = 0
+#              test = parent_buf.render_pixmap_and_mask(0)
+               parent_buf.add_alpha(True,chr(255),chr(255),chr(255))
                retimg.set_from_pixbuf(parent_buf)
+               print retimg.get_pixel_size()
                print "vpos after fixing image: " + str(self.imgvpos)
                return retimg
 
@@ -688,6 +899,7 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
 # ************************* OPTION DIALOGS ********************************
 
        def show_options(self, widget):
+               print "loading options dialog"
                dialog = gtk.Dialog("Comic Widget", None, gtk.DIALOG_DESTROY_WITH_PARENT)
 
                about_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
@@ -700,14 +912,17 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
                comics_button.set_alignment(0,0,0,0)
                comics_button.connect("clicked", self.show_comics)
 
-               dialog.vbox.pack_start(about_button, True, True, 0)
+               print "adding buttons to dialog, starting with comics"
                dialog.vbox.pack_start(comics_button, True, True, 0)
-
+               print "adding about button"
+               dialog.vbox.pack_start(about_button, True, True, 0)
+               print "show!"
                dialog.show_all()
                dialog.run()
                dialog.destroy()
 
        def show_about(self, widget):
+               print "in about dialog"
                dialog = gtk.AboutDialog()
                dialog.set_title("About")
                dialog.set_name("Comic widget")
@@ -723,10 +938,12 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
                dialog.destroy()
 
        def show_comics(self, widget):
+               print "in comics dialog! yay!"
                dialog = gtk.Dialog("Configure Search Engines", None, gtk.DIALOG_DESTROY_WITH_PARENT | gtk.DIALOG_NO_SEPARATOR)
                comiclist = comics.keys()
                comiclist.sort()
                buttonlist = {}
+               self.connlist = {}
                for comicid in comiclist:
                        buttonlist[comicid] = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
                        print buttonlist[comicid].get_name()
@@ -739,10 +956,10 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
                                        break
                        if active:
                                buttonlist[comicid].set_text(comics[comicid]['name'] + " *", comicid + ": click to remove")
-                               buttonlist[comicid].connect("clicked", self.remove_comic)
+                               self.connlist[comicid] = buttonlist[comicid].connect("clicked", self.remove_comic)
                        else:
                                buttonlist[comicid].set_text(comics[comicid]['name'], comicid + ": click to add")
-                               buttonlist[comicid].connect("clicked", self.add_comic)
+                               self.connlist[comicid] = buttonlist[comicid].connect("clicked", self.add_comic)
                        dialog.vbox.pack_start(buttonlist[comicid], True, True, 0)
 
                
@@ -765,6 +982,8 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
                        dbw.writerow(self.active_comics)
                        dbf.close()
                widget.set_text(comics[comicid]['name'] + " *", comicid + ": click to remove")
+               widget.disconnect(self.connlist[comicid])
+               self.connlist[comicid] = widget.connect("clicked", self.remove_comic)
 
        def remove_comic(self, widget):
                print "remove " + widget.get_title()
@@ -774,12 +993,17 @@ class ComicHomePlugin(hildondesktop.HomePluginItem):
                comicid = splt[0]
                if os.path.isfile(activecomics) == True:
                        print "removing " + comicid + " from " + str(self.active_comics)
-                       del self.active_comics[self.active_comics.index(comicid)]
+                       try:
+                               del self.active_comics[self.active_comics.index(comicid)]
+                       except:
+                               pass
                        dbf = open(activecomics, 'w')
                        dbw = csv.writer(dbf)
                        dbw.writerow(self.active_comics)
                        dbf.close()
                widget.set_text(comics[comicid]['name'], comicid + ": click to add")
+               widget.disconnect(self.connlist[comicid])
+               self.connlist[comicid] = widget.connect("clicked", self.add_comic)
                
 hd_plugin_type = ComicHomePlugin