Open Media Library

oml/user/api.py

@@ -0,0 +1,216 @@
+# -*- coding: utf-8 -*-
+# vi:si:et:sw=4:sts=4:ts=4
+
+import os
+from copy import deepcopy
+
+from flask import json
+from oxflask.api import actions
+from oxflask.shortcuts import returns_json
+
+import models
+from item.models import Item
+
+from utils import get_position_by_id
+
+import settings
+import state
+from changelog import Changelog
+
+@returns_json
+def init(request):
+    '''
+        this is an init request to test stuff
+    '''
+    #print 'init', request
+    response = {}
+    if os.path.exists(settings.oml_config_path):
+        with open(settings.oml_config_path) as fd:
+            config = json.load(fd)
+    else:
+        config = {}
+    response['config'] = config
+    response['user'] = deepcopy(config['user'])
+    if settings.preferences:
+        response['user']['preferences'] = settings.preferences
+    response['user']['id'] = settings.USER_ID
+    response['user']['online'] = state.online
+    if settings.ui:
+        response['user']['ui'] = settings.ui
+    return response
+actions.register(init)
+
+def update_dict(root, data):
+    for key in data:
+        keys = map(lambda p: p.replace('\0', '\\.'), key.replace('\\.', '\0').split('.'))
+        value = data[key]
+        p = root
+        while len(keys)>1:
+            key = keys.pop(0)
+            if isinstance(p, list):
+                p = p[get_position_by_id(p, key)]
+            else:
+                if key not in p:
+                    p[key] = {}
+                p = p[key]
+        if value == None and keys[0] in p:
+            del p[keys[0]]
+        else:
+            p[keys[0]] = value
+
+@returns_json
+def setPreferences(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    update_dict(settings.preferences, data)
+    return settings.preferences
+actions.register(setPreferences)
+
+@returns_json
+def setUI(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    update_dict(settings.ui, data)
+    return settings.ui
+actions.register(setUI)
+
+@returns_json
+def getUsers(request):
+    users = []
+    for u in models.User.query.filter(models.User.id!=settings.USER_ID).all():
+        users.append(u.json())
+    return {
+        "users": users
+    }
+actions.register(getUsers)
+
+@returns_json
+def getLists(request):
+    lists = {}
+    for u in models.User.query.filter((models.User.peered==True)|(models.User.id==settings.USER_ID)):
+        lists[u.id] = u.lists_json()
+    return {
+        'lists': lists
+    }
+actions.register(getLists)
+
+@returns_json
+def addList(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    user_id = settings.USER_ID
+    l = models.List.get(user_id, data['name'])
+    if not l:
+        l = models.List.create(user_id, data['name'], data.get('query'))
+        if 'items' in data:
+            l.add_items(data['items'])
+        return l.json()
+    return {}
+actions.register(addList, cache=False)
+
+@returns_json
+def removeList(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    l = models.List.get(data['id'])
+    if l:
+        l.remove()
+    return {}
+actions.register(removeList, cache=False)
+
+@returns_json
+def editList(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    l = models.List.get_or_create(data['id'])
+    name = l.name
+    if 'name' in data:
+        l.name = data['name']
+        l.type = 'static'
+    if 'query' in data:
+        l._query = data['query']
+        l.type = 'smart'
+    if l.type == 'static' and name != l.name:
+        Changelog.record(state.user(), 'editlist', name, {'name': l.name})
+    l.save()
+    return {}
+actions.register(editList, cache=False)
+
+@returns_json
+def addListItem(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    l = models.List.get_or_create(data['id'])
+    i = Item.get(data['item'])
+    if l and i:
+        l.items.append(i)
+        models.db.session.add(l)
+        i.update()
+    return {}
+actions.register(addListItem, cache=False)
+
+@returns_json
+def removeListItem(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    l = models.List.get(data['id'])
+    i = Item.get(data['item'])
+    if l and i:
+        l.items.remove(i)
+        models.db.session.add(l)
+        i.update()
+    return {}
+actions.register(removeListItem, cache=False)
+
+@returns_json
+def sortLists(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    n = 0
+    print 'sortLists', data
+    for id in data['ids']:
+        l = models.List.get(id)
+        l.position = n
+        n += 1
+        models.db.session.add(l)
+    models.db.session.commit()
+    return {}
+actions.register(sortLists, cache=False)
+
+@returns_json
+def editUser(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    if 'nickname' in data:
+        p = models.User.get_or_create(data['id'])
+        p.set_nickname(data['nickname'])
+        p.save()
+    return {}
+actions.register(editUser, cache=False)
+
+@returns_json
+def requestPeering(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    p = models.User.get_or_create(data['id'])
+    state.nodes.queue('add', p.id)
+    state.nodes.queue(p.id, 'requestPeering', data.get('message', ''))
+    return {}
+actions.register(requestPeering, cache=False)
+
+@returns_json
+def acceptPeering(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    p = models.User.get_or_create(data['id'])
+    state.nodes.queue('add', p.id)
+    state.nodes.queue(p.id, 'acceptPeering', data.get('message', ''))
+    return {}
+actions.register(acceptPeering, cache=False)
+
+@returns_json
+def rejectPeering(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    p = models.User.get_or_create(data['id'])
+    state.nodes.queue('add', p.id)
+    state.nodes.queue(p.id, 'rejectPeering', data.get('message', ''))
+    return {}
+actions.register(rejectPeering, cache=False)
+
+@returns_json
+def removePeering(request):
+    data = json.loads(request.form['data']) if 'data' in request.form else {}
+    p = models.User.get_or_create(data['id'])
+    state.nodes.queue('add', p.id)
+    state.nodes.queue(p.id, 'removePeering', data.get('message', ''))
+    return {}
+actions.register(removePeering, cache=False)