mirror of
https://github.com/TrentSPalmer/flask_photo_scaling_app.git
synced 2024-11-16 22:41:29 -08:00
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import psycopg2
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def delete_photo(photo, app_config):
|
||
|
|
conn = psycopg2.connect(
|
||
|
|
dbname=app_config['DATABASE_NAME'],
|
||
|
|
user=app_config['DATABASE_USER'],
|
||
|
|
password=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(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
|