from datetime import datetime

from flask import Flask, render_template, request, abort
import peewee
from playhouse.shortcuts import model_to_dict

import database

app = Flask(__name__)
app.config["DEBUG"] = True

@app.before_request
def before_request():
    database.instance.connect()

@app.after_request
def after_request(response):
    database.instance.close()
    return response

@app.route("/")
def home():
    puppies = ['hollie', 'grace', 'loki']
    return render_template('index.html', puppies=puppies)

@app.route("/users")
def users():
    args = request.args.to_dict()
    if 'username' not in args:
        abort(400)
    
    username = args['username'].strip()
    try:
        return model_to_dict(database.User.get(database.User.username == username))
    except peewee.DoesNotExist:
        abort(404)
    except:
        abort(500)

@app.route("/transactions", methods=["GET", "POST"])
def transactions():
    if request.method == "GET":
        args = request.args.to_dict()
        if 'user_id' not in args:
            abort(400)

        user_id = args['user_id'].strip()
        try:
            july_2022 = datetime(2022, 7, 1, 0)
            transactions = database.Transaction.select().where(
                (database.Transaction.user == user_id) &
                (database.Transaction.transaction_date > july_2022)
            )
            return [model_to_dict(t) for t in transactions]
        except peewee.DoesNotExist:
            abort(404)
        except:
            abort(500)

    if request.method == "POST":
        body = request.get_json()
        try:
            user = database.User.get(database.User.uuid == body['user_id'])
            category = database.TransactionCategory.get(database.TransactionCategory.name == body['category'])
            database.Transaction.create(
                user=user,
                subcategory=category,
                transaction_date=body['transaction_date'],
                description=body['description'],
                amount=body['amount'],
                type=body['type'],
                notes=body['notes']
            )
            return ('', 200)
        except peewee.DoesNotExist:
            abort(404)
        except KeyError:
            abort(400)

    # Default if no method is hit
    abort(400)

@app.route("/transactions/<uuid>", methods=["PUT"])
def transaction(uuid=None):
    if request.method == "PUT":
        if not uuid:
            abort(400)
        
        body = request.get_json()
        if 'category' not in body or 'notes' not in body:
            abort(400)

        try:
            category = database.TransactionCategory.get(database.TransactionCategory.name == body['category'])
            transaction = database.Transaction.get(database.Transaction.primary_key == uuid)
            transaction.subcategory = category
            transaction.notes = body['notes']
            transaction.save()
            return ('', 200)
        except peewee.DoesNotExist:
            abort(404)
        except:
            abort(500)

    # Default if no method is hit
    abort(400)

@app.route("/categories")
def categories():
    try:
        raw_categories = database.TransactionCategory.select()
        categories = {} 
        for category in raw_categories:
            if not category.parent:
                categories[category.name] = []

        for category in raw_categories:
            if category.parent:
                categories[category.parent.name].append(category.name)

        return categories
    except peewee.DoesNotExist:
        abort(404)
    except:
        abort(500)

app.run()