cycles/period/cycles.py

from itertools import cycle
from time import get_clock_info
from tracemalloc import start
from flask import (
    Blueprint, flash, g, redirect, render_template, request, url_for
)
from werkzeug.exceptions import abort
import pendulum

# from flaskr.auth import login_required
from period.db import get_db

bp = Blueprint('cycles', __name__)

@bp.route('/')
def index():
    db = get_db()
    current_cycle = db.execute('SELECT * from cycle ORDER BY id DESC LIMIT 1').fetchone()
    print(current_cycle['start_time'])
    start_time = pendulum.parse(current_cycle['start_time'])
    if current_cycle['end_time'] is None:
        cycle_length = (pendulum.now() - start_time).days

    db.execute(
        'UPDATE cycle SET cycle_length = ?'
        ' WHERE id = ?',
        (cycle_length, current_cycle['id'])
    )

    cycles = db.execute(
        'SELECT id, start_time, end_time, cycle_length'
        ' FROM cycle ORDER BY start_time DESC'
    ).fetchall()
    return render_template('cycle/index.html', cycles=cycles)

@bp.route('/create', methods=('GET', 'POST'))
def create():
    if request.method == 'POST':
        start_time_stamp = request.form['start_time']
        start_time = pendulum.from_format(start_time_stamp, 'YYYY-MM-DD')
        try:
            end_time_stamp = request.form['end_time']
            end_time = pendulum.from_format(request.form['end_time'], 'YYYY-MM-DD')
        except ValueError:
            end_time_stamp = "1900-01-01"
            end_time = pendulum.datetime(1900,1,1)
        error = None
        print(request.form['end_time'])

        if not start_time:
            error = "You must provide a start date"
        
        if end_time > start_time:
            cycle_length = (end_time - start_time).days
        else:
            cycle_length = (pendulum.now() - start_time).days

        if error is not None:
            flash(error)
        else:
            db = get_db()
            if end_time > start_time:
                db.execute(
                    'INSERT INTO cycle (start_time, end_time, cycle_length) VALUES (?, ?, ?)',
                    (start_time_stamp, end_time_stamp, cycle_length)
                )
            else:
                db.execute(
                    'INSERT INTO cycle (start_time)'
                    ' VALUES (?)',
                    (start_time.to_datetime_string(),)
                )
            db.commit()
            return redirect(url_for('index'))
    return render_template('cycle/create.html')

def get_cycle(id):
    cycle = get_db().execute(
        'SELECT * from cycle WHERE id = ?',
        (id,)
    ).fetchone()

    if cycle is None:
        abort(404, f"Cycle id {id} doesn't exist.")

    return cycle

@bp.route('/<int:id>/update', methods=('GET', 'POST'))
def update(id):
    cycle = get_cycle(id)
    print(cycle['end_time'])

    if request.method == 'POST':
        start_time_stamp = request.form['start_time']
        start_time = pendulum.from_format(start_time_stamp, 'YYYY-MM-DD')
        try:
            end_time_stamp = request.form['end_time']
            end_time = pendulum.from_format(end_time_stamp, 'YYYY-MM-DD')
            print(end_time)
        except ValueError:
            end_time_stamp = "1900-01-01"
            end_time = pendulum.datetime(1900,1,1)
        error = None
        if end_time is None:
            end_time = "2999-01-01"
        print(end_time)

        if not start_time:
                error = "You must provide a start date"

        if end_time > start_time:
                cycle_length = (end_time - start_time).days
        else:
            cycle_length = (pendulum.now() - start_time).days

        if error is not None:
            flash(error)
        else:
            db = get_db()
            if end_time > start_time:
                db.execute(
                    'UPDATE cycle SET start_time = ?, end_time = ?, cycle_length = ?'
                    ' WHERE id = ?',
                    (start_time.to_datetime_string(), end_time.to_datetime_string(), cycle_length, id)
                )
            else:
                db.execute(
                    'UPDATE cycle SET start_time = ?, cycle_length = ?'
                    ' WHERE id = ?',
                    (start_time.to_datetime_string(), cycle_length, id)
                )
            db.commit()
            return redirect(url_for('index'))
    return render_template('cycle/update.html', cycle=cycle)

@bp.route('/<int:id>/delete', methods=('POST',))
def delete(id):
    get_cycle(id)
    db = get_db()
    db.execute('DELETE FROM cycle WHERE id = ?', (id,))
    db.commit()
    return redirect(url_for('index'))