git.maemo.org Git - comic-widget/blob - src/usr/lib/hildon-desktop/comicwidget.py

   1 import gobject
   2 import hildon
   3 import hildondesktop
   4 import gtk
   5 import csv
   6 import urllib2
   7 import urllib
   8 import string
   9 import os
  10 import osso
  11 import cairo
  12 import datetime
  13 import shutil
  14 import sys
  15
  16 supports_alpha = False
  17
  18 # LOGGING!
  19 # sys.stdout = open('/home/user/.comic-widget/output_log.txt', 'a')
  20 # sys.stderr = open('/home/user/.comic-widget/error_log.txt', 'a')
  21 # print "Start logging!"
  22
  23 # constants. dbfile is the location of the csv
  24 # comiccache is the location of the images
  25 APP_VERSION = "0.3.2-2"
  26
  27 basedbdir = "/opt/comic-widget/db/"
  28 imagedir = "/opt/comic-widget/images/"
  29 dbdir = "/home/user/.comic-widget/"
  30 activecomics = dbdir + "activecomics.cfg"
  31 comiccache = "/home/user/MyDocs/.comics/"
  32 comics = {"xkcd":{"name":"xkcd","link":"http://xkcd.org/","start":666,"dbfile":dbdir + "comicdb.xkcd.csv"},
  33                 "sinfest":{"name":"Sinfest","link":"http://sinfest.com/","start":3400,"dbfile":dbdir + "comicdb.sinfest.csv"},
  34                 "phd":{"name":"PHD Comics","link":"http://www.phdcomics.com/","start":1240,"dbfile":dbdir + "comicdb.phd.csv"},
  35                 "dilbert":{"name":"Dilbert","link":"http://dilbert.com/","start":"2009-01-01","dbfile":dbdir + "comicdb.dilbert.csv"},
  36                 "cyanide":{"name":"C and H","link":"http://explosm.com/","start":"1920","dbfile":dbdir + "comicdb.cyanide.csv"},
  37                 }
  38 defaults = {'width':480,'height':230}
  39 #defaults = {'width':480,'height':240}
  40 sizefile = "/home/user/.comic-widget/size.cfg"
  41 previous = False
  42 next = False
  43
  44
  45 # handling of the comics
  46 class ComicDb():
  47         def __init__(self, comic):
  48                 self.comic = comic
  49                 self.start = comics[self.comic]["start"]
  50                 self.dbfile = comics[self.comic]["dbfile"]
  51
  52                 #if db file exist, read it
  53                 # if os.path.isfile(self.dbfile) == True:
  54                 #       dbf = open(self.dbfile, 'r')
  55                 # if not, create it.
  56                 dbf = self.get_dbfile()
  57                 dbr = csv.DictReader(dbf)
  58                 self.db = []
  59                 for row in dbr:
  60                         self.db.insert(0,row)
  61                 dbf.close()
  62                 if len(self.db) == 0:
  63                         self.refresh()
  64                 self.currentcomic = 0
  65
  66
  67
  68         def get_dbfile(self):
  69                 #if db file exist, read it
  70                 print "fetching " + self.dbfile
  71                 if os.path.isfile(self.dbfile) == True:
  72                         print "found " + self.dbfile
  73                         dbf = open(self.dbfile, 'r')
  74                         return dbf
  75                 # if not, copy it.
  76                 else:
  77                         print "file " + self.dbfile + " not found"
  78                         if not os.path.exists(dbdir):
  79                                 print "path not found: " + dbdir
  80                                 try:
  81                                         os.makedirs(dbdir)
  82                                 except:
  83                                         print "comic db creation failed on mkdir"
  84                                         exit()
  85                         try:
  86                                 shutil.copyfile(basedbdir + "comicdb." + self.comic + ".csv", dbdir + "comicdb." + self.comic + ".csv")
  87                         except:
  88                                 print "comic db creation failed on copy"
  89                                 print "creating new..."
  90                                 dbf = open(self.dbfile, 'w')
  91                                 dbf.write('comic,id,link,url,filename,title\n')
  92                                 dbf.close()
  93
  94                         if os.path.isfile(self.dbfile) == True:
  95                                 dbf = open(self.dbfile, 'r')
  96                                 return dbf
  97                         else:
  98                                 print "comic db creation failed after copy"
  99
 100
 101         def get_comic(self):
 102                 print str(self.currentcomic)
 103                 if self.currentcomic < 0:
 104                         self.currentcomic = 0
 105                         if len(self.db) > 0:
 106                                 fetchid = self.db[self.currentcomic]['id']
 107                         else:
 108                                 fetchid = comics[self.comic]['start']
 109                         self.fetch_newer(self.comic, fetchid)
 110                         self.refresh()
 111                 if  len(self.db) < (self.currentcomic + 1):
 112                         self.currentcomic = len(self.db) - 1
 113                         if len(self.db) > 0:
 114                                 fetchid = self.db[self.currentcomic]['id']
 115                         else:
 116                                 fetchid = comics[self.comic]['start']
 117                         self.fetch_earlier(self.comic, fetchid)
 118                         self.refresh()
 119                         self.currentcomic = len(self.db) - 1
 120
 121                 dbrow = self.db[self.currentcomic]
 122                 filename = comiccache + self.comic + "/" + dbrow['filename']
 123                 print filename + "\n"
 124
 125                 if os.path.isfile(filename):
 126                         print "found, returning " + filename + "\n"
 127                         return filename
 128                 else:
 129                         #fetch file
 130                         print filename + " not found\nretrieving " + dbrow['url'] + "\n"
 131                         if not os.path.exists(comiccache + self.comic + "/"):
 132                                 print "create dir " + comiccache + self.comic + "/"
 133                                 try:
 134                                         os.makedirs(comiccache + self.comic + "/")
 135                                         print "created dir"
 136                                 except:
 137                                         print "comic db creation failed on mkdir"
 138                         urllib.urlretrieve(dbrow['url'], filename)
 139                         print "success\n"
 140                         return filename
 141
 142         def get_current_row(self):
 143                 return self.db[self.currentcomic]
 144
 145         def get_link(self):
 146                 print str(self.currentcomic)
 147                 if self.currentcomic < 0:
 148                         self.refresh()
 149                         self.currentcomic = 0
 150                 if  len(self.db) < (self.currentcomic + 1):
 151                         self.currentcomic -= 1
 152                 return self.db[self.currentcomic]['link']
 153
 154
 155
 156         def insert_row(self, irow):
 157                 dbf = open(self.dbfile, 'a')
 158                 dbw = csv.writer(dbf)
 159                 dbw.writerow(irow)
 160                 dbf.close()
 161
 162         def insert_row_first(self, irow):
 163                 dbf = open(self.dbfile, 'w')
 164                 dbf.write('comic,id,link,url,filename,title\n')
 165                 dbw = csv.writer(dbf)
 166                 dbw.writerow(irow)
 167                 dbf.close()
 168                 dbf = open(self.dbfile, 'a')
 169                 dbw = csv.DictWriter(dbf, ['comic','id','link','url','filename','title'])
 170                 tmpdb = self.db
 171                 tmpdb.reverse()
 172                 for row in tmpdb:
 173                         dbw.writerow(row)
 174                 dbf.close()
 175
 176         def refresh(self):
 177         #       if len(self.db) < 1:
 178         #               self.currentcomic = -1
 179         #               self.fetch_latest_std(self.comic, self.start)
 180         #       elif self.currentcomic == 0 or self.currentcomic < 0:
 181         #               self.fetch_latest_std(self.comic, self.db[0]['id'])
 182         #       elif self.currentcomic == (len(self.db) - 1):
 183         #               self.fetch_earlier(self.comic, self.db[self.currentcomic]['id'])
 184                 dbf = open(self.dbfile, 'r')
 185                 dbr = csv.DictReader(dbf)
 186                 self.db = []
 187                 for row in dbr:
 188                         self.db.insert(0,row)
 189                 dbf.close()
 190
 191
 192 # fetch earlier
 193         def fetch_earlier(self, comic, earliest):
 194                 print "fetch before, " + comic + " earliest"
 195                 if comic == "cyanide":
 196                         comicid = self.get_prev_id(comic,earliest)
 197                         print "got " + comicid + " as the one before current..."
 198                 elif len(earliest) == 10:
 199                         # date id.
 200                         dt = string.split(earliest, "-")
 201                         d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
 202                         earlier = d - datetime.timedelta( 1 )
 203                         comicid = earlier.isoformat()
 204                 else:
 205                         comicid = int(earliest) - 1
 206
 207                 irow = self.get_irow(comic, comicid)
 208                 if irow and irow[0] == "skip":
 209                                 print "problem with this one, fetching " + str(irow[2]) + " instead."
 210                                 irow = self.get_irow(comic, irow[2])
 211
 212                 if irow and len(irow) > 3:
 213                         print "got irow: "
 214                         print irow
 215                         print "\ninserting...\n"
 216                         self.insert_row_first([irow[0],irow[1],irow[2],irow[3],irow[4],irow[5]])
 217                 else:
 218                         print "No comic found at " + comicid
 219
 220
 221
 222
 223
 224         def fetch_newer(self, comic, newest):
 225                 if comic == "cyanide":
 226                         comicid = self.get_next_id(comic,newest)
 227                 elif len(newest) == 10:
 228                         # date id.
 229                         dt = string.split(newest, "-")
 230                         d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
 231                         newest = d + datetime.timedelta( 1 )
 232                         comicid = newest.isoformat()
 233                 else:
 234                         comicid = int(newest) + 1
 235
 236                 irow = self.get_irow(comic, comicid)
 237                 if irow and irow[0] == "skip":
 238                                 print "problem with this one, fetching " + str(irow[1]) + " instead."
 239                                 irow = self.get_irow(comic, irow[1])
 240
 241                 if irow and len(irow) > 3:
 242                         print "got irow: "
 243                         print irow
 244                         print "\ninserting...\n"
 245                         self.insert_row([irow[0],irow[1],irow[2],irow[3],irow[4],irow[5]])
 246                 else:
 247                         print "No comic found at " + str(comicid)
 248
 249
 250
 251
 252         def fetch_latest_std(self, comic, latest):
 253                 print "fetching new after " + str(comic) + " " + str(latest)
 254                 next = False
 255                 dateid = False
 256                 if comic == 'cyanide':
 257                         next = self.get_next_id(comic, latest)
 258                         if not next:
 259                                 return
 260                         else:
 261                                 comicid = next
 262                 else:
 263                         if len(str(latest)) == 10:
 264                                 # date id.
 265                                 dateid = True
 266                                 dt = string.split(latest, "-")
 267                                 d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
 268                                 newer = d + datetime.timedelta( 1 )
 269                                 comicid = newer.isoformat()
 270                         else:
 271                                 comicid = int(latest) + 1
 272
 273                 if len(self.db) > 0:
 274                         lasturl = self.db[0]['url']
 275                 else:
 276                         lasturl = "http"
 277                 while True:
 278                         irow = self.get_irow(comic, comicid)
 279                         if irow:
 280                                 print "got irow: " + str(irow)
 281                                 if str(irow[0]) == 'skip':
 282                                         print "skipping this one..."
 283                                         next = int(irow[1])
 284                                 else:
 285                                         if irow[3] == lasturl:
 286                                                 print "Looping the same, break break break!"
 287                                                 break
 288                                         lasturl = irow[3]
 289                                         print "inserting..."
 290                                         self.insert_row([irow[0],irow[1],irow[2],irow[3],irow[4],irow[5]])
 291                                         if len(irow) > 6:
 292                                                 next = irow[6]
 293                                                 if not next:
 294                                                         break
 295                                 if dateid:
 296                                         dt = string.split(comicid, "-")
 297                                         d = datetime.date(int(dt[0]),int(dt[1]),int(dt[2]))
 298                                         newer = d + datetime.timedelta( 1 )
 299                                         comicid = newer.isoformat()
 300                                 elif next:
 301                                         comicid = next
 302                                 else:
 303                                         comicid += 1
 304                         else:
 305                                 break
 306
 307
 308
 309
 310         def get_next_id(self, comic, number):
 311                 if comic == 'cyanide':
 312                         link = "http://www.explosm.net/comics/" + str(number) + "/"
 313                         print "link: " + link
 314                         try:
 315                                 f = urllib2.urlopen(link)
 316                                 hcode = f.code
 317                         except:
 318                                 hcode = 404
 319                         print "got hcode = " + str(hcode) + "\n"
 320                         if (hcode != 200):
 321                                 return False
 322                         else:
 323                                 # print "Cyanide & Happiness is unreliable, so we need to track next and prev"
 324                                 s = f.read()
 325                                 f.close()
 326                                 # title:
 327                                 splt = string.split(s, 'Previous</a> | <a href="/comics/', 1)
 328                                 if len(splt) < 2:
 329                                         print "no 'next' found"
 330                                         return False
 331                                 splt = string.split(splt[1], '/">Next >', 1)
 332                                 if len(splt) < 2:
 333                                         print "no 'next' found"
 334                                         return False
 335                                 else:
 336                                         print "got next: " + splt[0]
 337                                         return splt[0]
 338
 339         def get_prev_id(self, comic, number):
 340                 if comic == 'cyanide':
 341                         link = "http://www.explosm.net/comics/" + str(number) + "/"
 342                         print "link: " + link
 343                         try:
 344                                 f = urllib2.urlopen(link)
 345                                 hcode = f.code
 346                         except:
 347                                 hcode = 404
 348                         print "got hcode = " + str(hcode) + "\n"
 349                         if (hcode != 200):
 350                                 return False
 351                         else:
 352                                 print "Cyanide & Happiness is unreliable, so we need to track next and prev"
 353                                 s = f.read()
 354                                 f.close()
 355                                 # title:
 356                                 splt = string.split(s, 'First</a> | <a href="/comics/', 1)
 357                                 if len(splt) < 2:
 358                                         print "no 'prev' found"
 359                                         return False
 360                                 splt = string.split(splt[1], '/">< Previous', 1)
 361                                 if len(splt) < 2 or len(splt[0]) > 5:
 362                                         print "no 'next' found"
 363                                         return False
 364                                 else:
 365                                         print "got previous: " + splt[0]
 366                                         return splt[0]
 367
 368
 369
 370
 371
 372         def get_irow(self, comic, number):
 373                 dateid = False
 374                 if len(str(number)) == 10:
 375                         dateid = True
 376                 else:
 377                         if number < 0:
 378                                 return False
 379                         print "number is now: " + str(number) + "\n"
 380                 if comic == 'xkcd':
 381                         link = "http://xkcd.org/" + str(number) + "/"
 382                 elif comic == 'sinfest':
 383                         link = "http://www.sinfest.net/archive_page.php?comicID=" + str(number)
 384                 elif comic == 'dilbert':
 385                         link = "http://dilbert.com/" + str(number) + "/"
 386                 elif comic == 'phd':
 387                         link = "http://www.phdcomics.com/comics/archive.php?comicid=" + str(number)
 388                 elif comic == 'cyanide':
 389                         link = "http://www.explosm.net/comics/" + str(number) + "/"
 390                 else:
 391                         return False
 392
 393                 print "link: " + link
 394                 try:
 395                         f = urllib2.urlopen(link)
 396                         hcode = f.code
 397                 except:
 398                         hcode = 404
 399                 print "got hcode = " + str(hcode) + "\n"
 400                 if (hcode != 200):
 401                         return False
 402                 else:
 403                         # Build db entry.
 404                         # they look like: comic,id,link,url,filename,title
 405                         # We already have comic, number/id and link
 406
 407                         if comic == 'xkcd':
 408                                 s = f.read()
 409                                 f.close()
 410                                 # This should be done with regex but...
 411                                 splt = string.split(s, "<h3>Image URL (for hotlinking/embedding): ", 1)
 412                                 splt2 = string.split(splt[1], "</h3>", 1)
 413                                 url = splt2[0]
 414                                 splt = string.split(splt[0], "<h1>", 1)
 415                                 splt = string.split(splt[1], "</h1>", 1)
 416
 417                                 title = splt[0]
 418
 419                         elif comic == 'sinfest':
 420                                 s = f.read()
 421                                 f.close()
 422                                 splt = string.split(s, 'height="107"', 1)
 423                                 splt = string.split(splt[1], '<img src="', 1)
 424                                 splt = string.split(splt[1], '" alt="', 1)
 425                                 url = splt[0]
 426                                 if len(url) < 49:
 427                                         print "Fake 404! Break break break!"
 428                                         return False
 429                                 splt = string.split(splt[1], '" border="0" />', 1)
 430                                 title = splt[0]
 431
 432                         elif comic == 'dilbert':
 433                                 s = f.read()
 434                                 f.close()
 435                                 splt = string.split(s, 'input type="hidden" name="PrintPath" value="', 1)
 436                                 splt = string.split(splt[1], '" />', 1)
 437                                 url = "http://dilbert.com" + splt[0]
 438                                 if len(url) < 50:
 439                                         print "Fake 404! Break break break!"
 440                                         return False
 441                                 title = ""
 442
 443                         elif comic == 'phd':
 444                                 s = f.read()
 445                                 f.close()
 446                                 # title:
 447                                 splt = string.split(s, '<title>PHD Comics: ', 1)
 448                                 splt = string.split(splt[1], '</title>', 1)
 449                                 title = splt[0];
 450                                 splt = string.split(splt[1], '<td bgcolor=#FFFFFF align=center><img src=', 1)
 451                                 splt = string.split(splt[1], ' align=top>', 1)
 452                                 url = splt[0]
 453                                 if len(url) < 49:
 454                                         print "Fake 404! Break break break!"
 455                                         return False
 456
 457                         elif comic == 'cyanide':
 458                                 s = f.read()
 459                                 f.close()
 460                                 # title:
 461                                 splt = string.split(s, ' First</a> | <a href="/comics/', 1)
 462                                 if len(splt) < 2:
 463                                         print "first comic?"
 464                                         prev = False
 465                                 else:
 466                                         splt = string.split(splt[1], '/">< ', 1)
 467                                         prev = splt[0];
 468                                         print "got prev: " + prev
 469                                 splt = string.split(s, 'Previous</a> | <a href="/comics/', 1)
 470                                 if len(splt) < 2:
 471                                         print "last comic?"
 472                                         next = False
 473                                 else:
 474                                         splt = string.split(splt[1], '/">Next ></a>', 1)
 475                                         if len(splt[0]) > 10:
 476                                                 next = False
 477                                         else:
 478                                                 next = splt[0]
 479                                                 print "got next: " + next
 480
 481                                 splt = string.split(s, '</tr><tr><td colspan=2>', 1)
 482                                 if len(splt) < 2:
 483                                         print "no comic?"
 484                                         return False
 485                                 splt = string.split(splt[1], ' <b>by <a href="', 1)
 486                                 title = splt[0];
 487                                 if len(title) > 10:
 488                                         print "title wrong..."
 489                                         return False
 490                                 splt = string.split(splt[1], 'a daily webcomic" src="http://www.explosm', 1)
 491                                 if len(splt) < 2:
 492                                         print "a video? Try skipping"
 493                                         return ['skip',next,prev]
 494
 495                                 splt = string.split(splt[1], '">', 1)
 496                                 url = "http://www.explosm" + splt[0]
 497                                 splt2 = string.rsplit(url, "/", 1)
 498                                 filename = splt2[1]
 499                                 irow = [comic,number,link,url,filename,title,next,prev]
 500                                 return irow
 501
 502                         splt2 = string.rsplit(url, "/", 1)
 503                         filename = splt2[1]
 504                         if filename == self.db[0]['filename']:
 505                                 print "already exists! Break break break!"
 506                                 return False
 507                         irow = [comic,number,link,url,filename,title]
 508                         return irow
 509
 510
 511 # ------------UI
 512
 513 class ComicHomePlugin(hildondesktop.HomePluginItem):
 514         # __gtype_name__ = 'ComicHomePlugin'
 515         def __init__(self):
 516                 hildondesktop.HomePluginItem.__init__(self)
 517                 global supports_alpha
 518                 self.set_settings(True)
 519                 self.active_comics = self.get_active_comics()
 520                 self.sizes = self.get_size_settings()
 521                 self.new_width = self.sizes['width']
 522                 self.new_height = self.sizes['height']
 523                 self.connect("show-settings", self.show_options)
 524                 self.osso_c = osso.Context("comic-widget", "0.3.0", False)
 525                 self.osso_rpc = osso.Rpc(self.osso_c)
 526                 self.keypointer = 0
 527                 if len(self.active_comics) < 1:
 528                         self.active_comics = ['xkcd']
 529                 self.db = ComicDb(self.active_comics[self.keypointer])
 530                 self.comicname = comics[self.active_comics[self.keypointer]]['name']
 531                 self.set_name = "comicwidget"
 532                 screen = self.get_screen()
 533                 colormap = screen.get_rgba_colormap()
 534                 if colormap == None:
 535                         colormap = screen.get_rgb_colormap()
 536                         supports_alpha = False
 537                 else:
 538                         supports_alpha = True
 539                 self.set_colormap(colormap)
 540                 self.set_app_paintable(True)
 541                 #setup internal area
 542                 self.set_size_request(self.sizes['width'], self.sizes['height'])
 543                 self.imgvpos = 0;
 544                 self.vbox = gtk.VBox()
 545                 self.hbox = gtk.HBox()
 546                 self.label = gtk.Label()
 547                 self.label.set_alignment(xalign = 0, yalign = 0.5)
 548                 self.label.set_use_markup(True)
 549                 titl = str(self.db.db[self.db.currentcomic]['title'])
 550                 if len(titl) > 22:
 551                         titl = titl[:19] + "..."
 552                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 553
 554
 555                 self.e_goweb = gtk.EventBox()
 556                 self.e_goweb.set_name('goweb')
 557                 self.e_goweb.set_size_request(220, 48)
 558                 # e_open is the eventbox for the image. Does nothing, but it's convenient to have it.
 559                 # it will change to be a cairo thing when I have the time.
 560                 self.e_open = gtk.EventBox()
 561                 self.e_open.set_name('open')
 562                 self.e_switch = gtk.EventBox()
 563                 self.e_switch.set_name('switch')
 564                 self.e_switch.set_size_request(48, 48)
 565                 self.e_next = gtk.EventBox()
 566                 self.e_next.set_name('next')
 567                 self.e_next.set_size_request(48, 48)
 568                 self.e_prev = gtk.EventBox()
 569                 self.e_prev.set_name('prev')
 570                 self.e_prev.set_size_request(48, 48)
 571                 self.e_up = gtk.EventBox()
 572                 self.e_up.set_name('up')
 573                 self.e_up.set_size_request(48, 48)
 574                 self.e_down = gtk.EventBox()
 575                 self.e_down.set_name('down')
 576                 self.e_down.set_size_request(48, 48)
 577                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 578                 self.comic_image.show()
 579
 580
 581                 # Preload images into a dictionary, so that we can access them by the names of the eventboxes:
 582
 583                 self.images = {'up0':cairo.ImageSurface.create_from_png(imagedir + "up0-48x48.png"),
 584                                         'up1':cairo.ImageSurface.create_from_png(imagedir + "up1-48x48.png"),
 585                                         'down0':cairo.ImageSurface.create_from_png(imagedir + "down0-48x48.png"),
 586                                         'down1':cairo.ImageSurface.create_from_png(imagedir + "down1-48x48.png"),
 587                                         'next0':cairo.ImageSurface.create_from_png(imagedir + "next0-48x48.png"),
 588                                         'next1':cairo.ImageSurface.create_from_png(imagedir + "next1-48x48.png"),
 589                                         'prev0':cairo.ImageSurface.create_from_png(imagedir + "prev0-48x48.png"),
 590                                         'prev1':cairo.ImageSurface.create_from_png(imagedir + "prev1-48x48.png"),
 591                                         'switch0':cairo.ImageSurface.create_from_png(imagedir + "switch0-48x48.png"),
 592                                         'switch1':cairo.ImageSurface.create_from_png(imagedir + "switch1-48x48.png"),
 593                                         }
 594
 595
 596
 597                 self.e_goweb.add(self.label)
 598                 self.e_goweb.set_events(gtk.gdk.BUTTON_RELEASE_MASK)
 599                 self.e_goweb.connect("button-release-event", self.view_comic)
 600
 601                 self.e_open.add(self.comic_image)
 602
 603                 # self.e_switch.add(self.switch)
 604                 self.e_switch.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 605                 self.e_switch.connect("button-press-event", self.button_press)
 606                 self.e_switch.connect("button-release-event", self.button_release)
 607                 self.e_switch.connect("leave-notify-event", self.button_release)
 608
 609
 610                 # self.e_next.add(self.next)
 611                 self.e_next.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 612                 self.e_next.connect("button-press-event", self.button_press)
 613                 self.e_next.connect("button-release-event", self.button_release)
 614                 self.e_next.connect("leave-notify-event", self.button_release)
 615
 616                 #self.e_prev.add(self.prev)
 617                 self.e_prev.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 618                 self.e_prev.connect("button-press-event", self.button_press)
 619                 self.e_prev.connect("button-release-event", self.button_release)
 620                 self.e_prev.connect("leave-notify-event", self.button_release)
 621
 622                 #self.e_up.add(self.up)
 623                 self.e_up.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 624                 self.e_up.connect("button-press-event", self.button_press)
 625                 self.e_up.connect("button-release-event", self.button_release)
 626                 self.e_up.connect("leave-notify-event", self.button_release)
 627
 628
 629                 #self.e_down.add(self.down)
 630                 self.e_down.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
 631                 self.e_down.connect("button-press-event", self.button_press)
 632                 self.e_down.connect("button-release-event", self.button_release)
 633                 self.e_down.connect("leave-notify-event", self.button_release)
 634
 635                 self.vbox.pack_start(self.e_open,False,False,0)
 636                 self.vbox.pack_end(self.hbox,False,False,0)
 637
 638                 self.hbox.pack_start(self.e_switch,False,False,0)
 639                 self.hbox.pack_start(self.e_goweb,False,False,10)
 640                 self.hbox.pack_end(self.e_next,False,False,0)
 641                 self.hbox.pack_end(self.e_prev,False,False,0)
 642                 self.hbox.pack_end(self.e_down,False,False,0)
 643                 self.hbox.pack_end(self.e_up,False,False,0)
 644                 self.vbox.show_all()
 645                 self.add(self.vbox)
 646
 647
 648
 649         # **************** Cairo draw functions ***************
 650
 651         def realize(self, widget):
 652                 global supports_alpha
 653
 654                 # To check if the display supports alpha channels, get the colormap
 655                 screen = self.get_screen()
 656                 colormap = screen.get_rgba_colormap()
 657                 if colormap == None:
 658                         print 'Your screen does not support alpha channels!'
 659                         colormap = screen.get_rgb_colormap()
 660                         supports_alpha = False
 661                 else:
 662                 #      print 'Your screen supports alpha channels!'
 663                         supports_alpha = True
 664
 665                 # Now we have a colormap appropriate for the screen, use it
 666                 self.set_colormap(colormap)
 667
 668                 return False
 669
 670         def do_expose_event(widget, event):
 671                 cr = widget.window.cairo_create()
 672                 widget.draw(widget.e_up, 0)
 673                 widget.draw(widget.e_down, 0)
 674                 widget.draw(widget.e_prev,0)
 675                 widget.draw(widget.e_next, 0)
 676                 widget.draw(widget.e_switch, 0)
 677 #               widget.writeit(widget.e_goweb, "asdf")
 678                 cr.set_operator(cairo.OPERATOR_OVER)
 679                 region = gtk.gdk.region_rectangle(event.area)
 680                 cr.region(region)
 681                 bg_color=gtk.gdk.color_parse('#000000')
 682                 if supports_alpha:
 683                         cr.set_source_rgba (0.0, 0.0, 0.0, 0.5)
 684                 else:
 685                         cr.set_source_rgb (0.0, 0.0, 0.0)
 686
 687                 cr.fill_preserve()
 688                 return False
 689
 690         # Function that draws a button preloaded in the images dictionary.
 691         def draw(self, target, postfix):
 692                 cr_e = target.window.cairo_create()
 693                 cr_e.set_source_rgba(0.0, 0.0, 0.0, 0.5) # Transparent
 694                 cr_e.set_operator(cairo.OPERATOR_SOURCE)
 695                 cr_e.paint()
 696                 cr_e.set_operator(cairo.OPERATOR_OVER)
 697                 cr_e.set_source_surface(self.images[target.get_name() + str(postfix)], 0, 0)
 698                 cr_e.paint()
 699
 700         # ********** Event handlers *************
 701
 702
 703         # Most events happen on release, so I moved all of them into the same event handler.
 704         def button_release(self, widget, event):
 705                 if event.type == gtk.gdk.BUTTON_RELEASE:
 706                         func = widget.get_name()
 707                         print "pressed: " + func
 708                         if func == 'up':
 709                                 self.e_open.remove(self.comic_image)
 710                                 self.imgvpos = self.imgvpos - 80
 711                                 if self.imgvpos < 0:
 712                                         self.imgvpos = 0
 713                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 714                                 self.e_open.add(self.comic_image)
 715                                 self.e_open.show_all()
 716
 717                         elif func == 'down':
 718                                 self.e_open.remove(self.comic_image)
 719                                 self.imgvpos = self.imgvpos + 80
 720                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 721                                 self.e_open.add(self.comic_image)
 722                                 self.e_open.show_all()
 723
 724                         elif func == 'next':
 725                                 self.db.currentcomic -= 1
 726                                 self.imgvpos = 0
 727                                 self.e_open.remove(self.comic_image)
 728                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 729                                 self.e_open.add(self.comic_image)
 730                                 self.e_open.show_all()
 731                                 self.e_goweb.remove(self.label)
 732                                 titl = str(self.db.db[self.db.currentcomic]['title'])
 733                                 if len(titl) > 22:
 734                                         titl = titl[:19] + "..."
 735                                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 736                                 self.e_goweb.add(self.label)
 737                                 self.e_goweb.show_all()
 738
 739                         elif func == 'prev':
 740                                 self.db.currentcomic += 1
 741                                 self.imgvpos = 0
 742                                 self.e_open.remove(self.comic_image)
 743                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 744                                 self.e_open.add(self.comic_image)
 745                                 self.e_open.show_all()
 746                                 self.e_goweb.remove(self.label)
 747                                 titl = str(self.db.db[self.db.currentcomic]['title'])
 748                                 if len(titl) > 22:
 749                                         titl = titl[:19] + "..."
 750                                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 751                                 self.e_goweb.add(self.label)
 752                                 self.e_goweb.show_all()
 753
 754                         elif func == 'switch':
 755                                 self.active_comics = self.get_active_comics()
 756                                 print "closing log"
 757                                 #sys.stdout.close()
 758                                 #sys.stdout = sys.__stdout__
 759                                 #sys.stderr.close()
 760                                 #sys.stderr = sys.__stderr__
 761                                 print "active comics: " + str(self.active_comics)
 762                                 self.keypointer = (self.keypointer + 1) % len(self.active_comics)
 763                                 self.imgvpos = 0
 764                                 self.db = []
 765                                 print "switching to " + self.active_comics[self.keypointer]
 766                                 self.comicname = comics[self.active_comics[self.keypointer]]['name']
 767                                 self.db = ComicDb(self.active_comics[self.keypointer])
 768                                 self.e_open.remove(self.comic_image)
 769                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
 770                                 self.e_goweb.remove(self.label)
 771                                 titl = str(self.db.db[self.db.currentcomic]['title'])
 772                                 if len(titl) > 22:
 773                                         titl = titl[:19] + "..."
 774                                 self.label.set_markup('<span size="12000" face="monospace"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + titl + '</span>')
 775                                 self.e_goweb.add(self.label)
 776                                 self.e_goweb.show_all()
 777                                 self.e_open.add(self.comic_image)
 778                                 self.e_open.show_all()
 779
 780                         else:
 781                                 return False
 782                 self.draw(widget, "0")
 783
 784         # this one only switches to pressed button image
 785         def button_press(self, widget, event):
 786                 if not event.type == gtk.gdk.BUTTON_PRESS:
 787                         return False
 788                 self.draw(widget, "1")
 789
 790         # this one cancels a button press when pointer is moved out from it after press.
 791         def button_out(self, widget, event):
 792                 if not event.type == gtk.gdk.BUTTON_PRESS:
 793                         return False
 794                 self.draw(widget, "0")
 795
 796         # no fancy cairo stuff here. simply launches url in browser.
 797         def view_comic(self, widget, event):
 798                 if not event.type == gtk.gdk.BUTTON_RELEASE:
 799                         return False
 800                 link = self.db.get_link()
 801                 self.osso_rpc.rpc_run_with_defaults("osso_browser", "open_new_window", (link,))
 802
 803
 804
 805         # this function modifies the image to fit the widget and then returns it.
 806
 807         def get_resized_pixmap(self, filename, vpos):
 808                 print "vpos: " + str(vpos)
 809
 810                 width = int(self.sizes['width'])
 811                 height = int(self.sizes['height'])
 812                 picwidth = width - 2
 813                 maxheight = height - 48
 814
 815                 pixbuf = gtk.gdk.pixbuf_new_from_file(filename)
 816                 parent_buf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, pixbuf.get_has_alpha(), 8, width, maxheight)
 817                 parent_buf.fill(0xffffffff)
 818                 retimg = gtk.Image()
 819                 # get correct width/height
 820                 new_height = 1
 821                 new_width = picwidth
 822                 new_height = int((float(picwidth) / pixbuf.get_width()) * pixbuf.get_height())
 823                 scaled_buf = pixbuf.scale_simple(new_width,new_height,gtk.gdk.INTERP_BILINEAR)
 824                 print "scaled height: " + str(scaled_buf.get_height())
 825                 if scaled_buf.get_height() > maxheight:
 826                         if scaled_buf.get_height() - (vpos + maxheight) < 0:
 827                                 vpos = scaled_buf.get_height() - maxheight
 828                         scaled_buf.copy_area(0,vpos,picwidth,maxheight,parent_buf,1,0)
 829                         # parent_buf.composite(scaled_buf, 0, 0, 478, maxheight, 0, 0, 478, maxheight, gtk.gdk.INTERP_BILINEAR, 0)
 830                         self.imgvpos = vpos
 831                 else:
 832                         tmpy = maxheight - scaled_buf.get_height()
 833                         if tmpy > 0:
 834                                 tmpy = tmpy / 2
 835
 836                         scaled_buf.copy_area(0,0,picwidth,scaled_buf.get_height(),parent_buf,1,tmpy)
 837                         self.imgvpos = 0
 838 #               test = parent_buf.render_pixmap_and_mask(0)
 839 #               parent_buf.add_alpha(True,chr(255),chr(255),chr(255))
 840                 retimg.set_from_pixbuf(parent_buf)
 841                 print retimg.get_pixel_size()
 842                 print "vpos after fixing image: " + str(self.imgvpos)
 843                 return retimg
 844
 845         def screen_changed(self, widget, old_screen=None):
 846                 global supports_alpha
 847
 848                 # To check if the display supports alpha channels, get the colormap
 849                 screen = self.get_screen()
 850                 colormap = screen.get_rgba_colormap()
 851                 if colormap == None:
 852                         print 'Your screen does not support alpha channels!'
 853                         colormap = screen.get_rgb_colormap()
 854                         supports_alpha = False
 855                 else:
 856                 #      print 'Your screen supports alpha channels!'
 857                         supports_alpha = True
 858
 859                 # Now we have a colormap appropriate for the screen, use it
 860                 self.set_colormap(colormap)
 861
 862                 return False
 863
 864
 865         def get_active_comics(self):
 866                 print "get list"
 867
 868                 if os.path.isfile(activecomics) == True:
 869                         print "found " + activecomics
 870                         dbf = open(activecomics, 'r')
 871                         dbr = csv.reader(dbf)
 872                         try:
 873                                 active = dbr.next()
 874                         except:
 875                                 active = []
 876                         dbf.close()
 877                         return active
 878                 # if not, copy it.
 879                 else:
 880                         print "file " + activecomics + " not found"
 881                         if not os.path.exists(dbdir):
 882                                 print "path not found: " + dbdir
 883                                 try:
 884                                         os.makedirs(dbdir)
 885                                 except:
 886                                         print "settings dir creation failed. Exiting."
 887                                         exit()
 888                         try:
 889 #                               dbf = file(activecomics, 'w')
 890 #                               dbf.close()
 891                                 dbf = open(activecomics, 'w')
 892                         except:
 893                                 print "error storing settings"
 894                                 exit()
 895                         dbw = csv.writer(dbf)
 896                         dbw.writerow(comics.keys())
 897                         print comics.keys()
 898                         dbf.close()
 899                         ret = comics.keys()
 900                         return ret
 901
 902
 903
 904
 905                 #check if settings file exists
 906
 907                 #if not, write defaults
 908
 909 # ************************* OPTION DIALOGS ********************************
 910
 911         def show_options(self, widget):
 912                 print "loading options dialog"
 913                 dialog = gtk.Dialog("Comic Widget", None, gtk.DIALOG_DESTROY_WITH_PARENT)
 914
 915                 about_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
 916                 about_button.set_text("About", "See Author, Copyright and License information")
 917                 about_button.set_alignment(0,0,0,0)
 918                 about_button.connect("clicked", self.show_about)
 919
 920                 comics_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
 921                 comics_button.set_text("Comics", "Choose which comics to view")
 922                 comics_button.set_alignment(0,0,0,0)
 923                 comics_button.connect("clicked", self.show_comics)
 924
 925                 # size_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
 926                 # size_button.set_text("Change size", "Change the size of the widget")
 927                 # size_button.set_alignment(0,0,0,0)
 928                 # size_button.connect("clicked", self.show_size)
 929
 930                 print "adding buttons to dialog, starting with comics"
 931                 dialog.vbox.pack_start(comics_button, True, True, 0)
 932                 print "adding about button"
 933                 dialog.vbox.pack_start(about_button, True, True, 0)
 934                 # dialog.vbox.pack_start(size_button, True, True, 0)
 935                 print "show!"
 936                 dialog.show_all()
 937                 dialog.run()
 938                 dialog.destroy()
 939
 940         def show_about(self, widget):
 941                 print "in about dialog"
 942                 dialog = gtk.AboutDialog()
 943                 dialog.set_title("About")
 944                 dialog.set_name("Comic widget")
 945                 dialog.set_version(APP_VERSION)
 946                 dialog.set_copyright("Copyright 2010 Marcus Wikstrom")
 947                 dialog.set_authors(["Marcus Wikstrom <mece@ovi.com>\nLogo by Martin Wikstrom\n\nSpecial thanks to all the great python developers on t.m.o who share their code, and the helpful wizards on #maemo."])
 948                 dialog.set_logo(gtk.gdk.pixbuf_new_from_file("/opt/comic-widget/images/icon2-64x64.png"))
 949                 dialog.set_comments("Silly rabbit, tricks are for kids.")
 950                 dialog.set_license("""This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License. <http://www.gnu.org/licenses/>.""")
 951                 dialog.set_wrap_license(True)
 952                 dialog.show_all()
 953                 dialog.run()
 954                 dialog.destroy()
 955
 956         def show_comics(self, widget):
 957                 print "in comics dialog! yay!"
 958                 dialog = gtk.Dialog("Configure Search Engines", None, gtk.DIALOG_DESTROY_WITH_PARENT | gtk.DIALOG_NO_SEPARATOR)
 959                 comiclist = comics.keys()
 960                 comiclist.sort()
 961                 buttonlist = {}
 962                 self.connlist = {}
 963                 for comicid in comiclist:
 964                         buttonlist[comicid] = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
 965                         print buttonlist[comicid].get_name()
 966
 967                         print buttonlist[comicid].set_alignment(0, 0.5, 0, 0.5)
 968                         active = False
 969                         for acomic in self.active_comics:
 970                                 if acomic == comicid:
 971                                         active = True
 972                                         break
 973                         if active:
 974                                 buttonlist[comicid].set_text(comics[comicid]['name'] + " *", comicid + ": click to remove")
 975                                 self.connlist[comicid] = buttonlist[comicid].connect("clicked", self.remove_comic)
 976                         else:
 977                                 buttonlist[comicid].set_text(comics[comicid]['name'], comicid + ": click to add")
 978                                 self.connlist[comicid] = buttonlist[comicid].connect("clicked", self.add_comic)
 979                         dialog.vbox.pack_start(buttonlist[comicid], True, True, 0)
 980
 981
 982                 dialog.show_all()
 983                 dialog.run()
 984                 dialog.destroy()
 985
 986         def add_comic(self, widget):
 987                 print "add " + widget.get_title()
 988                 splt = string.split(widget.get_value(), ':', 1)
 989                 comicid = splt[0]
 990                 if os.path.isfile(activecomics) == True:
 991                         print "added " + comicid + " to " + str(self.active_comics)
 992                         try:
 993                                 found = self.active_comics.index(comicid)
 994                         except:
 995                                 self.active_comics.append(comicid)
 996                         dbf = open(activecomics, 'w')
 997                         dbw = csv.writer(dbf)
 998                         dbw.writerow(self.active_comics)
 999                         dbf.close()
