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