HabitTracker/app.py

import datetime
import hashlib

from flask import Flask, render_template, redirect, url_for, request
from flask_login import login_required, LoginManager, login_user, logout_user, current_user

from models.Habit import Habit
from models.HabitTrackings import HabitTrackings
from models.User import User
from utils import anonymous_required

# Create a new Flask instance
app = Flask(__name__)
app.secret_key = 'PSSSSSHHHT!'

# Initialize the Flask-Login extension
login_manager = LoginManager()
login_manager.login_view = 'login'
login_manager.init_app(app)


@login_manager.user_loader
def load_user(user_id):
    return User.get(user_id)


@app.context_processor
def inject_user():
    return dict(user=current_user)


@app.route('/login')
@anonymous_required
def login():
    return render_template('auth/login.html', errors={})


@app.route('/signup')
@anonymous_required
def signup():
    return render_template('auth/signup.html', errors={})


@app.route('/login', methods=['POST'])
def login_post():
    email = request.form.get('email')
    password = request.form.get('password')

    # Check for errors
    errors = {}
    if not email:
        errors['email'] = 'Die E-Mail Adresse ist erforderlich.'
    if not password:
        errors['password'] = 'Das Passwort ist erforderlich.'

    # Check if user exists
    user = User.get_by_email(email)

    if not user:
        errors['email'] = 'E-Mail Adresse nicht gefunden.'
    elif user.password is None or hashlib.sha256(password.encode()).hexdigest() != user.password:
        errors['password'] = 'Das Passwort ist falsch.'

    if errors:
        return render_template(
            'auth/login.html',
            email=email,
            password=password,
            errors=errors
        )

    login_user(user)

    # Redirect to login page
    return redirect(url_for('index'))


@app.route('/signup', methods=['POST'])
def signup_post():
    email = request.form.get('email')
    name = request.form.get('name')
    password = request.form.get('password')

    # Check for errors
    errors = {}
    if not email:
        errors['email'] = 'Die E-Mail Adresse ist erforderlich.'
    if not name:
        errors['name'] = 'Der Name ist erforderlich.'
    if not password:
        errors['password'] = 'Das Passwort ist erforderlich.'

    if errors:
        return render_template(
            'auth/signup.html',
            email=email,
            name=name,
            password=password,
            errors=errors
        )

    # Save user to database. Maybe log the user in directly.
    user = User.create(name, email, password)
    login_user(user)

    # Redirect to login page
    return redirect(url_for('index'))


@app.route('/logout')
@login_required
def logout():
    # Log out functionality
    logout_user()

    return redirect(url_for('index'))


# Create a new route
@app.route('/')
def index():
    if current_user.is_authenticated:
        habits = current_user.get_habits()
        name = "Hallo " + current_user.name
    else:
        habits = []
        name = "Bitte melde dich an."

    # Sort habits by whether they have been checked today and then by slot
    habits.sort(key=lambda habit: (habit.checked, habit.slot))

    return render_template(
        'index.html',
        title=name,
        utc_dt=datetime.datetime.now().strftime("%d.%m.%Y %H:%M %A"),
        habits=habits,
        errors={},
    )


@app.route('/habit')
@login_required
def habit_creation():
    return render_template(
        'habit.html',
        title='Erstelle ein Habit',
        unit="Woche",
        errors={},
    )


