soter/soter/views/album.py

104 lines
4.0 KiB
Python

import uuid
import pkg_resources
from pyramid.response import FileResponse, Response
from pyramid.security import authenticated_userid
from pyramid.view import view_config
import transaction
from soter.models import DBSession
from soter.models.master import Album
from soter.models.validation_exception import TryCatchFunction, ValidationError
@view_config(route_name='album_list', permission='Albums')
@view_config(request_method='GET', route_name='album_id', permission='Albums')
@view_config(request_method='GET', route_name='album_id_view', permission='Albums')
@view_config(request_method='GET', route_name='album', permission='Albums')
def html(request):
package, resource = 'soter:static/index.html'.split(':', 1)
file = pkg_resources.resource_filename(package, resource)
return FileResponse(file, request=request)
@view_config(request_method='POST', route_name='api_album', renderer='json', permission='Albums')
@TryCatchFunction
def save(request):
item = Album(request.json_body['name'], request.json_body['description'],
user_id=uuid.UUID(authenticated_userid(request)), is_public=request.json_body['isPublic'])
DBSession.add(item)
transaction.commit()
return album_info(item.id)
@view_config(request_method='POST', route_name='api_album_id', renderer='json', permission='Albums')
@TryCatchFunction
def update(request):
item = Album.by_id(uuid.UUID(request.matchdict['id']))
if item.is_fixture:
raise ValidationError("{0} is a fixture and cannot be edited or deleted.".format(item.name))
item.name = request.json_body['name']
item.description = request.json_body['description']
item.is_public = request.json_body['isPublic']
item.user_id = uuid.UUID(authenticated_userid(request))
transaction.commit()
return album_info(item.id)
@view_config(request_method='DELETE', route_name='api_album_id', renderer='json', permission='Albums')
def delete(request):
item = request.matchdict['id']
item = None if item is None else Album.by_id(item)
if item.is_fixture:
raise ValidationError("{0} is a fixture and cannot be edited or deleted.".format(item.name))
else:
response = Response("Cost Center deletion not implemented")
response.status_int = 500
return response
@view_config(request_method='GET', route_name='api_album_id', renderer='json', permission='Albums')
def show_id(request):
return album_info(request.matchdict['id'])
@view_config(request_method='GET', route_name='api_album', renderer='json', permission='Albums')
def show_blank(request):
return album_info(None)
@view_config(request_method='GET', route_name='api_album', renderer='json', request_param='list',
permission='Authenticated')
def show_list(request):
list = Album.list()
albums = []
for item in list:
albums.append({'name': item.name, 'description': item.description, 'isPublic': item.is_public,
'user': item.user.name, 'isFixture': item.is_fixture, 'pictures': [],
'viewUrl': request.route_url('album_id_view', id=item.id, ),
'editUrl': request.route_url('album_id', id=item.id)})
return albums
@view_config(request_method='GET', route_name='api_album_id', renderer='json', request_param='pictures',
permission='Albums')
def show_pics(request):
album = Album.by_id(uuid.UUID(request.matchdict['id']))
info = album_info(uuid.UUID(request.matchdict['id']))
for item in album.pictures:
info['pictures'].append({'name': item.name, 'url': request.route_url('api_picture_id', id=item.id)})
return info
def album_info(id):
if id is None:
album = {'name': "", 'description': "", 'isPublic': False, 'pictures': []}
else:
album = Album.by_id(id)
album = {'id': album.id, 'name': album.name, 'description': album.description, 'isPublic': album.is_public,
'pictures': [], 'user': {'id': album.user_id, 'name': album.user.name},
'isFixture': album.is_fixture}
return album