def _process_args(self):
RHUserBase._process_args(self)
self.category = (Category.get_or_404(request.view_args['category_id'])
if 'category_id' in request.view_args else None)
self.suggestion = (self.user.suggested_categories.filter_by(
category=self.category).first()
if 'category_id' in request.view_args else None)
def _process_args(self):
if 'confId' in request.view_args:
self.obj = Event.get_or_404(request.view_args['confId'],
is_deleted=False)
self.obj_type = 'event'
elif 'category_id' in request.view_args:
self.obj = Category.get_or_404(request.view_args['category_id'],
is_deleted=False)
self.obj_type = 'category' if not self.obj.is_root else None
else:
self.obj = Category.get_root()
self.obj_type = None
def obj_deref(ref):
"""Return the object identified by `ref`."""
from indico_livesync.models.queue import EntryType
if ref['type'] == EntryType.category:
return Category.get_or_404(ref['category_id'])
elif ref['type'] == EntryType.event:
return Event.get_or_404(ref['event_id'])
elif ref['type'] == EntryType.session:
return Session.get_or_404(ref['session_id'])
elif ref['type'] == EntryType.contribution:
return Contribution.get_or_404(ref['contrib_id'])
elif ref['type'] == EntryType.subcontribution:
return SubContribution.get_or_404(ref['subcontrib_id'])
elif ref['type'] == EntryType.note:
return EventNote.get_or_404(ref['note_id'])
elif ref['type'] == EntryType.attachment:
return Attachment.get_or_404(ref['attachment_id'])
else:
raise ValueError('Unexpected object type: {}'.format(ref['type']))