@app.route('/habit', methods=['POST'])
@login_required
def habit_create():
    name = request.form.get('name')
    note = request.form.get('note')
    times = request.form.get('times')
    unit = request.form.get('unit')

    # Check for errors
    errors = {}
    if not name:
        errors['name'] = 'Der Name ist erforderlich.'
    if not times:
        errors['times'] = 'Die Anzahl ist erforderlich.'
    if not note:
        note = ''
    if not unit:
        errors['unit'] = 'Die Einheit ist erforderlich.'

    # Check if times is an integer
    try:
        times = int(times)

        # Check that times is greater than 0
        if times <= 0:
            errors['times'] = 'Die Anzahl muss größer als 0 sein.'
    except ValueError:
        errors['times'] = 'Die Anzahl muss eine Zahl sein.'

    # Check that unit is valid
    if unit not in ['Tag', 'Woche', 'Monat', 'Jahr']:
        errors['unit'] = 'Die Einheit ist ungültig.'

    if errors:
        return render_template(
            'habit.html',
            title='Erstelle ein Habit',
            name=name,
            note=note,
            times=times,
            unit=unit,
            errors=errors
        )

    # Map unit to integer
    if unit == 'Tag':
        unit = 0
    elif unit == 'Woche':
        unit = 1
    elif unit == 'Monat':
        unit = 2
    elif unit == 'Jahr':
        unit = 3
    else:
        unit = 1

    # Save habit to database
    habit = Habit.create(current_user.id, name, times, note, unit)

    # Back to index
    return redirect(url_for('index'))

    """return render_template(
        'habit.html',
        title='Erstelle ein Habit',
        name=name,
        note=note,
        times=times,
        unit=unit,
        errors=errors,
    )"""


@app.route('/profile')
@login_required
def profile():
    return render_template(
        "profile.html",
        name=current_user.name,
        email=current_user.email,
        errors={}
    )


@app.route('/profile', methods=['POST'])
@login_required
def profile_change():
    newName = request.form.get('newName')
    newEmail = request.form.get('newEmail')
    newPassword = request.form.get('newPassword')
    oldPassword = request.form.get('oldPassword')

    # Check for errors
    errors = {}
    if not newName:
        errors['newName'] = 'Der Name ist erforderlich.'

    if not newEmail:
        errors['newEmail'] = 'Die E-Mail Adresse ist erforderlich.'

    if not oldPassword:
        errors['oldPassword'] = 'Du musst dein aktuelles Passwort angeben.'
    else:
        if hashlib.sha256(oldPassword.encode()).hexdigest() != current_user.password:
            errors['oldPassword'] = 'Das Passwort ist falsch.'

    if errors:
        return render_template(
            "profile.html",
            name=current_user.name,
            email=current_user.email,
            errors=errors
        )

    # Update user
    current_user.name = newName
    current_user.email = newEmail
    if newPassword:
        current_user.password = hashlib.sha256(newPassword.encode()).hexdigest()
    current_user.update()

    # Back to profile
    return render_template(
        "profile.html",
        name=current_user.name,
        email=current_user.email,
        errors={}
    )


@app.route('/check', methods=['POST'])
@login_required
def check_habit():
    habit_id = request.get_json()["habitId"]

    habit = Habit.get(habit_id)

    if habit is None:
        return {"error": "Habit not found"}

    # Check if habit belongs to user
    if habit.user_id != current_user.id:
        return {"error": "Habit does not belong to user"}

    trackings = habit.get_habitTrackings()

    # Check if habit has been tracked today
    delete_tracking = None
    for tracking in trackings:
        # day
        if habit.unit == 0:
            if tracking.created_at.date() == datetime.date.today():
                delete_tracking = tracking
                break
        # week
        elif habit.unit == 1:
            if tracking.created_at.date().isocalendar()[1] == datetime.date.today().isocalendar()[1]:
                delete_tracking = tracking
                break
        # month
        elif habit.unit == 2:
            if tracking.created_at.date().month == datetime.date.today().month:
                delete_tracking = tracking
                break
        # year
        elif habit.unit == 3:
            if tracking.created_at.date().year == datetime.date.today().year:
                delete_tracking = tracking
                break

    if not delete_tracking:
        HabitTrackings.create(habit_id, 1)
    else:
        delete_tracking.delete()

    # Update habit
    habit.fill_statistics()

    return {
        "habitId": habit_id,
        "unchecked": not delete_tracking,
        "percentage": habit.percentage,
    }

@app.route('/delete', methods=['POST'])
@login_required
def delete_habit():
    habit_id = request.get_json()["habitId"]

    habit = Habit.get(habit_id)

    if habit is None:
        return {"error": "Habit not found"}

    # Check if habit belongs to user
    if habit.user_id != current_user.id:
        return {"error": "Habit does not belong to user"}

    habit.delete()

    return {}


# Run the application
if __name__ == '__main__':
    app.run(port=5000, debug=True)