tanshu/soter
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Updated picture_raw view to show different sizes.

Browse Source 
Albums shifted to semantic url names.
Uploads are now thumbnailed when first viewed.
This commit is contained in:
tanshu 2015-06-28 18:21:27 +05:30
parent 088310b652
commit cf72709ddf
7 changed files with 91 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
soter/__init__.py
View File

@ -39,7 +39,7 @@ def main(global_config, **settings):
config.add_route('logout', '/logout')
add_route(config, 'picture', '/picture')
config.add_route('picture_raw', '/p/{id}')
config.add_route('picture_raw', '/p/{size}/{id}')
config.add_route('upload', '/upload')
config.add_route('api_upload', '/v1/upload')

8
soter/models/master.py
View File

@ -78,7 +78,7 @@ class Album(Base):
@name.setter
def name(self, name):
self.name = name
self._name = name
self.semantic = semantic_name(name)
def __init__(self, name, description, user_id, is_public, id=None, is_fixture=False):
@ -105,6 +105,12 @@ class Album(Base):
return None
return DBSession.query(cls).filter(cls.name.ilike(name)).first()
@classmethod
def by_semantic(cls, semantic):
if not semantic:
return None
return DBSession.query(cls).filter(cls.semantic.ilike(semantic)).first()
@classmethod
def menu_item(cls):
return uuid.UUID('dad46805-f577-4e5b-8073-9b788e0173fc')

7
soter/static/app/picture/picture.controller.js
View File

@ -33,9 +33,10 @@
var UploadCtrlResolve = {};
var PictureCtrlResolve = {
album: ['$route', 'Picture', function ($route, Picture) {
var id = $route.current.params.id;
return Picture.get({id: id}).$promise;
picture: ['$route', 'Picture', function ($route, Picture) {
var id = $route.current.params.id,
size = $route.current.params.s;
return Picture.get({id: id, s: size}).$promise;
}]
};

4
soter/static/app/picture/view.html
View File

@ -1,5 +1,5 @@
<div class="carousel-inner">
<!--?<div class="carousel-inner">-->
<div class="item">
<img ng-src="{{picture.imageUrl}}">
</div>
</div>
<!--?</div>-->

1
soter/static/index.html
View File

@ -51,6 +51,7 @@
<script src="/app/album/album.service.js"></script>
<script src="/app/picture/picture.controller.js"></script>
<script src="/app/picture/picture.service.js"></script>
<script src="/script/angular_directive.js"></script>
<script src="/script/angular_filter.js"></script>

30
soter/views/album.py
View File

@ -2,8 +2,11 @@ import uuid
import re
import pkg_resources
from pyramid.response import FileResponse, Response
from pyramid.security import authenticated_userid
from pyramid.view import view_config
import transaction
@ -35,7 +38,13 @@ def save(request):
@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']))
p = re.compile('^[A-Fa-f0-9]{8}-[A-Fa-f0-9]{4}-[A-Fa-f0-9]{4}-[A-Fa-f0-9]{4}-[A-Fa-f0-9]{12}$')
id = request.matchdict['id']
if p.match(id):
item = Album.by_id(uuid.UUID(id))
else:
item = Album.by_semantic(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']
@ -61,7 +70,13 @@ def delete(request):
@view_config(request_method='GET', route_name='api_album_id', renderer='json', permission='Albums')
def show_id(request):
return album_info(request.matchdict['id'])
p = re.compile('^[A-Fa-f0-9]{8}-[A-Fa-f0-9]{4}-[A-Fa-f0-9]{4}-[A-Fa-f0-9]{4}-[A-Fa-f0-9]{12}$')
id = request.matchdict['id']
if p.match(id):
item = Album.by_id(uuid.UUID(id))
else:
item = Album.by_semantic(id)
return album_info(item)
@view_config(request_method='GET', route_name='api_album', renderer='json', permission='Albums')
@ -77,8 +92,8 @@ def show_list(request):
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)})
'viewUrl': request.route_url('album_id_view', id=item.semantic, ),
'editUrl': request.route_url('album_id', id=item.semantic)})
return albums
@ -90,11 +105,12 @@ def show_pics(request):
if p.match(id):
album = Album.by_id(uuid.UUID(id))
else:
album = Album.by_name(id)
album = Album.by_semantic(id)
info = album_info(album)
for item in album.pictures:
info['pictures'].append({'name': item.name, 'url': request.route_url('picture_id', id=item.id),
'imageUrl': request.route_url('picture_raw', id=item.id)})
info['pictures'].append({'name': item.name,
'url': request.route_url('picture_id', id=item.id, _query={'s': '1600x1200'}),
'imageUrl': request.route_url('picture_raw', size='242x200', id=item.id)})
return info

71
soter/views/picture.py
View File

