Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Upgrade export VCalendar to v2.0, so can export utf8

Browse Source 
Fixes #11316
This commit is contained in:
prculley 2019-09-21 09:56:33 -05:00 committed by Nick Hall
parent 57d8dcd574
commit 0cd6394b72
1 changed files with 57 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
gramps/plugins/export/exportvcalendar.py
View File

@ -30,7 +30,7 @@
#------------------------------------------------------------------------- #-------------------------------------------------------------------------
import os import os
import sys import sys
from time import localtime import time
#------------------------------------------------------------------------ #------------------------------------------------------------------------
# #
@ -51,10 +51,10 @@ _ = glocale.translation.gettext
from gramps.gui.plug.export import WriterOptionBox from gramps.gui.plug.export import WriterOptionBox
from gramps.gen.utils.db import family_name from gramps.gen.utils.db import family_name
from gramps.gen.lib import Date, EventType from gramps.gen.lib import Date, EventType
from gramps.gui.glade import Glade
from gramps.gen.display.name import displayer as name_displayer from gramps.gen.display.name import displayer as name_displayer
from gramps.gen.display.place import displayer as _pd from gramps.gen.display.place import displayer as _pd
class CalendarWriter: class CalendarWriter:
def __init__(self, database, filename, user, option_box=None): def __init__(self, database, filename, user, option_box=None):
self.db = database self.db = database
@ -95,24 +95,27 @@ class CalendarWriter:
def export_data(self, filename): def export_data(self, filename):
self.dirname = os.path.dirname (filename) self.dirname = os.path.dirname(filename)
try: try:
with open(filename,"w") as self.g: with open(filename, "w", encoding='utf8',
newline='\r\n') as self.g:
self.writeln("BEGIN:VCALENDAR") self.writeln("BEGIN:VCALENDAR")
self.writeln("PRODID:-//GNU//Gramps//EN") self.writeln("PRODID:-//GNU//Gramps//EN")
self.writeln("VERSION:1.0") self.writeln("VERSION:2.0")
self.total = (len([x for x in self.db.iter_person_handles()]) + p_hndls = self.db.get_person_handles()
len([x for x in self.db.iter_family_handles()])) p_hndls.sort()
for key in self.db.iter_person_handles(): f_hndls = self.db.get_family_handles()
f_hndls.sort()
self.total = len(p_hndls) + len(f_hndls)
for key in p_hndls:
self.write_person(key) self.write_person(key)
self.update() self.update()
for key in self.db.iter_family_handles(): for key in f_hndls:
self.write_family(key) self.write_family(key)
self.update() self.update()
self.writeln("")
self.writeln("END:VCALENDAR") self.writeln("END:VCALENDAR")
return True return True
@ -129,16 +132,11 @@ class CalendarWriter:
if family: if family:
for event_ref in family.get_event_ref_list(): for event_ref in family.get_event_ref_list():
event = self.db.get_event_from_handle(event_ref.ref) event = self.db.get_event_from_handle(event_ref.ref)
if event.get_type() == EventType.MARRIAGE: if event and event.get_type() == EventType.MARRIAGE:
m_date = event.get_date_object()
place_handle = event.get_place_handle()
# feature requests 2356, 1657: avoid genitive form # feature requests 2356, 1657: avoid genitive form
text = "%s - %s" % (family_name(family, self.db), _("Marriage")) text = "%s - %s" % (family_name(family, self.db),
if place_handle: _("Marriage"))
place_title = _pd.display_event(self.db, event) self.write_vevent(text, event)
self.write_vevent( text, m_date, place_title)
else:
self.write_vevent( text, m_date)
def write_person(self, person_handle): def write_person(self, person_handle):
person = self.db.get_person_from_handle(person_handle) person = self.db.get_person_from_handle(person_handle)
@ -147,45 +145,26 @@ class CalendarWriter:
if birth_ref: if birth_ref:
birth = self.db.get_event_from_handle(birth_ref.ref) birth = self.db.get_event_from_handle(birth_ref.ref)
if birth: if birth:
b_date = birth.get_date_object()
place_handle = birth.get_place_handle()
if place_handle:
# feature requests 2356, 1657: avoid genitive form
place_title = _pd.display_event(self.db, birth)
self.write_vevent("%s - %s" %
(name_displayer.display(person), _("Birth")),
b_date, place_title)
else:
# feature requests 2356, 1657: avoid genitive form # feature requests 2356, 1657: avoid genitive form
self.write_vevent("%s - %s" % self.write_vevent("%s - %s" %
(name_displayer.display(person), _("Birth")), (name_displayer.display(person),
b_date) _("Birth")), birth)
death_ref = person.get_death_ref() death_ref = person.get_death_ref()
if death_ref: if death_ref:
death = self.db.get_event_from_handle(death_ref.ref) death = self.db.get_event_from_handle(death_ref.ref)
if death: if death:
d_date = death.get_date_object()
place_handle = death.get_place_handle()
if place_handle:
# feature requests 2356, 1657: avoid genitive form
place_title = _pd.display_event(self.db, death)
self.write_vevent("%s - %s" %
(name_displayer.display(person), _("Death")),
d_date, place_title)
else:
# feature requests 2356, 1657: avoid genitive form # feature requests 2356, 1657: avoid genitive form
self.write_vevent("%s - %s" % self.write_vevent("%s - %s" %
(name_displayer.display(person), _("Death")), (name_displayer.display(person),
d_date) _("Death")), death)
def format_single_date(self, subdate, thisyear, cal): def format_single_date(self, subdate, thisyear, cal):
retval = "" retval = ""
(day, month, year, sl) = subdate (day, month, year, sl) = subdate
if thisyear: if thisyear:
year = localtime().tm_year year = time.localtime().tm_year
if not cal == Date.CAL_GREGORIAN: if not cal == Date.CAL_GREGORIAN:
return "" return ""
@ -196,7 +175,6 @@ class CalendarWriter:
retval = "%s%02d%02d" % (year, month, day) retval = "%s%02d%02d" % (year, month, day)
return retval return retval
def format_date(self, date, thisyear=0): def format_date(self, date, thisyear=0):
retval = "" retval = ""
if date.get_modifier() == Date.MOD_TEXTONLY: if date.get_modifier() == Date.MOD_TEXTONLY:
@ -220,23 +198,43 @@ class CalendarWriter:
start) start)
return retval return retval
def write_vevent(self, event_text, date, location=""): def write_vevent(self, event_text, event):
date_string = self.format_date(date, 1) date = event.get_date_object()
place_handle = event.get_place_handle()
date_string = self.format_date(date, 0)
if date_string is not "": if date_string is not "":
self.writeln("") # self.writeln("")
self.writeln("BEGIN:VEVENT") self.writeln("BEGIN:VEVENT")
self.writeln("SUMMARY:%s %s" % (date.get_year(), event_text)) time_s = time.gmtime(event.change)
self.writeln("DTSTAMP:%04d%02d%02dT%02d%02d%02dZ" % time_s[0:6])
self.writeln("UID:%s@gramps.com" % event.handle)
self.writeln(fold("SUMMARY:%s %s" % (date.get_year(), event_text)))
if place_handle:
location = _pd.display_event(self.db, event)
if location: if location:
self.writeln("LOCATION:%s" % location) self.writeln("LOCATION:%s" % location)
self.writeln("RRULE:FREQ=YEARLY") self.writeln("RRULE:FREQ=YEARLY")
self.writeln(date_string) self.writeln(date_string)
self.writeln("END:VEVENT") self.writeln("END:VEVENT")
#------------------------------------------------------------------------- # -------------------------------------------------------------------------
#
#
# def fold(txt):
#------------------------------------------------------------------------- """ Limit line length to 75 octets (per RFC 5545) """
l_len = 0
text = ''
for char in txt:
c_len = len(char.encode('utf8'))
if c_len + l_len > 75:
l_len = 1
text += '\n ' + char
else:
l_len += c_len
text += char
return text
def exportData(database, filename, user, option_box=None): def exportData(database, filename, user, option_box=None):
cw = CalendarWriter(database, filename, user, option_box) cw = CalendarWriter(database, filename, user, option_box)
return cw.export_data(filename) return cw.export_data(filename)