lucaspalomodevelop/indico-plugins
1
0
Fork 0
mirror of https://github.com/lucaspalomodevelop/indico-plugins.git synced 2026-03-12 23:27:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add iter_subentries to generate nested entries

Browse Source
This commit is contained in:
Alejandro Avilés 2014-11-04 17:30:04 +01:00
parent 71462cf69b
commit ac1939190f
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
livesync/indico_livesync/models/queue.py
View File

@ -139,3 +139,29 @@ class LiveSyncQueueEntry(db.Model):
entry = cls(agent=agent, change=change, **obj_ref)
db.session.add(entry)
db.session.flush()
def iter_subentries(self):
"""Iterates through all children"""
if self.type not in {'category', 'event', 'contribution'}:
return
data = {'change': self.change,
'category_id': self.category_id, 'event_id': self.event_id,
'contrib_id': self.contrib_id, 'subcontrib_id': self.subcontrib_id}
if self.type == 'category':
for event in self.object.iterAllConferences():
new_data = dict(data)
new_data['type'] = 'event'
new_data['event_id'] = event.getId()
yield LiveSyncQueueEntry(**new_data)
elif self.type == 'event':
for contrib in self.object.iterContributions():
new_data = dict(data)
new_data['type'] = 'contribution'
new_data['contrib_id'] = contrib.getId()
yield LiveSyncQueueEntry(**new_data)
elif self.type == 'contribution':
for subcontrib in self.object.iterSubContributions():
new_data = dict(data)
new_data['type'] = 'subcontribution'
new_data['subcontrib_id'] = subcontrib.getId()
yield LiveSyncQueueEntry(**new_data)