sjs/vdirsyncer
1
0
Fork 0
mirror of https://github.com/samsonjs/vdirsyncer.git synced 2026-04-27 14:57:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up split_collections

Browse source 
The additional function args aren't pretty, but I think this is easier
to grasp.
This commit is contained in:
Markus Unterwaditzer 2017-01-20 23:17:19 +01:00
parent 9c417248de
commit b10e82b344
1 changed files with 25 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
vdirsyncer/utils/vobject.py
View file

@ -125,36 +125,38 @@ def split_collection(text):
items = {} # uid => item items = {} # uid => item
ungrouped_items = [] ungrouped_items = []
def inner(item, main):
if item.name == u'VTIMEZONE':
inline.append(item)
elif item.name == u'VCARD':
ungrouped_items.append(item)
elif item.name in (u'VTODO', u'VEVENT', u'VJOURNAL'):
uid = item.get(u'UID', u'')
wrapper = _Component(main.name, main.props[:], [])
if uid.strip():
wrapper = items.setdefault(uid, wrapper)
else:
ungrouped_items.append(wrapper)
wrapper.subcomponents.append(item)
elif item.name in (u'VCALENDAR', u'VADDRESSBOOK'):
for subitem in item.subcomponents:
inner(subitem, item)
else:
raise ValueError('Unknown component: {}'
.format(item.name))
for main in _Component.parse(text, multiple=True): for main in _Component.parse(text, multiple=True):
inner(main, main) _split_collection_impl(main, main, inline, items, ungrouped_items)
for item in chain(items.values(), ungrouped_items): for item in chain(items.values(), ungrouped_items):
item.subcomponents.extend(inline) item.subcomponents.extend(inline)
yield u'\r\n'.join(item.dump_lines()) yield u'\r\n'.join(item.dump_lines())
def _split_collection_impl(item, main, inline, items, ungrouped_items):
if item.name == u'VTIMEZONE':
inline.append(item)
elif item.name == u'VCARD':
ungrouped_items.append(item)
elif item.name in (u'VTODO', u'VEVENT', u'VJOURNAL'):
uid = item.get(u'UID', u'')
wrapper = _Component(main.name, main.props[:], [])
if uid.strip():
wrapper = items.setdefault(uid, wrapper)
else:
ungrouped_items.append(wrapper)
wrapper.subcomponents.append(item)
elif item.name in (u'VCALENDAR', u'VADDRESSBOOK'):
for subitem in item.subcomponents:
_split_collection_impl(subitem, item, inline, items,
ungrouped_items)
else:
raise ValueError('Unknown component: {}'
.format(item.name))
_default_join_wrappers = { _default_join_wrappers = {
u'VCALENDAR': u'VCALENDAR', u'VCALENDAR': u'VCALENDAR',
u'VEVENT': u'VCALENDAR', u'VEVENT': u'VCALENDAR',