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()
 225                 self.label.set_alignment(xalign = 0, yalign = 0.5)
 226                 self.label.set_use_markup(True)
 227                 titl = str(self.db.db[self.db.currentcomic]['title'])
 228                 if len(titl) > 22:
 229                         titl = titl[:19] + "..."
 230                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 231
 232
 233                 self.e_goweb = gtk.EventBox()
 234                 self.e_goweb.set_name('goweb')
 235                 self.e_goweb.set_size_request(220, 48)
 236                 # e_open is the eventbox for the image. Does nothing, but it's convenient to have it.
 237                 # it will change to be a cairo thing when I have the time.
 238                 self.e_open = gtk.EventBox()
 239                 self.e_open.set_name('open')
 240                 self.e_switch = gtk.EventBox()
 241                 self.e_switch.set_name('switch')
 242                 self.e_switch.set_size_request(48, 48)
 243                 self.e_next = gtk.EventBox()
 244                 self.e_next.set_name('next')
 245                 self.e_next.set_size_request(48, 48)
 246                 self.e_prev = gtk.EventBox()
 247                 self.e_prev.set_name('prev')
 248                 self.e_prev.set_size_request(48, 48)
 249                 self.e_up = gtk.EventBox()
 250                 self.e_up.set_name('up')
 251                 self.e_up.set_size_request(48, 48)
 252                 self.e_down = gtk.EventBox()
 253                 self.e_down.set_name('down')
 254                 self.e_down.set_size_request(48, 48)
 255                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 256                 self.comic_image.show()
 257
 258
 259                 # Preload images into a dictionary, so that we can access them by the names of the eventboxes:
 260
 261                 self.images = {'up0':cairo.ImageSurface.create_from_png(imagedir + "up0-48x48.png"),
 262                                         'up1':cairo.ImageSurface.create_from_png(imagedir + "up1-48x48.png"),
 263                                         'down0':cairo.ImageSurface.create_from_png(imagedir + "down0-48x48.png"),
 264                                         'down1':cairo.ImageSurface.create_from_png(imagedir + "down1-48x48.png"),
 265                                         'next0':cairo.ImageSurface.create_from_png(imagedir + "next0-48x48.png"),
 266                                         'next1':cairo.ImageSurface.create_from_png(imagedir + "next1-48x48.png"),
 267                                         'prev0':cairo.ImageSurface.create_from_png(imagedir + "prev0-48x48.png"),
 268                                         'prev1':cairo.ImageSurface.create_from_png(imagedir + "prev1-48x48.png"),
 269                                         'switch0':cairo.ImageSurface.create_from_png(imagedir + "switch0-48x48.png"),
 270                                         'switch1':cairo.ImageSurface.create_from_png(imagedir + "switch1-48x48.png"),
 271                                         }
 272
 273
 274
 275                 self.e_goweb.add(self.label)
 276                 self.e_goweb.set_events(gtk.gdk.BUTTON_RELEASE_MASK)
 277                 self.e_goweb.connect("button-release-event", self.view_comic)
 278
 279                 self.e_open.add(self.comic_image)
 280
 281                 # self.e_switch.add(self.switch)
 282                 self.e_switch.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 283                 self.e_switch.connect("button-press-event", self.button_press)
 284                 self.e_switch.connect("button-release-event", self.button_release)
 285                 self.e_switch.connect("leave-notify-event", self.button_release)
 286
 287
 288                 # self.e_next.add(self.next)
 289                 self.e_next.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 290                 self.e_next.connect("button-press-event", self.button_press)
 291                 self.e_next.connect("button-release-event", self.button_release)
 292                 self.e_next.connect("leave-notify-event", self.button_release)
 293
 294                 #self.e_prev.add(self.prev)
 295                 self.e_prev.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 296                 self.e_prev.connect("button-press-event", self.button_press)
 297                 self.e_prev.connect("button-release-event", self.button_release)
 298                 self.e_prev.connect("leave-notify-event", self.button_release)
 299
 300                 #self.e_up.add(self.up)
 301                 self.e_up.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 302                 self.e_up.connect("button-press-event", self.button_press)
 303                 self.e_up.connect("button-release-event", self.button_release)
 304                 self.e_up.connect("leave-notify-event", self.button_release)
 305
 306
 307                 #self.e_down.add(self.down)
 308                 self.e_down.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 309                 self.e_down.connect("button-press-event", self.button_press)
 310                 self.e_down.connect("button-release-event", self.button_release)
 311                 self.e_down.connect("leave-notify-event", self.button_release)
 312
 313                 self.vbox.pack_start(self.e_open,False,False,0)
 314                 self.vbox.pack_end(self.hbox,False,False,0)
 315
 316                 self.hbox.pack_start(self.e_switch,False,False,0)
 317                 self.hbox.pack_start(self.e_goweb,False,False,10)
 318                 self.hbox.pack_end(self.e_next,False,False,0)
 319                 self.hbox.pack_end(self.e_prev,False,False,0)
 320                 self.hbox.pack_end(self.e_down,False,False,0)
 321                 self.hbox.pack_end(self.e_up,False,False,0)
 322                 self.vbox.show_all()
 323                 self.add(self.vbox)
 324
 325         # **************** Cairo draw functions ***************
 326
 327         def realize(self):
 328                 screen = self.get_screen()
 329                 colormap = screen.get_rgba_colormap()
 330                 self.set_colormap(colormap)
 331                 return None
 332
 333         def do_expose_event(widget, event):
 334                 cr = widget.window.cairo_create()
 335                 widget.draw(widget.e_up, 0)
 336                 widget.draw(widget.e_down, 0)
 337                 widget.draw(widget.e_prev,0)
 338                 widget.draw(widget.e_next, 0)
 339                 widget.draw(widget.e_switch, 0)
 340 #               widget.writeit(widget.e_goweb, "asdf")
 341                 cr.set_operator(cairo.OPERATOR_OVER)
 342                 region = gtk.gdk.region_rectangle(event.area)
 343                 cr.region(region)
 344                 bg_color=gtk.gdk.color_parse('#000000')
 345                 cr.set_source_rgba (0.0, 0.0, 0.0, 0.5)
 346                 cr.fill_preserve()
 347                 return False
 348
 349         # Function that draws a button preloaded in the images dictionary.
 350         def draw(self, target, postfix):
 351                 cr_e = target.window.cairo_create()
 352                 cr_e.set_source_rgba(0.0, 0.0, 0.0, 0.5) # Transparent
 353                 cr_e.set_operator(cairo.OPERATOR_SOURCE)
 354                 cr_e.paint()
 355                 cr_e.set_operator(cairo.OPERATOR_OVER)
 356                 cr_e.set_source_surface(self.images[target.get_name() + str(postfix)], 0, 0)
 357                 cr_e.paint()
 358
 359         # ********** Event handlers *************
 360
 361
 362         # Most events happen on release, so I moved all of them into the same event handler.
 363         def button_release(self, widget, event):
 364                 if event.type == gtk.gdk.BUTTON_RELEASE:
 365                         func = widget.get_name()
 366                         print "pressed: " + func
 367                         if func == 'up':
 368                                 self.e_open.remove(self.comic_image)
 369                                 self.imgvpos = self.imgvpos - 80
 370                                 if self.imgvpos < 0:
 371                                         self.imgvpos = 0
 372                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 373                                 self.e_open.add(self.comic_image)
 374                                 self.e_open.show_all()
 375
 376                         elif func == 'down':
 377                                 self.e_open.remove(self.comic_image)
 378                                 self.imgvpos = self.imgvpos + 80
 379                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 380                                 self.e_open.add(self.comic_image)
 381                                 self.e_open.show_all()
 382
 383                         elif func == 'next':
 384                                 self.db.currentcomic -= 1
 385                                 self.imgvpos = 0
 386                                 self.e_open.remove(self.comic_image)
 387                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 388                                 self.e_open.add(self.comic_image)
 389                                 self.e_open.show_all()
 390                                 self.e_goweb.remove(self.label)
 391                                 titl = str(self.db.db[self.db.currentcomic]['title'])
 392                                 if len(titl) > 22:
 393                                         titl = titl[:19] + "..."
 394                                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 395                                 self.e_goweb.add(self.label)
 396                                 self.e_goweb.show_all()
 397
 398                         elif func == 'prev':
 399                                 self.db.currentcomic += 1
 400                                 self.imgvpos = 0
 401                                 self.e_open.remove(self.comic_image)
 402                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 403                                 self.e_open.add(self.comic_image)
 404                                 self.e_open.show_all()
 405                                 self.e_goweb.remove(self.label)
 406                                 titl = str(self.db.db[self.db.currentcomic]['title'])
 407                                 if len(titl) > 22:
 408                                         titl = titl[:19] + "..."
 409                                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 410                                 self.e_goweb.add(self.label)
 411                                 self.e_goweb.show_all()
 412
 413                         elif func == 'switch':
 414                                 self.keypointer = (self.keypointer + 1) % 2
 415                                 self.imgvpos = 0
 416                                 self.db = []
 417                                 print "switching to " + self.keys[self.keypointer]
 418                                 self.comicname = comics[self.keys[self.keypointer]]['name']
 419                                 self.db = ComicDb(self.keys[self.keypointer])
 420                                 self.e_goweb.remove(self.label)
 421                                 titl = str(self.db.db[self.db.currentcomic]['title'])
 422                                 if len(titl) > 22:
 423                                         titl = titl[:19] + "..."
 424                                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 425                                 self.e_goweb.add(self.label)
 426                                 self.e_goweb.show_all()
 427                                 self.e_open.remove(self.comic_image)
 428                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 429                                 self.e_open.add(self.comic_image)
 430                                 self.e_open.show_all()
 431
 432                         else:
 433                                 return False
 434                 self.draw(widget, "0")
 435
 436         # this one only switches to pressed button image
 437         def button_press(self, widget, event):
 438                 if not event.type == gtk.gdk.BUTTON_PRESS:
 439                         return False
 440                 self.draw(widget, "1")
 441
 442         # this one cancels a button press when pointer is moved out from it after press.
 443         def button_out(self, widget, event):
 444                 if not event.type == gtk.gdk.BUTTON_PRESS:
 445                         return False
 446                 self.draw(widget, "0")
 447
 448         # no fancy cairo stuff here. simply launches url in browser.
 449         def view_comic(self, widget, event):
 450                 if not event.type == gtk.gdk.BUTTON_RELEASE:
 451                         return False
 452                 link = self.db.get_link()
 453                 self.osso_rpc.rpc_run_with_defaults("osso_browser", "open_new_window", (link,))
 454
 455
 456
 457         # this function modifies the image to fit the widget and then returns it.
 458
 459         def get_resized_pixmap(self, filename, vpos):
 460                 print "vpos: " + str(vpos)
 461
 462                 maxheight = 182
 463
 464                 pixbuf = gtk.gdk.pixbuf_new_from_file(filename)
 465                 parent_buf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, pixbuf.get_has_alpha(), 8, 480, maxheight)
 466                 parent_buf.fill(0xffffffff)
 467                 retimg = gtk.Image()
 468                 # get correct width/height
 469                 new_height = 1
 470                 new_width = 478
 471                 new_height = int((478.0 / pixbuf.get_width()) * pixbuf.get_height())
 472                 scaled_buf = pixbuf.scale_simple(new_width,new_height,gtk.gdk.INTERP_BILINEAR)
 473                 print "scaled height: " + str(scaled_buf.get_height())
 474                 if scaled_buf.get_height() > maxheight:
 475                         if scaled_buf.get_height() - (vpos + maxheight) < 0:
 476                                 vpos = scaled_buf.get_height() - maxheight
 477                         scaled_buf.copy_area(0,vpos,478,maxheight,parent_buf,1,0)
 478                         self.imgvpos = vpos
 479                 else:
 480                         tmpy = maxheight - scaled_buf.get_height()
 481                         if tmpy > 0:
 482                                 tmpy = tmpy / 2
 483
 484                         scaled_buf.copy_area(0,0,478,scaled_buf.get_height(),parent_buf,1,tmpy)
 485                         self.imgvpos = 0
 486                 retimg.set_from_pixbuf(parent_buf)
 487                 print "vpos after fixing image: " + str(self.imgvpos)
 488                 return retimg
 489
 490
 491 hd_plugin_type = ComicHomePlugin
 492
 493
 494 if __name__ == "__main__":
 495         import gobject
 496         gobject.type_register(hd_plugin_type)
 497         obj = gobject.new(hd_plugin_type, plugin_id="plugin_id")
 498         obj.show_all()
 499         gtk.main()