Ethan Dalool
adb1d0ef39
There was always some semblance that two blank lines has some kind of meaning or structure that's different from single blank lines, but in reality it was mostly arbitrary and I can't stand to look at it any more.
198 lines
7.2 KiB
Python
198 lines
7.2 KiB
Python
import flask; from flask import request
|
|
import urllib.parse
|
|
|
|
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 = etiquette.jsonify.album(album)
|
|
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 = common.P_album(request.form['child_id'], response_type='json')
|
|
album.add_child(child, commit=True)
|
|
response = etiquette.jsonify.album(child)
|
|
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 = common.P_album(request.form['child_id'], response_type='json')
|
|
album.remove_child(child, commit=True)
|
|
response = etiquette.jsonify.album(child)
|
|
return jsonify.make_json_response(response)
|
|
|
|
@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():
|
|
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({})
|
|
|
|
# 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.
|
|
'''
|
|
response = {}
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
photo_ids = etiquette.helpers.comma_space_split(request.form['photo_id'])
|
|
photos = list(common.P_photos(photo_ids, response_type='json'))
|
|
album.add_photos(photos, commit=True)
|
|
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.
|
|
'''
|
|
response = {}
|
|
album = common.P_album(album_id, response_type='json')
|
|
|
|
photo_ids = etiquette.helpers.comma_space_split(request.form['photo_id'])
|
|
photos = list(common.P_photos(photo_ids, response_type='json'))
|
|
album.remove_photos(photos, commit=True)
|
|
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 = etiquette.jsonify.exception(exc)
|
|
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 = etiquette.jsonify.album(album, minimal=True)
|
|
return jsonify.make_json_response(response)
|
|
|
|
# Album listings ###################################################################################
|
|
|
|
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 = [etiquette.jsonify.album(album, 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 = etiquette.jsonify.album(album, 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({})
|