@ -4,10 +4,12 @@ import shutil
import uuid
from PIL import Image, ExifTags
from PIL.Image import LANCZOS
import pkg_resources
from pyramid.response import FileResponse
from pyramid.response import FileResponse, Response
from pyramid.security import authenticated_userid
from pyramid.view import view_config
from sqlalchemy.exc import IntegrityError
import transaction
import numpy
@ -16,6 +18,7 @@ from soter.models.master import Picture, Album
@view_config(request_method='GET', route_name='upload', permission='Albums')
@view_config(request_method='GET', route_name='picture_id', permission='Albums')
def html(request):
package, resource = 'soter:static/index.html'.split(':', 1)
file = pkg_resources.resource_filename(package, resource)
@ -26,23 +29,34 @@ def html(request):
def upload(request):
input_file = request.POST['file'].file
file_id = uuid.uuid4()
file_path = pkg_resources.resource_filename('soter', 'upload/' + str(file_id) + '.jpg')
file_path = pkg_resources.resource_filename('soter', 'upload/o/' + str(file_id) + '.jpg')
album = request.POST['album']
temp_file_path = file_path + '~'
input_file.seek(0)
with open(temp_file_path, 'wb') as output_file:
shutil.copyfileobj(input_file, output_file)
os.rename(temp_file_path, file_path)
# Process picture here to get the hash and the exif tags to be added to the database
pic = process_picture(file_path, str(file_id), uuid.UUID(authenticated_userid(request)), file_id, uuid.UUID(album))
pic = process_picture(temp_file_path, str(file_id), uuid.UUID(authenticated_userid(request)), file_id,
uuid.UUID(album))
DBSession.add(pic)
transaction.commit()
try:
transaction.commit()
except IntegrityError as ex:
transaction.abort()
os.remove(temp_file_path)
response = Response("Duplicate file")
response.status_int = 500
return response
else:
os.rename(temp_file_path, file_path)
return {'location': file_path}
def process_picture(path, name, user_id, file_id, album_id):
image = get_image(path)
hash = hashlib.sha512(numpy.array(image)).digest()
img = numpy.array(image)
img = img[:, :, ::-1].copy()
hash = hashlib.sha512(numpy.array(img)).digest()
picture = Picture(name, hash, user_id, True, file_id)
picture.albums.append(Album.by_id(album_id))
return picture
@ -51,37 +65,58 @@ def process_picture(path, name, user_id, file_id, album_id):
def get_image(path):
image = Image.open(path)
orientation = [tag for tag in ExifTags.TAGS.keys() if ExifTags.TAGS[tag] == 'Orientation'][0]
exif = dict(image._getexif().items())
exif = getattr(image, "_getexif", None)
if exif is None:
return image
exif = image._getexif()
if exif is None:
return image
exif = dict(exif.items())
if exif[orientation] == 3:
image = image.rotate(180, expand=True)
elif exif[orientation] == 6:
image = image.rotate(270, expand=True)
elif exif[orientation] == 8:
image = image.rotate(90, expand=True)
img = numpy.array(image)
# CONVERT RGB TO BGR
return img[:, :, ::-1].copy()
return image
@view_config(request_method='GET', route_name='picture_raw', permission='Albums')
def show_raw(request):
package, resource = ('soter:upload/' + request.matchdict['id'] + '.jpg').split(':', 1)
size = request.matchdict['size']
id = request.matchdict['id']
package, resource = ('soter:upload/' + size + '/' + id + '.jpg').split(':', 1)
file = pkg_resources.resource_filename(package, resource)
if not os.path.isfile(file):
create_thumbnail(id, size)
return FileResponse(file, request=request)
@view_config(request_method='GET', route_name='api_album_id', renderer='json', request_param='pictures',
permission='Albums')
def create_thumbnail(id, size):
file_path = pkg_resources.resource_filename('soter', 'upload/' + size + '/' + id + '.jpg')
size = tuple([int(i) for i in size.split('x')])
file = pkg_resources.resource_filename('soter', 'upload/o/' + id + '.jpg')
image = get_image(file)
image.thumbnail(size, resample=LANCZOS)
image.save(file_path)
@view_config(request_method='GET', route_name='api_picture_id', renderer='json', request_param='s', permission='Albums')
def show_id(request):
return picture_info(request.matchdict['id'], request)
return picture_info(request.matchdict['id'], request.GET['s'], request)
def picture_info(id, request):
@view_config(request_method='GET', route_name='api_picture_id', renderer='json', permission='Albums')
def show_id_org(request):
return picture_info(request.matchdict['id'], 'o', request)
def picture_info(id, size, request):
if not isinstance(id, Picture):
picture = Picture.by_id(id)
else:
picture = id
album = {'id': picture.id, 'name': picture.name, 'imageUrl': request.route_url('picture_raw', id=picture.id),
album = {'id': picture.id, 'name': picture.name,
'imageUrl': request.route_url('picture_raw', id=picture.id, size=size),
'user': {'id': picture.user_id, 'name': picture.user.name}}
return album