njpma/app/text/models.py

import logging
import json

from django.conf import settings
from django.contrib.auth import get_user_model
from django.db import models
import ox

logger = logging.getLogger(__name__)
User = get_user_model()

DEFAULT_LAYER = 'descriptions'

class Page(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    slug = models.SlugField(blank=True, unique=True)
    public = models.BooleanField(default=False)

    title = models.TextField(blank=True)
    body = models.TextField(blank=True)
    data = models.JSONField(default=dict, blank=True)

    def __str__(self):
        return '%s (%s)' % (self.title, self.slug)

    def get_absolute_url(self):
        if self.slug:
            return '/' + settings.URL_PREFIX + '' + self.slug
        else:
            return '/' + settings.URL_PREFIX[:-1]


class Text(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    slug = models.SlugField()
    public = models.BooleanField(default=False)
    position = models.IntegerField(default=0)

    title = models.TextField()
    byline = models.TextField(default="", blank=True)
    body = models.TextField(default="", blank=True)

    data = models.JSONField(default=dict)
    annotations = models.JSONField(default=dict, blank=True, editable=False)

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return '/' + settings.URL_PREFIX + 'assemblies/' + self.slug

    def get_annotations(self):
        api = ox.api.signin(settings.DEFAULT_PANDORA_API)
        if 'item' in self.data:
            return self.get_item_annotations(api)
        elif 'edit' in self.data:
            return self.get_edit_annotations(api)
        else:
            return []

    def get_item_annotations(self, api):
        query = {
            'id': self.data['item'],
            'keys': [
                'layers'
            ]
        }
        layer = self.data.get('layer', None) or DEFAULT_LAYER
        annotations = api.get(**query)['data']['layers'][layer]
        user = self.data.get('user', None)
        if user:
            annotations = list(filter(lambda annot: annot['user'] == user, annotations))
        return self.get_clips(api, annotations)

    def get_edit_annotations(self, api):
        query = {
            'id': self.data['edit'],
            'keys': []
        }
        clips = api.getEdit(**query)['data']['clips']
        annotations = []
        layer = self.data.get('layer', None) or DEFAULT_LAYER
        for clip in clips:
            for annotation in clip['layers'][layer]:
                if 'user' in self.data and annotation['user'] == self.data['user']:
                    continue
                annotation['title'] = clip['title']
                annotations.append(annotation)
        return self.get_clips(api, annotations)

    def get_clips(self, api, annotations):
        item_ids = {annot['id'].split('/')[0] for annot in annotations}
        query = {
            'itemsQuery': {
                'conditions': [{
                    'key': 'id',
                    'operator': '&',
                    'value': item_ids
                }]
            },
            'range': [0, 10000],
            'keys': [
                'id',
                'hue',
                'saturation',
                'lightness'
            ]
        }
        items = api.findClips(**query)['data']['items']
        colors = {}
        for item in items:
            colors[item['id']] = item
        previous = None
        for annot in annotations:
            annot_id = annot['id'].split('/')[0]
            annot_in = '{:.3f}'.format(annot['in'])
            annot_out = '{:.3f}'.format(annot['out'])
            clip_id = f'{annot_id}/{annot_in}-{annot_out}'
            annot['color1'] = colors[clip_id]
            if previous:
                previous['color2'] = annot['color1']
            previous = annot
        if len(annotations) > 0:
            annotations[len(annotations) - 1]['color2'] = annotations[0]['color1']
        return annotations

    def save(self, *args, **kwargs):
        annotations = self.get_annotations()
        self.annotations = annotations
        super().save(*args, **kwargs)

    def json(self):
        data = {}
        data['title'] = self.title
        data['byline'] = self.byline
        data['body'] = self.body
        if isinstance(self.annotations, list) and len(self.annotations) > 0:
            data['annotations'] = self.annotations
        data.update(self.data)
        return json.dumps(data)