144 lines
4.4 KiB
Python
Executable File
144 lines
4.4 KiB
Python
Executable File
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", methods=["GET", "POST"])
|
|
def categories():
|
|
if request.method == "GET":
|
|
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)
|
|
if request.method == "POST":
|
|
body = request.get_json()
|
|
try:
|
|
if 'parent' in body:
|
|
parent = database.TransactionCategory.get(database.TransactionCategory.name == body['parent'])
|
|
database.TransactionCategory.create(name=body['name'], parent=parent)
|
|
else:
|
|
database.TransactionCategory.create(name=body['name'])
|
|
return ('', 200)
|
|
except peewee.DoesNotExist:
|
|
abort(404)
|
|
except:
|
|
abort(500)
|
|
|
|
# Should never get here.
|
|
abort(500)
|
|
|
|
app.run() |