combo/combo/apps/fargo/models.py

91 lines
3.3 KiB
Python

# combo - content management system
# Copyright (C) 2014-2016 Entr'ouvert
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from django.conf import settings
from django.db import models
from django.forms import models as model_forms
from django.forms import Select
from django.utils.translation import ugettext_lazy as _
from combo.data.models import CellBase
from combo.data.library import register_cell_class
from combo.utils import requests
@register_cell_class
class RecentDocumentsCell(CellBase):
template_name = 'combo/fargo/recent-documents-cell.html'
fargo_site = models.CharField(_('Site'), max_length=50, blank=True)
class Meta:
verbose_name = _('Recent Documents')
def get_form_fields(self):
if len(get_fargo_services()) == 1:
return []
return ['fargo_site']
def get_form_widgets(self):
if len(get_fargo_services()) == 1:
return {}
combo_fargo_sites = [('', _('Default'))]
fargo_sites = [(x, y.get('title')) for x, y in get_fargo_services().items()]
fargo_sites.sort(key=lambda x: x[1])
combo_fargo_sites.extend(fargo_sites)
return {'fargo_site': Select(choices=combo_fargo_sites)}
def get_default_form_class(self):
return model_forms.modelform_factory(self.__class__,
fields=self.get_form_fields(),
widgets=self.get_form_widgets())
def is_visible(self, user=None):
if not user or user.is_anonymous:
return False
return super(RecentDocumentsCell, self).is_visible(user)
@classmethod
def is_enabled(cls):
return hasattr(settings, 'KNOWN_SERVICES') and settings.KNOWN_SERVICES.get('fargo')
def get_json(self, path, context):
remote_service = get_fargo_site(self.fargo_site)
response = requests.get(path,
remote_service=get_fargo_site(self.fargo_site),
user=self.get_concerned_user(context),
raise_if_not_cached=not(context.get('synchronous')),
headers={'accept': 'application/json'})
if response.status_code == 200:
return response.json()
return {}
def render(self, context):
context.update(self.get_json('api/documents/recently-added/', context))
return super(RecentDocumentsCell, self).render(context)
def get_fargo_services():
return settings.KNOWN_SERVICES.get('fargo') or []
def get_fargo_site(fargo_site):
if fargo_site:
return settings.KNOWN_SERVICES['fargo'].get(fargo_site)
for site in settings.KNOWN_SERVICES['fargo'].values():
if not site.get('secondary'):
return site
return settings.KNOWN_SERVICES['fargo'].values()[0]