Files
elo-rezept-rechner/src/Run.py
2024-01-30 16:27:16 +01:00

139 lines
3.9 KiB
Python

from flask import Flask, request, render_template, jsonify, redirect, url_for, g
import sqlite3
from flask_oidc import OpenIDConnect
from werkzeug.middleware.proxy_fix import ProxyFix
import os
import json
import psycopg2
app = Flask(__name__)
app.config.update({
'SECRET_KEY': os.environ['SECRET'],
'DEBUG': False,
'OIDC_CLIENT_SECRETS': json.loads(os.environ['CLIENT_SECRETS']),
'OIDC_ID_TOKEN_COOKIE_SECURE': False,
'OIDC_USER_INFO_ENABLED': True,
'OIDC_OPENID_REALM': 'hottis',
'OIDC_SCOPES': ['openid', 'email', 'profile']
})
oidc = OpenIDConnect(app)
def calculate_nutrition(food, weight):
try:
conn = psycopg2.connect()
with conn.cursor() as cursor:
# Abfrage der Nährwertdaten aus der Datenbank
cursor.execute('SELECT kcal, EW, Fett, KH, BST, CA FROM nutrition_table WHERE name = %s', (food,))
result = cursor.fetchone()
if result:
# Runden und Berechnen der Nährwerte basierend auf dem Gewicht
kcal, ew, fett, kh, bst, ca = result
nutrition_values = [
round(kcal * weight / 100), # kcal gerundet auf ganze Zahl
round(ew * weight / 100, 1), # EW gerundet auf eine Dezimalstelle
round(fett * weight / 100, 1), # Fett gerundet auf eine Dezimalstelle
round(kh * weight / 100, 1), # KH gerundet auf eine Dezimalstelle
round(bst * weight / 100, 1), # BST gerundet auf eine Dezimalstelle
round(ca * weight / 100) # CA gerundet auf ganze Zahl
]
return nutrition_values
else:
return None
finally:
if conn:
conn.close()
# Index-Route
@app.route('/')
@oidc.require_login
def index():
return render_template('index.html')
# ...
@app.route('/get_products')
@oidc.require_login
def get_products():
try:
conn = psycopg2.connect()
with conn.cursor() as cursor:
cursor.execute('SELECT name FROM nutrition_table')
products = cursor.fetchall()
print("ter")
return {'products': [product[0] for product in products]}
finally:
if conn:
conn.close()
# Route zum Hinzufügen und Berechnen von Lebensmitteln
@app.route('/add_lm', methods=['GET'])
@oidc.require_login
def add_lm():
food = request.args.get('food')
weight = float(request.args.get('weight'))
nutrition = calculate_nutrition(food, weight)
if nutrition:
# Extrahieren der einzelnen Nährwerte
kcal, ew, fett, kh, bst, ca = nutrition
return jsonify({
"food": food,
"kcal": kcal,
"ew": ew,
"fett": fett,
"kh": kh,
"bst": bst,
"ca": ca
})
else:
return "Lebensmittel nicht gefunden.", 404
@app.route('/add_nutrition', methods=['POST'])
@oidc.accept_token(require_token=True, scopes_required=['openid'])
def add_nutrition():
food = request.form.get('food')
kcal = float(request.form.get('kcal'))
ew = float(request.form.get('ew'))
fett = float(request.form.get('fett'))
kh = float(request.form.get('kh'))
bst = float(request.form.get('bst'))
ca = float(request.form.get('ca'))
print("test")
# Verbindung zur Datenbank herstellen und Daten einfügen
try:
conn = psycopg2.connect()
with conn.cursor() as cursor:
cursor.execute("INSERT INTO nutrition_table (name, kcal, ew, fett, kh, bst, ca) VALUES (%s, %s, %s, %s, %s, %s, %s)",
(food, kcal, ew, fett, kh, bst, ca))
return redirect(url_for('nutrition'))
finally:
if conn:
conn.close()
@app.route('/nutrition')
@oidc.require_login
def nutrition():
return render_template('nutrition.html')
app = ProxyFix(app, x_for=1, x_host=1)