diff --git a/budget.py b/budget.py
index 52c8cdb..08fbd1a 100644
--- a/budget.py
+++ b/budget.py
@@ -15,6 +15,7 @@ def init_db():
credit REAL DEFAULT 0,
debit REAL DEFAULT 0,
amount REAL NOT NULL,
+ description TEXT,
created_at TEXT NOT NULL
)
""")
@@ -69,6 +70,7 @@ input, button {{
<form method="POST" action="/add">
<input name="amount" type="text" placeholder="Amount (+ or -)" required>
+ <input name="description" type="text" placeholder="Description (optional)">
<button type="submit">Add</button>
</form>
@@ -112,6 +114,7 @@ class Handler(BaseHTTPRequestHandler):
data = parse_qs(body)
amount = float(data["amount"][0])
+ description = data.get("description", [""])[0]
# Determine if the amount should be credit or debit
if amount > 0:
@@ -123,8 +126,8 @@ class Handler(BaseHTTPRequestHandler):
with sqlite3.connect(DB) as conn:
conn.execute(
- "INSERT INTO entries(credit, debit, amount, created_at) VALUES (?, ?, ?, ?)",
- (credit, debit, amount, datetime.now().isoformat())
+ "INSERT INTO entries(credit, debit, amount, description, created_at) VALUES (?, ?, ?, ?, ?)",
+ (credit, debit, amount, description, datetime.now().isoformat())
)
self.send_response(303)