From e8568f88420b4517f6be14c11d89220207bc88b6 Mon Sep 17 00:00:00 2001 From: Don Allingham Date: Fri, 1 Jun 2001 01:01:45 +0000 Subject: [PATCH] Handling images better. svn: r74 --- gramps/src/RelImage.py | 9 ++- gramps/src/plugins/ReadGedcom.py | 98 +++++++++++++++++++++++++++++--- 2 files changed, 97 insertions(+), 10 deletions(-) diff --git a/gramps/src/RelImage.py b/gramps/src/RelImage.py index 15df08453..af96a3061 100644 --- a/gramps/src/RelImage.py +++ b/gramps/src/RelImage.py @@ -115,9 +115,12 @@ def mk_thumb(source,dest,size): print cmd os.system(cmd) else: - im = Image.open(source) - im.thumbnail((size,size)) - im.save(dest,"JPEG") + try: + im = Image.open(source) + im.thumbnail((size,size)) + im.save(dest,"JPEG") + except: + pass #------------------------------------------------------------------------- # diff --git a/gramps/src/plugins/ReadGedcom.py b/gramps/src/plugins/ReadGedcom.py index e9ccac8cf..cb67f3719 100644 --- a/gramps/src/plugins/ReadGedcom.py +++ b/gramps/src/plugins/ReadGedcom.py @@ -36,6 +36,7 @@ import shutil from gtk import * from gnome.ui import * from libglade import * +import gnome.mime ANSEL = 1 UNICODE = 2 @@ -46,6 +47,9 @@ callback = None glade_file = None clear_data = 0 +photo_types = [ "jpeg", "bmp", "pict", "pntg", "tpic", "png", "gif", + "tiff", "pcx" ] + ged2rel = {} for val in const.personalConstantEvents.keys(): key = const.personalConstantEvents[val] @@ -67,11 +71,20 @@ nameRegexp = re.compile(r"([\S\s]*\S)?\s*/([^/]+)?/\s*,?\s*([\S]+)?") # #------------------------------------------------------------------------- def find_file(fullname,altpath): + print "trying",fullname if os.path.isfile(fullname): - return fullname + type = gnome.mime.type(fullname) + if type[0:6] != "image/": + return "" + else: + return fullname other = altpath + os.sep + os.path.basename(fullname) if os.path.isfile(other): - return other + type = gnome.mime.type(other) + if type[0:6] != "image/": + return "" + else: + return other else: return "" @@ -85,7 +98,6 @@ def importData(database, filename): global topDialog global glade_file global statusWindow - # add some checking here @@ -154,6 +166,7 @@ class GedcomParser: self.fmap = {} self.smap = {} self.nmap = {} + self.dir_path = os.path.dirname(file) f = open(file,"r") self.lines = f.readlines() f.close() @@ -216,6 +229,16 @@ class GedcomParser: self.update(self.errors_obj,str(self.error_count)) self.ignore_sub_junk(level) + #--------------------------------------------------------------------- + # + # + # + #--------------------------------------------------------------------- + def warn(self,msg): + self.error_text_obj.insert_defaults(msg) + self.error_count = self.error_count + 1 + self.update(self.errors_obj,str(self.error_count)) + #--------------------------------------------------------------------- # # @@ -424,6 +447,12 @@ class GedcomParser: event.setName("Divorce") self.family.setDivorce(event) self.parse_family_event(event,2) + elif matches[1] == "OBJE": + if matches[2] and matches[2][0] == '@': + self.barf(2) +# self.ignore_sub_junk(2) + else: + self.parse_family_object(2) elif matches[1] == "NOTE": if matches[2] and matches[2][0] != "@": note = matches[1] + self.parse_continue_data(1) @@ -486,7 +515,8 @@ class GedcomParser: self.ignore_sub_junk(1) elif matches[1] == "OBJE": if matches[2] and matches[2][0] == '@': - self.ignore_sub_junk(2) + self.barf(2) +# self.ignore_sub_junk(2) else: self.parse_person_object(2) elif matches[1] == "NOTE": @@ -625,7 +655,7 @@ class GedcomParser: while 1: matches = self.get_next() if matches[1] == "FORM": - form = matches[2] + form = string.lower(matches[2]) elif matches[1] == "TITL": title = matches[2] elif matches[1] == "FILE": @@ -638,11 +668,65 @@ class GedcomParser: else: self.barf(level+1) - if form == "URL": + if form == "url": url = Url(file,title) self.person.addUrl(url) + elif form in photo_types: + path = find_file(file,self.dir_path) + if path == "": + self.warn(_("Could not import %s: either the file could not be found, or it was not a valid image") % file + "\n") + else: + photo = Photo() + photo.setPath(path) + photo.setDescription(title) + self.person.addPhoto(photo) else: - print "*",form,title + self.warn(_("Could not import %s: currently an unknown file type") % \ + file + "\n") + + #--------------------------------------------------------------------- + # + # + # + #--------------------------------------------------------------------- + def parse_family_object(self,level): + form = "" + file = "" + title = "" + note = "" + while 1: + matches = self.get_next() + if matches[1] == "FORM": + form = string.lower(matches[2]) + elif matches[1] == "TITL": + title = matches[2] + elif matches[1] == "FILE": + file = matches[2] + elif matches[1] == "NOTE": + note = matches[2] + self.parse_continue_data(level+1) + elif int(matches[0]) < level: + self.backup() + break + else: + self.barf(level+1) + + if form == "url": + pass +# url = Url(file,title) +# self.family.addUrl(url) + elif form in photo_types: + path = find_file(file,self.dir_path) + if path == "": + self.warn("Could not import %s: the file could not be found\n" % \ + file) + else: + photo = Photo() + photo.setPath(path) + photo.setDescription(title) + self.family.addPhoto(photo) + else: + self.warn("Could not import %s: current an unknown file type\n" % \ + file) #--------------------------------------------------------------------- #