1000                 widget.set_text(comics[comicid]['name'] + " *", comicid + ": click to remove")
1001                 widget.disconnect(self.connlist[comicid])
1002                 self.connlist[comicid] = widget.connect("clicked", self.remove_comic)
1003
1004         def remove_comic(self, widget):
1005                 print "remove " + widget.get_title()
1006                 if len(self.active_comics) == 1:
1007                         return
1008                 splt = string.split(widget.get_value(), ':', 1)
1009                 comicid = splt[0]
1010                 if os.path.isfile(activecomics) == True:
1011                         print "removing " + comicid + " from " + str(self.active_comics)
1012                         try:
1013                                 del self.active_comics[self.active_comics.index(comicid)]
1014                         except:
1015                                 pass
1016                         dbf = open(activecomics, 'w')
1017                         dbw = csv.writer(dbf)
1018                         dbw.writerow(self.active_comics)
1019                         dbf.close()
1020                 widget.set_text(comics[comicid]['name'], comicid + ": click to add")
1021                 widget.disconnect(self.connlist[comicid])
1022                 self.connlist[comicid] = widget.connect("clicked", self.add_comic)
1023
1024         def get_size_settings(self):
1025                 # defaults = {'width':480,'height':230}
1026                 # options: w 798 700 600 500 480 400
1027                 # options: h 420 360 310 270 240 200
1028                 print "checking to see if config file is available"
1029                 if os.path.isfile(sizefile) == True:
1030                         print "found " + sizefile
1031                         dbf = open(sizefile, 'r')
1032                         dbr = csv.reader(dbf)
1033                         try:
1034                                 cfg = dbr.next()
1035                         except:
1036                                 dbf.close()
1037                                 print "empty config file"
1038                                 return defaults
1039                         dbf.close()
1040
1041                         sizes = {'width':int(cfg[0]),'height':int(cfg[1])}
1042                         if sizes['width'] > 798:
1043                                 sizes['width'] = 798
1044                         if sizes['width'] < 400:
1045                                 sizes['width'] = 400
1046                         if sizes['height'] > 420:
1047                                 sizes['height'] = 420
1048                         if sizes['height'] < 200:
1049                                 sizes['height'] = 200
1050
1051                         return sizes
1052
1053
1054                 # if not, copy it.
1055                 else:
1056                         print "no config file"
1057                         return defaults
1058
1059 hd_plugin_type = ComicHomePlugin
1060
1061
1062 if __name__ == "__main__":
1063         import gobject
1064         gobject.type_register(hd_plugin_type)
1065         obj = gobject.new(hd_plugin_type, plugin_id="plugin_id")
1066         obj.show_all()
1067         gtk.main()