mirror of
https://github.com/lucaspalomodevelop/indico-plugins.git
synced 2026-03-12 23:27:22 +00:00
96 lines
3.9 KiB
Python
96 lines
3.9 KiB
Python
# This file is part of the Indico plugins.
|
|
# Copyright (C) 2002 - 2022 CERN
|
|
#
|
|
# The Indico plugins are free software; you can redistribute
|
|
# them and/or modify them under the terms of the MIT License;
|
|
# see the LICENSE file for more details.
|
|
|
|
from datetime import timedelta
|
|
|
|
from werkzeug.datastructures import ImmutableDict
|
|
|
|
from indico.core.db import db
|
|
from indico.modules.attachments.models.attachments import Attachment
|
|
from indico.modules.categories import Category
|
|
from indico.modules.events import Event
|
|
from indico.modules.events.contributions.models.contributions import Contribution
|
|
from indico.modules.events.contributions.models.subcontributions import SubContribution
|
|
from indico.modules.events.notes.models.notes import EventNote
|
|
from indico.modules.events.sessions.models.sessions import Session
|
|
from indico.util.caching import memoize_request
|
|
from indico.util.date_time import now_utc
|
|
|
|
|
|
def obj_ref(obj):
|
|
"""Return a tuple identifying a category/event/contrib/subcontrib/note."""
|
|
from indico_livesync.models.queue import EntryType
|
|
if isinstance(obj, Category):
|
|
ref = {'type': EntryType.category, 'category_id': obj.id}
|
|
elif isinstance(obj, Event):
|
|
ref = {'type': EntryType.event, 'event_id': obj.id}
|
|
elif isinstance(obj, Session):
|
|
ref = {'type': EntryType.session, 'session_id': obj.id}
|
|
elif isinstance(obj, Contribution):
|
|
ref = {'type': EntryType.contribution, 'contrib_id': obj.id}
|
|
elif isinstance(obj, SubContribution):
|
|
ref = {'type': EntryType.subcontribution, 'subcontrib_id': obj.id}
|
|
elif isinstance(obj, EventNote):
|
|
ref = {'type': EntryType.note, 'note_id': obj.id}
|
|
elif isinstance(obj, Attachment):
|
|
ref = {'type': EntryType.attachment, 'attachment_id': obj.id}
|
|
else:
|
|
raise ValueError(f'Unexpected object: {obj.__class__.__name__}')
|
|
return ImmutableDict(ref)
|
|
|
|
|
|
@memoize_request
|
|
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']))
|
|
|
|
|
|
def clean_old_entries():
|
|
"""Delete obsolete entries from the queues."""
|
|
from indico_livesync.models.queue import LiveSyncQueueEntry
|
|
from indico_livesync.plugin import LiveSyncPlugin
|
|
|
|
queue_entry_ttl = LiveSyncPlugin.settings.get('queue_entry_ttl')
|
|
if not queue_entry_ttl:
|
|
return
|
|
expire_threshold = now_utc() - timedelta(days=queue_entry_ttl)
|
|
LiveSyncQueueEntry.query.filter(LiveSyncQueueEntry.processed,
|
|
LiveSyncQueueEntry.timestamp < expire_threshold).delete(synchronize_session='fetch')
|
|
|
|
|
|
@memoize_request
|
|
def get_excluded_categories(*, deep=False):
|
|
"""Get excluded category IDs.
|
|
|
|
:param deep: Whether to get all subcategory ids as well
|
|
"""
|
|
from indico_livesync.plugin import LiveSyncPlugin
|
|
ids = {int(x['id']) for x in LiveSyncPlugin.settings.get('excluded_categories')}
|
|
if not deep or not ids:
|
|
return ids
|
|
cte = Category.get_tree_cte()
|
|
query = (db.session.query(Category.id)
|
|
.join(cte, Category.id == cte.c.id)
|
|
.filter(cte.c.path.overlap(ids), ~cte.c.is_deleted))
|
|
return {x.id for x in query}
|