79 lines
3.2 KiB
Python
79 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import psycopg2
|
|
import psycopg2.extras
|
|
from shutil import disk_usage
|
|
|
|
|
|
def find_next_previous(photo, app_config):
|
|
conn = psycopg2.connect(
|
|
dbname=app_config['DATABASE_NAME'],
|
|
user=app_config['DATABASE_USER'],
|
|
host=app_config['DATABASE_HOST'],
|
|
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))
|
|
count = cur.fetchone()[0]
|
|
if count == 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))
|
|
photo.next_photo_id = cur.fetchone()[0]
|
|
cur.execute("SELECT count(id) FROM photo WHERE contributor_id=%s AND id < %s", (photo.contributor_id, photo.id))
|
|
count = cur.fetchone()[0]
|
|
if count == 0:
|
|
cur.execute("SELECT id FROM photo WHERE contributor_id=%s ORDER BY id DESC", (photo.contributor_id, ))
|
|
else:
|
|
cur.execute("SELECT id FROM photo WHERE contributor_id=%s AND id < %s ORDER BY id DESC", (photo.contributor_id, photo.id))
|
|
photo.previous_photo_id = cur.fetchone()[0]
|
|
conn.close()
|
|
|
|
|
|
def calc_additional_data(photo):
|
|
photo.UploadDate = photo.timestamp.strftime("%Y-%m-%d %H:%M:%S")
|
|
if photo.photo_raw_size >= 1048576:
|
|
photo.SizeOnDisc = str(round(photo.photo_raw_size / 1048576, 1)) + 'M'
|
|
else:
|
|
photo.SizeOnDisc = str(round(photo.photo_raw_size / 1024, 1)) + 'K'
|
|
if photo.photo_1280_size >= 1048576:
|
|
photo.SizeOnDisc1280 = str(round(photo.photo_1280_size / 1048576, 1)) + 'M'
|
|
else:
|
|
photo.SizeOnDisc1280 = str(round(photo.photo_1280_size / 1024, 1)) + 'K'
|
|
if photo.photo_480_size >= 1048576:
|
|
photo.SizeOnDisc480 = str(round(photo.photo_480_size / 1048576, 1)) + 'M'
|
|
else:
|
|
photo.SizeOnDisc480 = str(round(photo.photo_480_size / 1024, 1)) + 'K'
|
|
if photo.GPSAltitude is not None:
|
|
photo.GPSAltitudeFeet = round(photo.GPSAltitude * 3.28084, 1)
|
|
else:
|
|
photo.GPSAltitudeFeet = None
|
|
if photo.GPSLatitude is not None and photo.GPSLongitude is not None:
|
|
photo.LatLong = "{},{}".format(photo.GPSLatitude, photo.GPSLongitude)
|
|
photo.MapUrl = "https://www.google.com/maps/search/?api=1&query={}".format(photo.LatLong)
|
|
else:
|
|
photo.LatLong, photo.MapUrl = None, None
|
|
|
|
|
|
def get_photo_list(contributor_id, app_config):
|
|
conn = psycopg2.connect(
|
|
dbname=app_config['DATABASE_NAME'],
|
|
user=app_config['DATABASE_USER'],
|
|
host=app_config['DATABASE_HOST'],
|
|
password=app_config['DATABASE_PASSWORD']
|
|
)
|
|
cur = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
|
|
cur.execute("SELECT photo_name,id FROM photo WHERE contributor_id=%s ORDER BY timestamp,\"DateTimeOriginal\" DESC", (contributor_id, ))
|
|
photos = cur.fetchall()
|
|
conn.close()
|
|
return photos
|
|
|
|
|
|
def get_disk_stats():
|
|
disk_stats = disk_usage('/')
|
|
return("Used {}GB of {}GB, {}GB free".format(
|
|
round(disk_stats.used / 1073741824, 1),
|
|
round(disk_stats.total / 1073741824, 1),
|
|
round(disk_stats.free / 1073741824, 1)
|
|
))
|