195789f288dddd8f9a02b40b466c2630da46edeb
[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(str(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, 'png" title="', 1)
486                                 splt = string.split(splt[1], '" alt="', 1)
487                                 title = splt[0]
488                                 splt = string.split(splt[1], "<h3>Image URL (for hotlinking/embedding): ", 1)
489                                 splt = string.split(splt[1], "</h3>", 1)
490                                 url = splt[0]
491
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                 
635                 self.db = ComicDb(self.active_comics[self.keypointer])
636                 self.comicname = comics[self.active_comics[self.keypointer]]['name']
637                 self.set_name = "comicwidget"
638                 screen = self.get_screen()
639                 colormap = screen.get_rgba_colormap()
640                 if colormap == None:
641                         colormap = screen.get_rgb_colormap()
642                         supports_alpha = False
643                 else:
644                         supports_alpha = True
645                 self.set_colormap(colormap)
646                 self.set_app_paintable(True)
647                 #setup internal area
648                 self.set_size_request(self.sizes['width'], self.sizes['height'])
649                 self.imgvpos = 0;
650                 self.vbox = gtk.VBox()
651                 self.hbox = gtk.HBox()
652                 self.label = gtk.Label()
653                 self.label.set_alignment(xalign = 0, yalign = 0.5)
654                 self.label.set_use_markup(True)
655                 self.label.set_markup(self.get_markup())
656
657
658                 self.e_goweb = gtk.EventBox()
659                 self.e_goweb.set_name('goweb')
660                 self.e_goweb.set_size_request(self.sizes['width'] - 250, 48)
661                 # e_open is the eventbox for the image. Does nothing, but it's convenient to have it.
662                 # it will change to be a cairo thing when I have the time.
663                 self.e_open = gtk.EventBox()
664                 self.e_open.set_name('open')
665                 self.e_switch = gtk.EventBox()
666                 self.e_switch.set_name('switch')
667                 self.e_switch.set_size_request(48, 48)
668                 self.e_next = gtk.EventBox()
669                 self.e_next.set_name('next')
670                 self.e_next.set_size_request(48, 48)
671                 self.e_prev = gtk.EventBox()
672                 self.e_prev.set_name('prev')
673                 self.e_prev.set_size_request(48, 48)
674                 self.e_up = gtk.EventBox()
675                 self.e_up.set_name('up')
676                 self.e_up.set_size_request(48, 48)
677                 self.e_down = gtk.EventBox()
678                 self.e_down.set_name('down')
679                 self.e_down.set_size_request(48, 48)
680                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
681                 self.comic_image.show()
682                 
683
684                 # Preload images into a dictionary, so that we can access them by the names of the eventboxes:
685
686                 self.images = {'up0':cairo.ImageSurface.create_from_png(imagedir + "up0-48x48.png"),
687                                         'up1':cairo.ImageSurface.create_from_png(imagedir + "up1-48x48.png"),
688                                         'down0':cairo.ImageSurface.create_from_png(imagedir + "down0-48x48.png"),
689                                         'down1':cairo.ImageSurface.create_from_png(imagedir + "down1-48x48.png"),
690                                         'next0':cairo.ImageSurface.create_from_png(imagedir + "next0-48x48.png"),
691                                         'next1':cairo.ImageSurface.create_from_png(imagedir + "next1-48x48.png"),
692                                         'prev0':cairo.ImageSurface.create_from_png(imagedir + "prev0-48x48.png"),
693                                         'prev1':cairo.ImageSurface.create_from_png(imagedir + "prev1-48x48.png"),
694                                         'switch0':cairo.ImageSurface.create_from_png(imagedir + "switch0-48x48.png"),
695                                         'switch1':cairo.ImageSurface.create_from_png(imagedir + "switch1-48x48.png"),
696                                         }
697
698
699
700                 self.e_goweb.add(self.label)
701                 self.e_goweb.set_events(gtk.gdk.BUTTON_RELEASE_MASK)
702                 self.e_goweb.connect("button-release-event", self.view_comic)
703
704                 self.e_open.add(self.comic_image)
705
706                 # self.e_switch.add(self.switch)
707                 self.e_switch.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
708                 self.e_switch.connect("button-press-event", self.button_press)
709                 self.e_switch.connect("button-release-event", self.button_release)
710                 self.e_switch.connect("leave-notify-event", self.button_release)
711
712
713                 # self.e_next.add(self.next)
714                 self.e_next.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
715                 self.e_next.connect("button-press-event", self.button_press)
716                 self.e_next.connect("button-release-event", self.button_release)
717                 self.e_next.connect("leave-notify-event", self.button_release)
718
719                 #self.e_prev.add(self.prev)
720                 self.e_prev.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
721                 self.e_prev.connect("button-press-event", self.button_press)
722                 self.e_prev.connect("button-release-event", self.button_release)
723                 self.e_prev.connect("leave-notify-event", self.button_release)
724
725                 #self.e_up.add(self.up)
726                 self.e_up.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
727                 self.e_up.connect("button-press-event", self.button_press)
728                 self.e_up.connect("button-release-event", self.button_release)
729                 self.e_up.connect("leave-notify-event", self.button_release)
730                 
731
732                 #self.e_down.add(self.down)
733                 self.e_down.set_events(gtk.gdk.BUTTON_PRESS_MASK|gtk.gdk.BUTTON_RELEASE_MASK|gtk.gdk.LEAVE_NOTIFY)
734                 self.e_down.connect("button-press-event", self.button_press)
735                 self.e_down.connect("button-release-event", self.button_release)
736                 self.e_down.connect("leave-notify-event", self.button_release)
737
738                 self.vbox.pack_start(self.e_open,False,False,0)
739                 self.vbox.pack_end(self.hbox,False,False,0)
740
741                 self.hbox.pack_start(self.e_switch,False,False,0)
742                 self.hbox.pack_start(self.e_goweb,False,False,5)
743                 self.hbox.pack_end(self.e_next,False,False,0)
744                 self.hbox.pack_end(self.e_prev,False,False,0)
745                 self.hbox.pack_end(self.e_down,False,False,0)
746                 self.hbox.pack_end(self.e_up,False,False,0)
747                 self.vbox.show_all()
748                 self.add(self.vbox)
749
750
751
752         # **************** Cairo draw functions ***************
753
754         def realize(self, widget):
755                 global supports_alpha
756
757                 # To check if the display supports alpha channels, get the colormap
758                 screen = self.get_screen()
759                 colormap = screen.get_rgba_colormap()
760                 if colormap == None:
761                         print 'Your screen does not support alpha channels!'
762                         colormap = screen.get_rgb_colormap()
763                         supports_alpha = False
764                 else:
765                 #      print 'Your screen supports alpha channels!'
766                         supports_alpha = True
767
768                 # Now we have a colormap appropriate for the screen, use it
769                 self.set_colormap(colormap)
770
771                 return False
772
773         def do_expose_event(widget, event):
774                 cr = widget.window.cairo_create()
775                 widget.draw(widget.e_up, 0)
776                 widget.draw(widget.e_down, 0)
777                 widget.draw(widget.e_prev,0)
778                 widget.draw(widget.e_next, 0)
779                 widget.draw(widget.e_switch, 0)
780 #               widget.writeit(widget.e_goweb, "asdf")
781                 cr.set_operator(cairo.OPERATOR_OVER)
782                 region = gtk.gdk.region_rectangle(event.area)
783                 cr.region(region)
784                 bg_color=gtk.gdk.color_parse('#000000')
785                 if supports_alpha:
786                         cr.set_source_rgba (0.0, 0.0, 0.0, 0.5)
787                 else:
788                         cr.set_source_rgb (0.0, 0.0, 0.0)
789                         
790                 cr.fill_preserve()
791                 return False
792
793         # Function that draws a button preloaded in the images dictionary.
794         def draw(self, target, postfix):
795                 cr_e = target.window.cairo_create()
796                 cr_e.set_source_rgba(0.0, 0.0, 0.0, 0.5) # Transparent
797                 cr_e.set_operator(cairo.OPERATOR_SOURCE)
798                 cr_e.paint()
799                 cr_e.set_operator(cairo.OPERATOR_OVER)
800                 cr_e.set_source_surface(self.images[target.get_name() + str(postfix)], 0, 0)
801                 cr_e.paint()
802
803         # ********** Event handlers *************
804
805
806         # Most events happen on release, so I moved all of them into the same event handler.
807         def button_release(self, widget, event):
808                 if event.type == gtk.gdk.BUTTON_RELEASE:
809                         func = widget.get_name()
810                         print "pressed: " + func
811                         if func == 'up':
812                                 self.e_open.remove(self.comic_image)
813                                 self.imgvpos = self.imgvpos - ((self.sizes['height'] - 48) / 2.1)
814                                 if self.imgvpos < 0:
815                                         self.imgvpos = 0
816                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
817                                 self.e_open.add(self.comic_image)
818                                 self.e_open.show_all()
819
820                         elif func == 'down':
821                                 self.e_open.remove(self.comic_image)
822                                 self.imgvpos = self.imgvpos + ((self.sizes['height'] - 48) / 2.1)
823                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
824                                 self.e_open.add(self.comic_image)
825                                 self.e_open.show_all()
826                         
827                         elif func == 'next':
828                                 self.db.currentcomic -= 1
829                                 self.imgvpos = 0
830                                 self.e_open.remove(self.comic_image)
831                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
832                                 self.e_open.add(self.comic_image)
833                                 self.e_open.show_all()
834                                 self.e_goweb.remove(self.label)
835                                 titl = str(self.db.db[self.db.currentcomic]['title'])
836                                 if len(titl) > 22:
837                                         titl = titl[:19] + "..."
838                                 self.label.set_markup(self.get_markup())
839                                 self.e_goweb.add(self.label)
840                                 self.e_goweb.show_all()
841
842                         elif func == 'prev':
843                                 self.db.currentcomic += 1
844                                 self.imgvpos = 0
845                                 self.e_open.remove(self.comic_image)
846                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
847                                 self.e_open.add(self.comic_image)
848                                 self.e_open.show_all()
849                                 self.e_goweb.remove(self.label)
850                                 titl = str(self.db.db[self.db.currentcomic]['title'])
851                                 if len(titl) > 22:
852                                         titl = titl[:19] + "..."
853                                 self.label.set_markup(self.get_markup())
854                                 self.e_goweb.add(self.label)
855                                 self.e_goweb.show_all()
856
857                         elif func == 'switch':
858                                 self.active_comics = self.get_active_comics()
859                                 print "closing log"
860                                 #sys.stdout.close()
861                                 #sys.stdout = sys.__stdout__
862                                 #sys.stderr.close()
863                                 #sys.stderr = sys.__stderr__
864                                 print "active comics: " + str(self.active_comics)
865                                 self.keypointer = (self.keypointer + 1) % len(self.active_comics)
866                                 self.imgvpos = 0
867                                 self.db = []
868                                 print "switching to " + self.active_comics[self.keypointer]
869                                 self.comicname = comics[self.active_comics[self.keypointer]]['name']
870                                 self.db = ComicDb(self.active_comics[self.keypointer])
871                                 self.e_open.remove(self.comic_image)
872                                 self.comic_image = self.get_resized_pixmap(self.db.get_comic(), self.imgvpos)
873                                 self.e_goweb.remove(self.label)
874                                 titl = str(self.db.db[self.db.currentcomic]['title'])
875                                 if len(titl) > 22:
876                                         titl = titl[:19] + "..."
877                                 self.label.set_markup(self.get_markup())
878                                 self.e_goweb.add(self.label)
879                                 self.e_goweb.show_all()
880                                 self.e_open.add(self.comic_image)
881                                 self.e_open.show_all()  
882
883                         else:
884                                 return False                    
885                 self.draw(widget, "0")
886
887         # this one only switches to pressed button image
888         def button_press(self, widget, event):
889                 if not event.type == gtk.gdk.BUTTON_PRESS:
890                         return False
891                 self.draw(widget, "1")
892
893         # this one cancels a button press when pointer is moved out from it after press.
894         def button_out(self, widget, event):
895                 if not event.type == gtk.gdk.BUTTON_PRESS:
896                         return False
897                 self.draw(widget, "0")
898
899         # no fancy cairo stuff here. simply launches url in browser.
900         def view_comic(self, widget, event):
901                 if not event.type == gtk.gdk.BUTTON_RELEASE:
902                         return False
903                 link = self.db.get_link()
904                 self.osso_rpc.rpc_run_with_defaults("osso_browser", "open_new_window", (link,))
905
906
907
908         # this function modifies the image to fit the widget and then returns it.
909
910         def get_resized_pixmap(self, filename, vpos):
911                 print "vpos: " + str(vpos)
912                 
913                 width = int(self.sizes['width'])
914                 height = int(self.sizes['height'])
915                 picwidth = width - 2
916                 maxheight = height - 48
917
918                 pixbuf = gtk.gdk.pixbuf_new_from_file(filename)
919                 parent_buf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, pixbuf.get_has_alpha(), 8, width, maxheight)
920                 parent_buf.fill(0xffffffff)
921                 retimg = gtk.Image()
922                 # get correct width/height
923                 new_height = 1
924                 new_width = picwidth
925                 new_height = int((float(picwidth) / pixbuf.get_width()) * pixbuf.get_height())
926                 scaled_buf = pixbuf.scale_simple(new_width,new_height,gtk.gdk.INTERP_BILINEAR)
927                 print "scaled height: " + str(scaled_buf.get_height())
928                 if scaled_buf.get_height() > maxheight:
929                         if scaled_buf.get_height() - (vpos + maxheight) < 0:
930                                 vpos = scaled_buf.get_height() - maxheight
931                         scaled_buf.copy_area(0,vpos,picwidth,maxheight,parent_buf,1,0)
932                         # parent_buf.composite(scaled_buf, 0, 0, 478, maxheight, 0, 0, 478, maxheight, gtk.gdk.INTERP_BILINEAR, 0)
933                         self.imgvpos = vpos
934                 else:
935                         tmpy = maxheight - scaled_buf.get_height()
936                         if tmpy > 0:
937                                 tmpy = tmpy / 2
938
939                         scaled_buf.copy_area(0,0,picwidth,scaled_buf.get_height(),parent_buf,1,tmpy)
940                         self.imgvpos = 0
941 #               test = parent_buf.render_pixmap_and_mask(0)
942 #               parent_buf.add_alpha(True,chr(255),chr(255),chr(255))
943                 retimg.set_from_pixbuf(parent_buf)
944                 print retimg.get_pixel_size()
945                 print "vpos after fixing image: " + str(self.imgvpos)
946                 return retimg
947
948         def screen_changed(self, widget, old_screen=None):
949                 global supports_alpha
950
951                 # To check if the display supports alpha channels, get the colormap
952                 screen = self.get_screen()
953                 colormap = screen.get_rgba_colormap()
954                 if colormap == None:
955                         print 'Your screen does not support alpha channels!'
956                         colormap = screen.get_rgb_colormap()
957                         supports_alpha = False
958                 else:
959                 #      print 'Your screen supports alpha channels!'
960                         supports_alpha = True
961
962                 # Now we have a colormap appropriate for the screen, use it
963                 self.set_colormap(colormap)
964
965                 return False
966
967
968         def get_active_comics(self):
969                 print "get list"
970
971                 if os.path.isfile(activecomics) == True:
972                         print "found " + activecomics
973                         dbf = open(activecomics, 'r')
974                         dbr = csv.reader(dbf)
975                         try:
976                                 active = dbr.next()
977                         except:
978                                 active = []
979                         dbf.close()
980                         return active
981                 # if not, copy it.
982                 else:
983                         print "file " + activecomics + " not found"
984                         if not os.path.exists(dbdir):
985                                 print "path not found: " + dbdir
986                                 try:
987                                         os.makedirs(dbdir)
988                                 except:
989                                         print "settings dir creation failed. Exiting."
990                                         exit()
991                         try:
992 #                               dbf = file(activecomics, 'w')
993 #                               dbf.close()
994                                 dbf = open(activecomics, 'w')
995                         except:
996                                 print "error storing settings"
997                                 exit()
998                         dbw = csv.writer(dbf)
999                         dbw.writerow(comics.keys())
1000                         print comics.keys()
1001                         dbf.close()
1002                         ret = comics.keys()
1003                         return ret
1004
1005                 
1006         def get_markup(self):
1007                 if len(self.comicname + " " + str(self.db.db[self.db.currentcomic]['id'])) < 20:
1008                         return '<span size="12000" face="arial"> <b>' + self.comicname + '</b> ' + str(self.db.db[self.db.currentcomic]['id']) + '\n ' + str(self.db.db[self.db.currentcomic]['title']) + '</span>'
1009                 else:
1010                         return '<span size="12000" face="arial"> <b>' + self.comicname + '</b>' + '\n ' + str(self.db.db[self.db.currentcomic]['title']) + '</span>'
1011
1012                 
1013                 #check if settings file exists
1014
1015                 #if not, write defaults
1016
1017 # ************************* OPTION DIALOGS ********************************
1018
1019         def show_options(self, widget):
1020                 print "loading options dialog"
1021                 dialog = gtk.Dialog("Comic Widget", None, gtk.DIALOG_DESTROY_WITH_PARENT)
1022
1023                 about_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
1024                 about_button.set_text("About", "See Author, Copyright and License information")
1025                 about_button.set_alignment(0,0,0,0)
1026                 about_button.connect("clicked", self.show_about)
1027
1028                 comics_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
1029                 comics_button.set_text("Comics", "Choose which comics to view")
1030                 comics_button.set_alignment(0,0,0,0)
1031                 comics_button.connect("clicked", self.show_comics)
1032
1033                 # size_button = hildon.Button(gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT, hildon.BUTTON_ARRANGEMENT_VERTICAL)
1034                 # size_button.set_text("Change size", "Change the size of the widget")
1035                 # size_button.set_alignment(0,0,0,0)
1036                 # size_button.connect("clicked", self.show_size)
1037
1038                 print "adding buttons to dialog, starting with comics"
1039                 dialog.vbox.pack_start(comics_button, True, True, 0)
1040                 print "adding about button"
1041                 dialog.vbox.pack_start(about_button, True, True, 0)
1042                 # dialog.vbox.pack_start(size_button, True, True, 0)
1043                 print "show!"
1044                 dialog.show_all()
1045                 dialog.run()
1046                 dialog.destroy()
1047
1048         def show_about(self, widget):
1049                 print "in about dialog"
1050                 dialog = gtk.AboutDialog()
1051                 dialog.set_title("About")
1052                 dialog.set_name("Comic widget")
1053                 dialog.set_version(APP_VERSION)
1054                 dialog.set_copyright("Copyright 2010 Marcus Wikstrom")
1055                 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."])
1056                 dialog.set_logo(gtk.gdk.pixbuf_new_from_file("/opt/comic-widget/images/icon2-64x64.png"))
1057                 dialog.set_comments("Silly rabbit, tricks are for kids.")
1058                 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/>.""")
1059                 dialog.set_wrap_license(True)
1060                 dialog.show_all()
1061                 dialog.run()
1062                 dialog.destroy()
1063
1064         def show_comics(self, widget, data = None):
1065                 print "in comics dialog! yay!"
1066                 dialog = gtk.Dialog("Choose comics", None, gtk.DIALOG_DESTROY_WITH_PARENT | gtk.DIALOG_NO_SEPARATOR)
1067                 comiclist = comics.keys()
1068                 comiclist.sort()
1069                 buttonlist = {}
1070                 self.connlist = {}
1071                 self.liststore = gtk.ListStore(str,str,bool)
1072                 for comicid in comiclist:
1073                         active = False
1074                         for acomic in self.active_comics:
1075                                 if acomic == comicid:
1076                                         active = True
1077                                         break
1078                         if active:
1079                                 self.liststore.append(["* " + comics[comicid]['name'],comicid, active])
1080                         else:
1081                                 self.liststore.append(["  " + comics[comicid]['name'],comicid, active])
1082                 self.treeview = hildon.GtkTreeView(gtk.HILDON_UI_MODE_EDIT)
1083                 self.treeview.set_model(self.liststore)
1084                 self.treeview.set_reorderable(True)
1085                 tvcolumn = gtk.TreeViewColumn('Column 0')
1086                 self.treeview.append_column(tvcolumn)
1087                 cell = gtk.CellRendererText()
1088                 tvcolumn.pack_start(cell, True)
1089                 tvcolumn.add_attribute(cell, 'text', 0)
1090
1091                 ts = hildon.PannableArea()
1092         
1093                 ts.set_size_request(-1, 280)
1094                 ts.add(self.treeview)
1095                 selection = self.treeview.get_selection()
1096
1097                 dialog.vbox.pack_start(ts,True,True,0)
1098                 dialog.show_all()
1099                 selection.connect("changed", self.pick_comic, comiclist)
1100                 dialog.run()
1101                 dialog.destroy()
1102
1103         def pick_comic(self, widget, data):
1104                 selected = widget.get_selected()
1105                 if selected == None:
1106                         return False
1107                 
1108                 print selected[0].get_value(selected[1],0) + ", " + selected[0].get_value(selected[1],1) + ", " + str(selected[0].get_value(selected[1],2))
1109                 if selected[0].get_value(selected[1],2):
1110                         self.remove_comic(selected[0].get_value(selected[1],1))
1111                         splt = selected[0].get_value(selected[1],0)[1:]
1112                         selected[0].set_value(selected[1],0," " + splt)
1113                         selected[0].set_value(selected[1],2,False)
1114                 else:
1115                         self.add_comic(selected[0].get_value(selected[1],1))
1116                         splt = selected[0].get_value(selected[1],0)[1:]
1117                         selected[0].set_value(selected[1],0,"*" + splt)
1118                         selected[0].set_value(selected[1],2,True)
1119
1120                                 
1121         def add_comic(self, comicid):
1122                 if os.path.isfile(activecomics) == True:
1123                         print "added " + comicid + " to " + str(self.active_comics)
1124                         try:
1125                                 found = self.active_comics.index(comicid)
1126                         except:
1127                                 self.active_comics.append(comicid)
1128                         dbf = open(activecomics, 'w')
1129                         dbw = csv.writer(dbf)
1130                         dbw.writerow(self.active_comics)
1131                         dbf.close()
1132
1133         def remove_comic(self, comicid):
1134                 if os.path.isfile(activecomics) == True:
1135                         print "removing " + comicid + " from " + str(self.active_comics)
1136                         try:
1137                                 del self.active_comics[self.active_comics.index(comicid)]
1138                         except:
1139                                 pass
1140                         dbf = open(activecomics, 'w')
1141                         dbw = csv.writer(dbf)
1142                         dbw.writerow(self.active_comics)
1143                         dbf.close()
1144
1145
1146         def get_size_settings(self):
1147                 # defaults = {'width':480,'height':230}
1148                 # options: w 798 700 600 500 480 400
1149                 # options: h 420 360 310 270 240 200
1150                 print "checking to see if config file is available"
1151                 if os.path.isfile(sizefile) == True:
1152                         print "found " + sizefile
1153                         dbf = open(sizefile, 'r')
1154                         dbr = csv.reader(dbf)
1155                         try:
1156                                 cfg = dbr.next()
1157                         except:
1158                                 dbf.close()
1159                                 print "empty config file"
1160                                 return defaults
1161                         dbf.close()
1162                         
1163                         sizes = {'width':int(cfg[0]),'height':int(cfg[1])}
1164                         if sizes['width'] > 798:
1165                                 sizes['width'] = 798
1166                         if sizes['width'] < 400:
1167                                 sizes['width'] = 400
1168                         if sizes['height'] > 420:
1169                                 sizes['height'] = 420
1170                         if sizes['height'] < 200:
1171                                 sizes['height'] = 200
1172                         
1173                         return sizes
1174                         
1175                         
1176                 # if not, copy it.
1177                 else:
1178                         print "no config file"
1179                         return defaults
1180                 
1181 hd_plugin_type = ComicHomePlugin
1182
1183
1184 if __name__ == "__main__":
1185         import gobject
1186         gobject.type_register(hd_plugin_type)
1187         obj = gobject.new(hd_plugin_type, plugin_id="plugin_id")
1188         obj.show_all()
1189         gtk.main()