mirror of
https://github.com/samsonjs/vdirsyncer.git
synced 2026-03-27 09:15:50 +00:00
134 lines
3.8 KiB
Python
134 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
vdirsyncer.utils.vobject
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
:copyright: (c) 2014 Markus Unterwaditzer & contributors
|
|
:license: MIT, see LICENSE for more details.
|
|
'''
|
|
import hashlib
|
|
|
|
import icalendar.cal
|
|
import icalendar.parser
|
|
|
|
from . import text_type, itervalues
|
|
|
|
IGNORE_PROPS = frozenset((
|
|
# PRODID is changed by radicale for some reason after upload
|
|
'PRODID',
|
|
# VERSION can get lost in singlefile storage
|
|
'VERSION',
|
|
# X-RADICALE-NAME is used by radicale, because hrefs don't really exist in
|
|
# their filesystem backend
|
|
'X-RADICALE-NAME',
|
|
# REV is from the VCARD specification and is supposed to change when the
|
|
# item does -- however, we can determine that ourselves
|
|
'REV'
|
|
))
|
|
|
|
|
|
def normalize_item(text, ignore_props=IGNORE_PROPS):
|
|
try:
|
|
lines = to_unicode_lines(icalendar.cal.Component.from_ical(text))
|
|
except Exception:
|
|
lines = sorted(text.splitlines())
|
|
|
|
return u'\r\n'.join(line.strip()
|
|
for line in lines
|
|
if line.strip() and
|
|
not any(line.startswith(p + ':')
|
|
for p in IGNORE_PROPS))
|
|
|
|
|
|
def hash_item(text):
|
|
return hashlib.sha256(normalize_item(text).encode('utf-8')).hexdigest()
|
|
|
|
|
|
def split_collection(text, inline=(u'VTIMEZONE',),
|
|
wrap_items_with=(u'VCALENDAR',)):
|
|
'''Emits items in the order they occur in the text.'''
|
|
assert isinstance(text, text_type)
|
|
collection = icalendar.cal.Component.from_ical(text)
|
|
items = collection.subcomponents
|
|
|
|
if collection.name in wrap_items_with:
|
|
start = u'BEGIN:{}'.format(collection.name)
|
|
end = u'END:{}'.format(collection.name)
|
|
else:
|
|
start = end = u''
|
|
|
|
inlined_items = {}
|
|
for item in items:
|
|
if item.name in inline:
|
|
inlined_items[item.name] = item
|
|
|
|
for item in items:
|
|
if item.name not in inline:
|
|
lines = []
|
|
lines.append(start)
|
|
for inlined_item in itervalues(inlined_items):
|
|
lines.extend(to_unicode_lines(inlined_item))
|
|
|
|
lines.extend(to_unicode_lines(item))
|
|
lines.append(end)
|
|
|
|
yield u''.join(line + u'\r\n' for line in lines if line)
|
|
|
|
|
|
def to_unicode_lines(item):
|
|
'''icalendar doesn't provide an efficient way of getting the ical data as
|
|
unicode. So let's do it ourselves.'''
|
|
|
|
for content_line in item.content_lines():
|
|
if content_line:
|
|
yield icalendar.parser.foldline(content_line)
|
|
|
|
|
|
def join_collection(items, wrappers={
|
|
u'VCALENDAR': (u'VCALENDAR', (u'VTIMEZONE',)),
|
|
u'VCARD': (u'VADDRESSBOOK', ())
|
|
}):
|
|
'''
|
|
:param wrappers: {
|
|
item_type: wrapper_type, items_to_inline
|
|
}
|
|
'''
|
|
inline = {}
|
|
components = []
|
|
wrapper_type = None
|
|
inline_types = None
|
|
item_type = None
|
|
|
|
def handle_item(item):
|
|
if item.name in inline_types:
|
|
inline[item.name] = item
|
|
else:
|
|
components.append(item)
|
|
|
|
for item in items:
|
|
component = icalendar.cal.Component.from_ical(item)
|
|
|
|
if item_type is None:
|
|
item_type = component.name
|
|
wrapper_type, inline_types = wrappers[item_type]
|
|
|
|
if component.name == item_type:
|
|
if item_type == wrapper_type:
|
|
for subcomponent in component.subcomponents:
|
|
handle_item(subcomponent)
|
|
else:
|
|
handle_item(component)
|
|
|
|
start = end = u''
|
|
if wrapper_type is not None:
|
|
start = u'BEGIN:{}'.format(wrapper_type)
|
|
end = u'END:{}'.format(wrapper_type)
|
|
|
|
lines = [start]
|
|
for inlined_item in itervalues(inline):
|
|
lines.extend(to_unicode_lines(inlined_item))
|
|
for component in components:
|
|
lines.extend(to_unicode_lines(component))
|
|
lines.append(end)
|
|
|
|
return u''.join(line + u'\r\n' for line in lines if line)
|