openmedialibrary/oml/item/title_api.py

# -*- coding: utf-8 -*-
import hashlib
import json
import unicodedata

import ox
from sqlalchemy.orm import load_only

from oxtornado import actions
from . import models
import utils
import state

import logging
logger = logging.getLogger(__name__)


def editTitle(data):
    '''
        takes {
            id       string
            sorttitle string
        }
    '''
    response = {}
    ids = data['id'].split(':')
    data = {
        'sorttitle': unicodedata.normalize('NFKD', data['sorttitle'])
    }
    for item in models.Item.query.filter(models.Item.id.in_(ids)):
        if item.meta.get('sorttitle') != data['sorttitle']:
            item.edit(data)
    response['title'] = item.meta.get('title', 'Untitled')
    response['sorttitle'] = item.meta['sorttitle']
    return response
actions.register(editTitle)

def findTitles(data):
    '''
        takes {
            query {
                conditions [{}]
                operator   string
            }
            keys  [string]
            sort  [{}]
            range [int, int]
        }
    '''
    response = {}
    conditions = [
        {'key': 'list', 'value': ':', 'operator': '=='}
    ]
    if data.get('query', {}).get('conditions'):
        value = data.get('query', {}).get('conditions')[0].get('value')
        if value:
            conditions.append({
                'key': 'title', 'operator': '=', 'value': value
            })
    q = {
        'qs': models.Item.find({
            'query': {
                'conditions': conditions, 'operator': '&'
            }
        }),
    }
    _keydata = data.copy()
    for key in ('range', 'position', 'positions'):
        if key in _keydata:
            del _keydata[key]
    key = 'title:' + hashlib.sha1(json.dumps(_keydata, sort_keys=True).encode('utf-8')).hexdigest()
    titles = state.cache.get(key)
    if titles is None:
        state.cache.lock(key)
        response['items'] = []
        _titles = {}
        for i in q['qs']:
            title = i.meta.get('title')
            if not title in _titles:
                _titles[title] = {
                    'id': i.id,
                    'title': title,
                    'sorttitle': i.get_sorttitle()
                }
            else:
                _titles[title]['id'] += ':' + i.id
        titles = list(_titles.values())
        if 'sort' in data:
            titles.sort(key=lambda t: ox.sort_string(t[data['sort'][0]['key']]).lower(),
                reverse=data['sort'][0]['operator']=='-')
        state.cache.set(key, titles)

    if 'position' in data:
        pass
    elif 'positions' in data:
        ids = [t['id'] for t in titles]
        response['positions'] = utils.get_positions(ids, data['positions'])
    elif 'keys' in data:
        response['items'] = titles[data['range'][0]:data['range'][1]]
    else:
        response['items'] = len(titles)
    return response
actions.register(findTitles)