69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from flask import Blueprint, request, redirect, url_for, render_template, current_app, send_file
|
|
from flask_login import current_user
|
|
from app.models import Photo
|
|
from app.forms import ConfirmPhotoDelete
|
|
import psycopg2
|
|
import os
|
|
|
|
d_d = Blueprint(
|
|
"d_d", __name__, template_folder="templates"
|
|
)
|
|
|
|
|
|
@d_d.route('/download')
|
|
def download():
|
|
if current_user.is_authenticated:
|
|
f = request.args['file']
|
|
try:
|
|
return send_file('/var/lib/photo_app/photos/{}'.format(f), attachment_filename=f)
|
|
except Exception as e:
|
|
return str(e)
|
|
|
|
|
|
@d_d.route('/delete', methods=['GET', 'POST'])
|
|
def delete():
|
|
photo = Photo.query.get(request.args['photo_id'])
|
|
if photo is None:
|
|
return(redirect(url_for('proute.index')))
|
|
if not current_user.is_authenticated or photo.contributor_id != current_user.id:
|
|
return(redirect(url_for('proute.index')))
|
|
form = ConfirmPhotoDelete()
|
|
if request.method == 'POST' and form.validate_on_submit():
|
|
return(redirect(url_for('p_route.photo', photo_id=delete_photo(photo))))
|
|
return(render_template(
|
|
'delete_photo.html',
|
|
title="Delete Photo?",
|
|
photo=photo,
|
|
photo_url=current_app.config['PHOTO_URL'],
|
|
form=form
|
|
))
|
|
|
|
|
|
def delete_photo(photo):
|
|
conn = psycopg2.connect(
|
|
dbname=current_app.config['DATABASE_NAME'],
|
|
user=current_app.config['DATABASE_USER'],
|
|
host=current_app.config['DATABASE_HOST'],
|
|
password=current_app.config['DATABASE_PASSWORD']
|
|
)
|
|
cur = conn.cursor()
|
|
cur.execute("SELECT count(id) FROM photo WHERE contributor_id=%s AND id>%s", (photo.contributor_id, photo.id))
|
|
if cur.fetchone()[0] == 0:
|
|
cur.execute("SELECT id FROM photo WHERE contributor_id=%s ORDER BY id", (photo.contributor_id, ))
|
|
else:
|
|
cur.execute("SELECT id FROM photo WHERE contributor_id=%s AND id>%s ORDER BY id", (photo.contributor_id, photo.id))
|
|
next_photo_id = cur.fetchone()[0]
|
|
os.chdir(current_app.config['PHOTO_SAVE_PATH'])
|
|
if os.path.exists('raw_' + photo.photo_name):
|
|
os.remove('raw_' + photo.photo_name)
|
|
if os.path.exists('1280_' + photo.photo_name):
|
|
os.remove('1280_' + photo.photo_name)
|
|
if os.path.exists('480_' + photo.photo_name):
|
|
os.remove('480_' + photo.photo_name)
|
|
cur.execute("DELETE FROM photo WHERE id=%s", (photo.id, ))
|
|
conn.commit()
|
|
conn.close()
|
|
return next_photo_id
|