250 lines
8.9 KiB
Python
250 lines
8.9 KiB
Python
import flask; from flask import request
|
|
import os
|
|
import time
|
|
import urllib.parse
|
|
|
|
from voussoirkit import stringtools
|
|
|
|
import etiquette
|
|
|
|
from .. import common
|
|
from .. import decorators
|
|
from .. import jsonify
|
|
|
|
site = common.site
|
|
session_manager = common.session_manager
|
|
|
|
# Individual albums ################################################################################
|
|
|
|
@site.route('/album/<album_id>')
|
|
def get_album_html(album_id):
|
|
album = common.P_album(album_id, response_type='html')
|
|
response = common.render_template(
|
|
request,
|
|
'album.html',
|
|
album=album,
|
|
view=request.args.get('view', 'grid'),
|
|
)
|
|
return response
|
|
|
|
@site.route('/album/<album_id>.json')
|
|
def get_album_json(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
album = album.jsonify()
|
|
return jsonify.make_json_response(album)
|
|
|
|
@site.route('/album/<album_id>.zip')
|
|
def get_album_zip(album_id):
|
|
album = common.P_album(album_id, response_type='html')
|
|
|
|
recursive = request.args.get('recursive', True)
|
|
recursive = etiquette.helpers.truthystring(recursive)
|
|
|
|
streamed_zip = etiquette.helpers.zip_album(album, recursive=recursive)
|
|
|
|
if album.title:
|
|
download_as = f'album {album.id} - {album.title}.zip'
|
|
else:
|
|
download_as = f'album {album.id}.zip'
|
|
|
|
download_as = etiquette.helpers.remove_path_badchars(download_as)
|
|
download_as = urllib.parse.quote(download_as)
|
|
outgoing_headers = {
|
|
'Content-Type': 'application/octet-stream',
|
|
'Content-Disposition': f'attachment; filename*=UTF-8\'\'{download_as}',
|
|
}
|
|
|
|
return flask.Response(streamed_zip, headers=outgoing_headers)
|
|
|
|
@site.route('/album/<album_id>/add_child', methods=['POST'])
|
|
@decorators.required_fields(['child_id'], forbid_whitespace=True)
|
|
def post_album_add_child(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
child_ids = stringtools.comma_space_split(request.form['child_id'])
|
|
children = list(common.P_albums(child_ids, response_type='json'))
|
|
print(children)
|
|
album.add_children(children, commit=True)
|
|
response = album.jsonify()
|
|
return jsonify.make_json_response(response)
|
|
|
|
@site.route('/album/<album_id>/remove_child', methods=['POST'])
|
|
@decorators.required_fields(['child_id'], forbid_whitespace=True)
|
|
def post_album_remove_child(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
child_ids = stringtools.comma_space_split(request.form['child_id'])
|
|
children = list(common.P_albums(child_ids, response_type='json'))
|
|
album.remove_children(children, commit=True)
|
|
response = album.jsonify()
|
|
return jsonify.make_json_response(response)
|
|
|
|
@site.route('/album/<album_id>/remove_thumbnail_photo', methods=['POST'])
|
|
def post_album_remove_thumbnail_photo(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
album.set_thumbnail_photo(None)
|
|
common.P.commit(message='album remove thumbnail photo endpoint')
|
|
return jsonify.make_json_response(album.jsonify())
|
|
|
|
@site.route('/album/<album_id>/refresh_directories', methods=['POST'])
|
|
def post_album_refresh_directories(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
for directory in album.get_associated_directories():
|
|
if not directory.is_dir:
|
|
continue
|
|
digest = common.P.digest_directory(directory, new_photo_ratelimit=0.1)
|
|
etiquette.helpers.run_generator(digest)
|
|
common.P.commit(message='refresh album directories endpoint')
|
|
return jsonify.make_json_response({})
|
|
|
|
@site.route('/album/<album_id>/set_thumbnail_photo', methods=['POST'])
|
|
@decorators.required_fields(['photo_id'], forbid_whitespace=True)
|
|
def post_album_set_thumbnail_photo(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
photo = common.P_photo(request.form['photo_id'], response_type='json')
|
|
album.set_thumbnail_photo(photo)
|
|
common.P.commit(message='album set thumbnail photo endpoint')
|
|
return jsonify.make_json_response(album.jsonify())
|
|
|
|
# Album photo operations ###########################################################################
|
|
|
|
@site.route('/album/<album_id>/add_photo', methods=['POST'])
|
|
@decorators.required_fields(['photo_id'], forbid_whitespace=True)
|
|
def post_album_add_photo(album_id):
|
|
'''
|
|
Add a photo or photos to this album.
|
|
'''
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
photo_ids = stringtools.comma_space_split(request.form['photo_id'])
|
|
photos = list(common.P_photos(photo_ids, response_type='json'))
|
|
album.add_photos(photos, commit=True)
|
|
response = album.jsonify()
|
|
return jsonify.make_json_response(response)
|
|
|
|
@site.route('/album/<album_id>/remove_photo', methods=['POST'])
|
|
@decorators.required_fields(['photo_id'], forbid_whitespace=True)
|
|
def post_album_remove_photo(album_id):
|
|
'''
|
|
Remove a photo or photos from this album.
|
|
'''
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
photo_ids = stringtools.comma_space_split(request.form['photo_id'])
|
|
photos = list(common.P_photos(photo_ids, response_type='json'))
|
|
album.remove_photos(photos, commit=True)
|
|
response = album.jsonify()
|
|
return jsonify.make_json_response(response)
|
|
|
|
# Album tag operations #############################################################################
|
|
|
|
@site.route('/album/<album_id>/add_tag', methods=['POST'])
|
|
def post_album_add_tag(album_id):
|
|
'''
|
|
Apply a tag to every photo in the album.
|
|
'''
|
|
response = {}
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
tag = request.form['tagname'].strip()
|
|
try:
|
|
tag = common.P_tag(tag, response_type='json')
|
|
except etiquette.exceptions.NoSuchTag as exc:
|
|
response = exc.jsonify()
|
|
return jsonify.make_json_response(response, status=404)
|
|
recursive = request.form.get('recursive', False)
|
|
recursive = etiquette.helpers.truthystring(recursive)
|
|
album.add_tag_to_all(tag, nested_children=recursive, commit=True)
|
|
response['action'] = 'add_tag'
|
|
response['tagname'] = tag.name
|
|
return jsonify.make_json_response(response)
|
|
|
|
# Album metadata operations ########################################################################
|
|
|
|
@site.route('/album/<album_id>/edit', methods=['POST'])
|
|
def post_album_edit(album_id):
|
|
'''
|
|
Edit the title / description.
|
|
'''
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
title = request.form.get('title', None)
|
|
description = request.form.get('description', None)
|
|
album.edit(title=title, description=description, commit=True)
|
|
response = album.jsonify(minimal=True)
|
|
return jsonify.make_json_response(response)
|
|
|
|
@site.route('/album/<album_id>/show_in_folder', methods=['POST'])
|
|
def post_album_show_in_folder(album_id):
|
|
if not request.is_localhost:
|
|
flask.abort(403)
|
|
|
|
album = common.P_album(album_id, response_type='json')
|
|
directories = album.get_associated_directories()
|
|
if len(directories) != 1:
|
|
flask.abort(400)
|
|
directory = directories.pop()
|
|
|
|
if os.name == 'nt':
|
|
command = f'start explorer.exe "{directory.absolute_path}"'
|
|
os.system(command)
|
|
return jsonify.make_json_response({})
|
|
|
|
flask.abort(501)
|
|
|
|
# Album listings ###################################################################################
|
|
|
|
@site.route('/all_albums.json')
|
|
@decorators.cached_endpoint(max_age=15)
|
|
def get_all_album_names():
|
|
all_albums = {album.display_name: album.id for album in common.P.get_albums()}
|
|
response = {'albums': all_albums}
|
|
return jsonify.make_json_response(response)
|
|
|
|
def get_albums_core():
|
|
albums = list(common.P.get_root_albums())
|
|
albums.sort(key=lambda x: x.display_name.lower())
|
|
return albums
|
|
|
|
@site.route('/albums')
|
|
def get_albums_html():
|
|
albums = get_albums_core()
|
|
response = common.render_template(
|
|
request,
|
|
'album.html',
|
|
albums=albums,
|
|
view=request.args.get('view', 'grid'),
|
|
)
|
|
return response
|
|
|
|
@site.route('/albums.json')
|
|
def get_albums_json():
|
|
albums = get_albums_core()
|
|
albums = [album.jsonify(minimal=True) for album in albums]
|
|
return jsonify.make_json_response(albums)
|
|
|
|
# Album create and delete ##########################################################################
|
|
|
|
@site.route('/albums/create_album', methods=['POST'])
|
|
def post_albums_create():
|
|
title = request.form.get('title', None)
|
|
description = request.form.get('description', None)
|
|
parent_id = request.form.get('parent_id', None)
|
|
if parent_id is not None:
|
|
parent = common.P_album(parent_id, response_type='json')
|
|
|
|
user = session_manager.get(request).user
|
|
|
|
album = common.P.new_album(title=title, description=description, author=user)
|
|
if parent_id is not None:
|
|
parent.add_child(album)
|
|
common.P.commit('create album endpoint')
|
|
|
|
response = album.jsonify(minimal=False)
|
|
return jsonify.make_json_response(response)
|
|
|
|
@site.route('/album/<album_id>/delete', methods=['POST'])
|
|
def post_album_delete(album_id):
|
|
album = common.P_album(album_id, response_type='json')
|
|
album.delete(commit=True)
|
|
return jsonify.make_json_response({})
|