1 files changed, 61 insertions, 0 deletions
diff --git a/src/expense/tracker/create.py b/src/expense/tracker/create.py
new file mode 100644
index 0000000..c6f8308
--- /dev/null
+++ b/src/expense/tracker/create.py
@@ -0,0 +1,61 @@
+from datetime import date
+from flask import (
+    redirect,
+    render_template,
+    url_for,
+)
+from flask_login import login_required, current_user
+
+from .. import db
+
+from flask_wtf import FlaskForm
+from wtforms import (
+    DateField,
+    TextAreaField,
+    DecimalField,
+    SubmitField,
+    SelectField,
+)
+from wtforms.widgets import NumberInput
+from wtforms.validators import InputRequired
+
+from ..model import Category, Expense
+from .blueprint import tracker
+
+
+class CreateForm(FlaskForm):
+    """
+    The Form to create a entry
+    """
+    description = TextAreaField()
+    date = DateField(
+        validators=[InputRequired()], format="%Y-%m-%d", default=date.today
+    )
+    amount = DecimalField(places=2, validators=[InputRequired()], widget=NumberInput(), default=0)  # type: ignore
+    category = SelectField(coerce=int)  # type: ignore
+    submit = SubmitField()
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.category.choices = [(cat.id, cat.name) for cat in current_user.categories]  # type: ignore
+
+
+@tracker.route("/create", methods=["GET", "POST"])
+@login_required
+def create():
+    """The view to handle new entry"""
+    form = CreateForm()
+    if form.validate_on_submit():
+        cat = db.session.scalars(
+            db.select(Category).where(Category.id == form.category.data)
+        ).one()
+        expense = Expense(
+            description=form.description.data,
+            date=form.date.data,
+            amount=form.amount.data,
+            category=cat,
+        )  # type: ignore
+        current_user.expenses.append(expense)  # type: ignore
+        db.session.commit()
+        return redirect(url_for("index"))
+    return render_template("tracker/create.html", form=form)