openmedialibrary/oml/user/api.py

360 lines
7.7 KiB
Python
Raw Normal View History

2014-05-04 17:26:43 +00:00
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
2014-05-17 00:14:15 +00:00
from __future__ import division
2014-05-04 17:26:43 +00:00
import os
from copy import deepcopy
2014-05-17 00:14:15 +00:00
import json
2014-05-04 17:26:43 +00:00
2014-05-19 20:14:24 +00:00
from oxtornado import actions
2014-05-04 17:26:43 +00:00
import models
2014-05-18 23:24:04 +00:00
from utils import update_dict
2014-05-04 17:26:43 +00:00
import settings
import state
from changelog import Changelog
2014-05-17 14:26:59 +00:00
import logging
logger = logging.getLogger('oml.user.api')
2014-05-19 20:14:24 +00:00
def init(data):
2014-05-04 17:26:43 +00:00
'''
2014-05-18 23:24:04 +00:00
takes {
}
returns {
config
user
preferences
ui
}
2014-05-04 17:26:43 +00:00
'''
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)
2014-05-19 20:14:24 +00:00
def setPreferences(data):
2014-05-18 23:24:04 +00:00
'''
takes {
key: value,
'sub.key': value
}
'''
2014-05-04 17:26:43 +00:00
update_dict(settings.preferences, data)
return settings.preferences
actions.register(setPreferences)
2014-05-19 20:14:24 +00:00
def setUI(data):
2014-05-18 23:24:04 +00:00
'''
takes {
key: value,
'sub.key': value
}
'''
2014-05-04 17:26:43 +00:00
update_dict(settings.ui, data)
return settings.ui
actions.register(setUI)
2014-05-19 20:14:24 +00:00
def getUsers(data):
2014-05-18 23:24:04 +00:00
'''
returns {
users: []
}
'''
2014-05-04 17:26:43 +00:00
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)
2014-05-19 20:14:24 +00:00
def getLists(data):
2014-05-18 23:24:04 +00:00
'''
returns {
lists: []
}
'''
from item.models import Item
2014-05-12 12:57:47 +00:00
lists = []
2014-05-18 23:24:04 +00:00
lists.append({
'id': '',
'items': Item.query.count(),
'name': 'Libraries',
'type': 'libraries',
'user': '',
})
2014-05-04 17:26:43 +00:00
for u in models.User.query.filter((models.User.peered==True)|(models.User.id==settings.USER_ID)):
2014-05-12 12:57:47 +00:00
lists += u.lists_json()
2014-05-04 17:26:43 +00:00
return {
'lists': lists
}
actions.register(getLists)
2014-05-18 23:24:04 +00:00
def validate_query(query):
for condition in query['conditions']:
if not list(sorted(condition.keys())) in (
['conditions', 'operator'],
['key', 'operator', 'value']
):
raise Exception('invalid query condition', condition)
2014-05-19 20:14:24 +00:00
def addList(data):
2014-05-18 23:24:04 +00:00
'''
takes {
name
items
query
}
'''
logger.debug('addList %s', data)
2014-05-04 17:26:43 +00:00
user_id = settings.USER_ID
2014-05-18 23:24:04 +00:00
if 'query' in data:
validate_query(data['query'])
if data['name']:
2014-05-04 17:26:43 +00:00
l = models.List.create(user_id, data['name'], data.get('query'))
if 'items' in data:
l.add_items(data['items'])
return l.json()
2014-05-18 23:24:04 +00:00
else:
raise Exception('name not set')
2014-05-04 17:26:43 +00:00
return {}
actions.register(addList, cache=False)
2014-05-19 20:14:24 +00:00
def editList(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
name
query
}
'''
2014-05-17 14:26:59 +00:00
logger.debug('editList %s', data)
2014-05-04 17:26:43 +00:00
l = models.List.get_or_create(data['id'])
name = l.name
if 'name' in data:
l.name = data['name']
if 'query' in data:
2014-05-18 23:24:04 +00:00
validate_query(data['query'])
2014-05-04 17:26:43 +00:00
l._query = data['query']
if l.type == 'static' and name != l.name:
Changelog.record(state.user(), 'editlist', name, {'name': l.name})
l.save()
2014-05-17 14:26:59 +00:00
return l.json()
2014-05-04 17:26:43 +00:00
actions.register(editList, cache=False)
2014-05-19 20:14:24 +00:00
def removeList(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
}
'''
l = models.List.get(data['id'])
if l:
l.remove()
return {}
actions.register(removeList, cache=False)
2014-05-19 20:14:24 +00:00
def addListItems(data):
2014-05-18 23:24:04 +00:00
'''
takes {
list
items
}
'''
2014-05-17 23:14:29 +00:00
if data['list'] == ':':
from item.models import Item
user = state.user()
for item_id in data['items']:
i = Item.get(item_id)
if user not in i.users:
i.queue_download()
elif data['list']:
l = models.List.get_or_create(data['list'])
if l:
l.add_items(data['items'])
return l.json()
2014-05-04 17:26:43 +00:00
return {}
2014-05-12 12:57:47 +00:00
actions.register(addListItems, cache=False)
2014-05-04 17:26:43 +00:00
2014-05-19 20:14:24 +00:00
def removeListItems(data):
2014-05-18 23:24:04 +00:00
'''
takes {
list
items
}
'''
2014-05-12 12:57:47 +00:00
l = models.List.get(data['list'])
if l:
l.remove_items(data['items'])
return l.json()
2014-05-04 17:26:43 +00:00
return {}
2014-05-12 12:57:47 +00:00
actions.register(removeListItems, cache=False)
2014-05-04 17:26:43 +00:00
2014-05-19 20:14:24 +00:00
def sortLists(data):
2014-05-18 23:24:04 +00:00
'''
takes {
ids
}
'''
2014-05-04 17:26:43 +00:00
n = 0
2014-05-17 14:26:59 +00:00
logger.debug('sortLists %s', data)
2014-05-04 17:26:43 +00:00
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)
2014-05-19 20:14:24 +00:00
def editUser(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
nickname
}
'''
2014-05-04 17:26:43 +00:00
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)
2014-05-19 20:14:24 +00:00
def dataPeering(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
message
}
'''
2014-05-17 00:14:15 +00:00
if len(data.get('id', '')) != 43:
2014-05-17 14:26:59 +00:00
logger.debug('invalid user id')
2014-05-17 00:14:15 +00:00
return {}
2014-05-18 03:01:24 +00:00
u = models.User.get_or_create(data['id'])
u.pending = 'sent'
u.queued = True
u.info['message'] = data.get('message', '')
u.save()
state.nodes.queue('add', u.id)
2014-05-19 20:14:24 +00:00
state.nodes.queue(u.id, 'peering', 'dataPeering')
2014-05-04 17:26:43 +00:00
return {}
2014-05-19 20:14:24 +00:00
actions.register(dataPeering, cache=False)
2014-05-04 17:26:43 +00:00
2014-05-19 20:14:24 +00:00
def acceptPeering(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
message
}
'''
2014-05-17 00:14:15 +00:00
if len(data.get('id', '')) != 43:
2014-05-17 14:26:59 +00:00
logger.debug('invalid user id')
2014-05-17 00:14:15 +00:00
return {}
2014-05-17 14:26:59 +00:00
logger.debug('acceptPeering... %s', data)
2014-05-18 03:01:24 +00:00
u = models.User.get_or_create(data['id'])
u.info['message'] = data.get('message', '')
u.update_peering(True)
state.nodes.queue('add', u.id)
state.nodes.queue(u.id, 'peering', 'acceptPeering')
2014-05-04 17:26:43 +00:00
return {}
actions.register(acceptPeering, cache=False)
2014-05-19 20:14:24 +00:00
def rejectPeering(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
message
}
'''
2014-05-17 00:14:15 +00:00
if len(data.get('id', '')) != 43:
2014-05-17 14:26:59 +00:00
logger.debug('invalid user id')
2014-05-17 00:14:15 +00:00
return {}
2014-05-18 03:01:24 +00:00
u = models.User.get_or_create(data['id'])
u.info['message'] = data.get('message', '')
u.update_peering(False)
state.nodes.queue('add', u.id)
state.nodes.queue(u.id, 'peering', 'rejectPeering')
2014-05-04 17:26:43 +00:00
return {}
actions.register(rejectPeering, cache=False)
2014-05-19 20:14:24 +00:00
def removePeering(data):
2014-05-18 23:24:04 +00:00
'''
takes {
id
message
}
'''
2014-05-17 00:14:15 +00:00
if len(data.get('id', '')) != 43:
2014-05-17 14:26:59 +00:00
logger.debug('invalid user id')
2014-05-17 00:14:15 +00:00
return {}
2014-05-13 10:36:02 +00:00
u = models.User.get_or_create(data['id'])
2014-05-18 03:01:24 +00:00
u.info['message'] = data.get('message', '')
u.update_peering(False)
2014-05-13 10:36:02 +00:00
state.nodes.queue('add', u.id)
2014-05-18 03:01:24 +00:00
state.nodes.queue(u.id, 'peering', 'removePeering')
2014-05-04 17:26:43 +00:00
return {}
actions.register(removePeering, cache=False)
2014-05-12 23:43:27 +00:00
2014-05-19 20:14:24 +00:00
def cancelPeering(data):
2014-05-18 23:24:04 +00:00
'''
takes {
}
'''
2014-05-17 00:14:15 +00:00
if len(data.get('id', '')) != 43:
2014-05-17 14:26:59 +00:00
logger.debug('invalid user id')
2014-05-17 00:14:15 +00:00
return {}
2014-05-18 03:01:24 +00:00
u = models.User.get_or_create(data['id'])
u.info['message'] = data.get('message', '')
u.update_peering(False)
state.nodes.queue('add', u.id)
state.nodes.queue(u.id, 'peering', 'cancelPeering')
2014-05-12 23:43:27 +00:00
return {}
actions.register(cancelPeering, cache=False)
2014-05-17 00:14:15 +00:00
2014-05-19 20:14:24 +00:00
def getActivity(data):
2014-05-18 23:24:04 +00:00
'''
return {
activity
progress
}
'''
2014-05-17 00:14:15 +00:00
return state.activity
actions.register(getActivity, cache=False)