git.maemo.org Git - comic-widget/blob - comic-widget.py

   1 import hildon
   2 import hildondesktop
   3 import gtk
   4 import csv
   5 import urllib2
   6 import urllib
   7 import string
   8 import os
   9 import osso
  10 import cairo
  11 import pango
  12
  13 # constants. dbfile is the location of the csv
  14 # comiccache is the location of the images
  15
  16 imagedir = "/home/user/comic-widget/images/"
  17 dbdir = "/home/user/comic-widget/db/"
  18 comiccache = "/home/user/MyDocs/comics/"
  19 comics = {"xkcd":{"name":"xkcd","link":"http://xkcd.org/","start":666,"dbfile":dbdir + ".comicdb.xkcd.csv"},
  20                 "sinfest":{"name":"Sinfest","link":"http://sinfest.com/","start":3400,"dbfile":dbdir + ".comicdb.sinfest.csv"}}
  21
  22 # handling of the comics
  23 class ComicDb():
  24         def __init__(self, comic):
  25                 #if no db, start from this comic id.
  26                 self.comic = comic
  27                 self.start = comics[self.comic]["start"]
  28                 self.dbfile = comics[self.comic]["dbfile"]
  29
  30                 #if db file exist, read it
  31                 if os.path.isfile(self.dbfile) == True:
  32                         dbf = open(self.dbfile, 'r')
  33                 # if not, create it.
  34                 else:
  35                         dbf = open(self.dbfile, 'w')
  36                         dbf.write('comic,id,link,url,filename,title\n')
  37                         dbf.close()
  38                         dbf = open(self.dbfile, 'r')
  39                         self.fetch_latest_std(self.comic, self.start)
  40
  41                 if not os.path.exists(comiccache + self.comic + "/"):
  42                         os.makedirs(comiccache + self.comic + "/")
  43                 dbr = csv.DictReader(dbf)
  44                 self.db = []
  45                 for row in dbr:
  46                         self.db.insert(0,row)
  47                 dbf.close()
  48                 self.currentcomic = 0
  49
  50         def get_comic(self):
  51                 print str(self.currentcomic)
  52                 if self.currentcomic < 0:
  53                         self.refresh()
  54                         self.currentcomic = 0
  55                 if  len(self.db) < (self.currentcomic + 1):
  56                         self.currentcomic -= 1
  57                         self.refresh()
  58                         self.currentcomic = len(self.db) - 1
  59
  60                 dbrow = self.db[self.currentcomic]
  61                 filename = comiccache + self.comic + "/" + dbrow['filename']
  62                 print filename + "\n"
  63
  64                 if os.path.isfile(filename):
  65                         print "found, returning " + filename + "\n"
  66                         return filename
  67                 else:
  68                         #fetch file
  69                         print filename + " not found\nretrieving " + dbrow['url'] + "\n"
  70                         urllib.urlretrieve(dbrow['url'], filename)
  71                         print "success\n"
  72                         return filename
  73
  74         def get_current_row(self):
  75                 return self.db[self.currentcomic]
  76
  77         def get_link(self):
  78                 print str(self.currentcomic)
  79                 if self.currentcomic < 0:
  80                         self.refresh()
  81                         self.currentcomic = 0
  82                 if  len(self.db) < (self.currentcomic + 1):
  83                         self.currentcomic -= 1
  84                 return self.db[self.currentcomic]['link']
  85
  86
  87
  88         def insert_row(self, irow):
  89                 dbf = open(self.dbfile, 'a')
  90                 dbw = csv.writer(dbf)
  91                 dbw.writerow(irow)
  92                 dbf.close()
  93
  94         def insert_row_first(self, irow):
  95                 dbf = open(self.dbfile, 'w')
  96                 dbf.write('comic,id,link,url,filename,title\n')
  97                 dbw = csv.writer(dbf)
  98                 dbw.writerow(irow)
  99                 dbf.close()
 100                 dbf = open(self.dbfile, 'a')
 101                 dbw = csv.DictWriter(dbf, ['comic','id','link','url','filename','title'])
 102                 tmpdb = self.db
 103                 tmpdb.reverse()
 104                 for row in tmpdb:
 105                         dbw.writerow(row)
 106                 dbf.close()
 107
 108         def refresh(self):
 109                 if len(self.db) < 1:
 110                         self.fetch_latest_std(self.comic, self.start)
 111                 elif self.currentcomic == 0 or self.currentcomic < 0:
 112                         self.fetch_latest_std(self.comic, self.db[0]['id'])
 113                 elif self.currentcomic == (len(self.db) - 1):
 114                         self.fetch_earlier(self.comic, self.db[self.currentcomic]['id'])
 115                 dbf = open(self.dbfile, 'r')
 116                 dbr = csv.DictReader(dbf)
 117                 self.db = []
 118                 for row in dbr:
 119                         self.db.insert(0,row)
 120                 dbf.close()
 121
 122         def fetch_earlier(self, comic, earliest):
 123                 number = int(earliest) - 1
 124                 irow = self.get_irow(comic, number)
 125                 if irow:
 126                         print "got irow: "
 127                         print irow
 128                         print "\ninserting first...\n"
 129                         self.insert_row_first(irow)
 130                 else:
 131                         print "No comic found at " + number
 132
 133         def get_irow(self, comic, number):
 134                 if number < 0:
 135                         return False
 136                 print "number is now: " + str(number) + "\n"
 137                 if comic == 'xkcd':
 138                         link = "http://xkcd.org/" + str(number) + "/"
 139                 elif comic == 'sinfest':
 140                         link = "http://www.sinfest.net/archive_page.php?comicID=" + str(number)
 141                 else:
 142                         return False
 143
 144                 print "link: " + link + "\n"
 145                 try:
 146                         f = urllib2.urlopen(link)
 147                         hcode = f.code
 148                 except:
 149                         hcode = 404
 150                 print "got hcode = " + str(hcode) + "\n"
 151                 if (hcode != 200):
 152                         return False
 153                 else:
 154                         # Build db entry.
 155                         # they look like: comic,id,link,url,filename,title
 156                         # We already have comic, number/id and link
 157
 158                         if comic == 'xkcd':
 159                                 s = f.read()
 160                                 f.close()
 161                                 # This should be done with regex but...
 162                                 splt = string.split(s, "<h3>Image URL (for hotlinking/embedding): ", 1)
 163                                 splt2 = string.split(splt[1], "</h3>", 1)
 164                                 url = splt2[0]
 165                                 splt = string.split(splt[0], "<h1>", 1)
 166                                 splt = string.split(splt[1], "</h1>", 1)
 167
 168                                 title = splt[0]
 169
 170                         elif comic == 'sinfest':
 171                                 s = f.read()
 172                                 f.close()
 173                                 splt = string.split(s, 'height="107"', 1)
 174                                 splt = string.split(splt[1], '<img src="', 1)
 175                                 splt = string.split(splt[1], '" alt="', 1)
 176                                 url = splt[0]
 177                                 if len(url) < 49:
 178                                         print "Fake 404! Break break break!"
 179                                         return False
 180                                 splt = string.split(splt[1], '" border="0" />', 1)
 181                                 title = splt[0]
 182                         splt2 = string.rsplit(url, "/", 1)
 183                         filename = splt2[1]
 184                         irow = [comic,number,link,url,filename,title]
 185                         return irow
 186                         print "got irow: "
 187
 188         def fetch_latest_std(self, comic, latest):
 189
 190                 print "fetching new after " + str(comic) + " " + str(latest) + "\n"
 191                 number = int(latest) + 1
 192                 while True:
 193                         irow = self.get_irow(comic, number)
 194                         if irow:
 195                                 print "got irow: "
 196                                 print irow
 197                                 print "\ninserting...\n"
 198                                 self.insert_row(irow)
 199                                 number += 1
 200                         else:
 201                                 break
 202
 203 # ------------
 204
 205 class ComicHomePlugin(hildondesktop.HomePluginItem):
 206         def __init__(self):
 207                 hildondesktop.HomePluginItem.__init__(self)
 208                 self.osso_c = osso.Context("comic-widget", "0.0.1", False)
 209                 self.osso_rpc = osso.Rpc(self.osso_c)
 210                 self.keys = comics.keys()
 211                 self.keypointer = 1
 212                 self.db = ComicDb(self.keys[self.keypointer])
 213                 self.comicname = comics[self.keys[self.keypointer]]['name']
 214                 self.set_name = "comicwidget"
 215                 screen = self.get_screen()
 216                 colormap = screen.get_rgba_colormap()
 217                 self.set_colormap(colormap)
 218                 self.set_app_paintable(True)
 219                 #setup internal area
 220                 self.set_size_request(480, 230)
 221                 self.imgvpos = 0;
 222                 self.vbox = gtk.VBox()
 223                 self.hbox = gtk.HBox()
 224                 self.label = gtk.Label(self.comicname + " " + str(self.db.db[self.db.currentcomic]['id']))
 225
 226                 self.e_goweb = gtk.EventBox()
 227                 self.e_open = gtk.EventBox()
 228                 self.e_switch = gtk.EventBox()
 229                 self.e_switch.set_size_request(48, 48)
 230                 self.e_next = gtk.EventBox()
 231                 self.e_next.set_size_request(48, 48)
 232                 self.e_prev = gtk.EventBox()
 233                 self.e_prev.set_size_request(48, 48)
 234                 self.e_up = gtk.EventBox()
 235                 self.e_up.set_size_request(48, 48)
 236                 self.e_down = gtk.EventBox()
 237                 self.e_down.set_size_request(48, 48)
 238                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 239                 self.comic_image.show()
 240
 241                 # self.next = gtk.image_new_from_file(imagedir + "right0-48x48_mask.png")
 242                 # self.prev = gtk.image_new_from_file(imagedir + "left0-48x48_mask.png")
 243                 # self.up = gtk.image_new_from_file(imagedir + "up0-48x48_mask.png")
 244                 # self.down = gtk.image_new_from_file(imagedir + "down0-48x48_mask.png")
 245                 # self.switch = gtk.image_new_from_file(imagedir + "star0-48x48_mask.png")
 246                 self.e_goweb.add(self.label)
 247                 self.e_goweb.set_events(gtk.gdk.BUTTON_PRESS_MASK)
 248                 self.e_goweb.connect("button-press-event", self.view_comic)
 249
 250                 self.e_open.add(self.comic_image)
 251
 252                 # self.e_switch.add(self.switch)
 253                 self.e_switch.set_events(gtk.gdk.BUTTON_PRESS_MASK)
 254                 self.e_switch.connect("button-press-event", self.switch_comic)
 255
 256
 257                 # self.e_next.add(self.next)
 258                 self.e_next.set_events(gtk.gdk.BUTTON_PRESS_MASK)
 259                 self.e_next.connect("button-press-event", self.get_next)
 260
 261                 #self.e_prev.add(self.prev)
 262                 self.e_prev.set_events(gtk.gdk.BUTTON_PRESS_MASK)
 263                 self.e_prev.connect("button_press_event", self.get_prev)
 264
 265                 #self.e_up.add(self.up)
 266                 self.e_up.set_events(gtk.gdk.BUTTON_PRESS_MASK)
 267                 self.e_up.connect("button-press-event", self.scroll_up)
 268
 269                 #self.e_down.add(self.down)
 270                 self.e_down.set_events(gtk.gdk.BUTTON_PRESS_MASK)
 271                 self.e_down.connect("button-press-event", self.scroll_down)
 272
 273                 self.vbox.pack_start(self.e_open,False,False,0)
 274                 self.vbox.pack_end(self.hbox,False,False,0)
 275
 276                 self.hbox.pack_start(self.e_switch,False,False,5)
 277                 # self.hbox.add(self.label)
 278                 self.hbox.pack_start(self.e_goweb,False,False,5)
 279                 self.hbox.pack_end(self.e_next,False,False,5)
 280                 self.hbox.pack_end(self.e_down,False,False,5)
 281                 self.hbox.pack_end(self.e_up,False,False,5)
 282                 self.hbox.pack_end(self.e_prev,False,False,5)
 283                 self.vbox.show_all()
 284                 self.add(self.vbox)
 285
 286         def realize(self):
 287                 screen = self.get_screen()
 288                 colormap = screen.get_rgba_colormap()
 289                 self.set_colormap(colormap)
 290                 return None
 291
 292         def do_expose_event(widget, event):
 293                 cr = widget.window.cairo_create()
 294                 cr.set_operator(cairo.OPERATOR_OVER)
 295                 widget.draw(widget.e_up, imagedir + "up0-48x48.png")
 296                 widget.draw(widget.e_down, imagedir + "down0-48x48.png")
 297                 widget.draw(widget.e_prev, imagedir + "left0-48x48.png")
 298                 widget.draw(widget.e_next, imagedir + "right0-48x48.png")
 299                 widget.draw(widget.e_switch, imagedir + "star0-48x48.png")
 300                 region = gtk.gdk.region_rectangle(event.area)
 301                 cr.region(region)
 302                 bg_color=gtk.gdk.color_parse('#000000')
 303                 cr.set_source_rgba (0.0, 0.0, 0.0, 0.7)
 304                 cr.fill()
 305                 return False
 306
 307         def draw(self, target, pic):
 308                 cr_e = target.window.cairo_create()
 309                 cr_e.set_source_rgba(0.0, 0.0, 0.0, 0.7) # Transparent
 310                 cr_e.set_operator(cairo.OPERATOR_SOURCE)
 311                 cr_e.paint()
 312                 if pic != None:
 313                         cr_e.set_operator(cairo.OPERATOR_OVER)
 314                         image = cairo.ImageSurface.create_from_png(pic)
 315                         cr_e.set_source_surface(image, 0, 0)
 316                         cr_e.paint()
 317
 318
 319         def switch_comic(self, widget, event):
 320                 if not event.type == gtk.gdk.BUTTON_PRESS:
 321                         return False
 322                 self.keypointer = (self.keypointer + 1) % 2
 323                 self.imgvpos = 0
 324                 self.db = []
 325                 print "switching to " + self.keys[self.keypointer]
 326                 self.comicname = comics[self.keys[self.keypointer]]['name']
 327                 self.db = ComicDb(self.keys[self.keypointer])
 328                 self.e_goweb.remove(self.label)
 329                 self.label = gtk.Label(self.comicname + " " + str(self.db.db[self.db.currentcomic]['id']))
 330                 self.e_goweb.add(self.label)
 331                 self.e_goweb.show_all()
 332                 self.e_open.remove(self.comic_image)
 333                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 334                 self.e_open.add(self.comic_image)
 335                 self.e_open.show_all()
 336
 337         def switch_rel(self, widget, event):
 338                 if not event.type == gtk.gdk.BUTTON_RELEASE:
 339                         return False
 340                 widget.remove(self.switch)
 341                 self.switch = gtk.image_new_from_file(imagedir + "star.png")
 342                 widget.add(self.switch)
 343                 widget.show_all()
 344
 345         def view_comic(self, widget, event):
 346                 if not event.type == gtk.gdk.BUTTON_PRESS:
 347                         return False
 348                 link = self.db.get_link()
 349                 self.osso_rpc.rpc_run_with_defaults("osso_browser", "open_new_window", (link,))
 350
 351         def scroll_down(self, widget, event):
 352                 if not event.type == gtk.gdk.BUTTON_PRESS:
 353                         return False
 354                 self.e_open.remove(self.comic_image)
 355                 self.imgvpos = self.imgvpos + 80
 356                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 357                 self.e_open.add(self.comic_image)
 358                 self.e_open.show_all()
 359
 360         def scroll_up(self, widget, event):
 361                 if not event.type == gtk.gdk.BUTTON_PRESS:
 362                         return False
 363                 self.e_open.remove(self.comic_image)
 364                 self.imgvpos = self.imgvpos - 80
 365                 if self.imgvpos < 0:
 366                         self.imgvpos = 0
 367                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 368                 self.e_open.add(self.comic_image)
 369                 self.e_open.show_all()
 370
 371         def get_next(self, widget, event):
 372                 if not event.type == gtk.gdk.BUTTON_PRESS:
 373                         return False
 374                 self.db.currentcomic -= 1
 375                 self.imgvpos = 0
 376                 self.e_open.remove(self.comic_image)
 377                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 378                 self.e_open.add(self.comic_image)
 379                 self.e_open.show_all()
 380                 self.e_goweb.remove(self.label)
 381                 self.label = gtk.Label(self.comicname + " " + str(self.db.db[self.db.currentcomic]['id']))
 382                 self.e_goweb.add(self.label)
 383                 self.e_goweb.show_all()
 384
 385         def get_prev(self, widget, event):
 386                 if not event.type == gtk.gdk.BUTTON_PRESS:
 387                         return False
 388                 self.db.currentcomic += 1
 389                 self.imgvpos = 0
 390                 self.e_open.remove(self.comic_image)
 391                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 392                 self.e_open.add(self.comic_image)
 393                 self.e_open.show_all()
 394                 self.e_goweb.remove(self.label)
 395                 self.label = gtk.Label(self.comicname + " " + str(self.db.db[self.db.currentcomic]['id']))
 396                 self.e_goweb.add(self.label)
 397                 self.e_goweb.show_all()
 398
 399         def get_resized_pixmap(self, filename, vpos):
 400                 print "vpos: " + str(vpos)
 401
 402                 maxheight = 182
 403
 404                 pixbuf = gtk.gdk.pixbuf_new_from_file(filename)
 405                 parent_buf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, pixbuf.get_has_alpha(), 8, 480, maxheight)
 406                 parent_buf.fill(0xffffffff)
 407                 retimg = gtk.Image()
 408                 # get correct width/height
 409                 new_height = 1
 410                 new_width = 478
 411                 new_height = int((478.0 / pixbuf.get_width()) * pixbuf.get_height())
 412                 scaled_buf = pixbuf.scale_simple(new_width,new_height,gtk.gdk.INTERP_BILINEAR)
 413                 print "scaled height: " + str(scaled_buf.get_height())
 414                 if scaled_buf.get_height() > maxheight:
 415                         if scaled_buf.get_height() - (vpos + maxheight) < 0:
 416                                 vpos = scaled_buf.get_height() - maxheight
 417                         scaled_buf.copy_area(0,vpos,478,maxheight,parent_buf,1,0)
 418                         self.imgvpos = vpos
 419                 else:
 420                         tmpy = maxheight - scaled_buf.get_height()
 421                         if tmpy > 0:
 422                                 tmpy = tmpy / 2
 423
 424                         scaled_buf.copy_area(0,0,478,scaled_buf.get_height(),parent_buf,1,tmpy)
 425                         self.imgvpos = 0
 426                 retimg.set_from_pixbuf(parent_buf)
 427                 print "vpos after fixing image: " + str(self.imgvpos)
 428                 return retimg
 429
 430
 431 hd_plugin_type = ComicHomePlugin
 432
 433
 434 if __name__ == "__main__":
 435         import gobject
 436         gobject.type_register(hd_plugin_type)
 437         obj = gobject.new(hd_plugin_type, plugin_id="plugin_id")
 438         obj.show_all()
 439         gtk.main()