mirror of
https://github.com/TrentSPalmer/flask_photo_scaling_app.git
synced 2025-12-13 22:01:29 -08:00
refactor with blueprints
This commit is contained in:
48
app/photo_routes/proutes.py
Normal file
48
app/photo_routes/proutes.py
Normal file
@@ -0,0 +1,48 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
from flask import Blueprint, flash, render_template, current_app
|
||||
from flask_login import current_user
|
||||
from shutil import disk_usage
|
||||
import psycopg2
|
||||
|
||||
proute = Blueprint(
|
||||
"proute", __name__, template_folder="templates"
|
||||
)
|
||||
|
||||
|
||||
@proute.route("/")
|
||||
@proute.route("/index")
|
||||
def index():
|
||||
if current_user.is_authenticated:
|
||||
photos = get_photo_list(current_user.id)
|
||||
flash(get_disk_stats())
|
||||
return(render_template(
|
||||
'index.html',
|
||||
title="Photos",
|
||||
photos=photos,
|
||||
photo_url=current_app.config['PHOTO_URL']
|
||||
))
|
||||
return render_template('index.html', title="Photos")
|
||||
|
||||
|
||||
def get_photo_list(contributor_id):
|
||||
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(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)
|
||||
))
|
||||
Reference in New Issue
Block a user