From 7f11aa00673b0f77523db44969699c54289ace5b Mon Sep 17 00:00:00 2001
From: pacien
Date: Sun, 25 Jul 2021 21:53:59 +0200
Subject: app: working web prototype

---
 app/app_wallet.py | 95 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100644 app/app_wallet.py

(limited to 'app/app_wallet.py')

diff --git a/app/app_wallet.py b/app/app_wallet.py
new file mode 100644
index 0000000..9dffbd5
--- /dev/null
+++ b/app/app_wallet.py
@@ -0,0 +1,95 @@
+# UGE / L2 / Intro to relational databases / Python project prototype
+# Author: Pacien TRAN-GIRARD
+# Licence: EUPL-1.2
+
+from decimal import Decimal
+
+from fastapi import APIRouter, Form, Depends, status
+
+from embrace.exceptions import IntegrityError
+from psycopg2.errors import CheckViolation
+from psycopg2.extensions import ISOLATION_LEVEL_SERIALIZABLE
+
+from app_sessions import UserSession
+from app_database import db_transaction
+
+
+router = APIRouter()
+
+
+# TODO: add paging for the transaction history
+@router.get('/wallet')
+def wallet(
+    session: UserSession=Depends(UserSession.authenticated),
+):
+    with db_transaction() as tx:
+        history = tx.fetch_transactions(user_id=session.get_user_id())
+        return list(history)
+
+
+@router.post('/wallet/transfer')
+def wallet_transfer(
+    session: UserSession=Depends(UserSession.authenticated),
+    recipient: str=Form(...),
+    amount: Decimal=Form(...),
+):
+    if amount <= 0:
+        return 'error: Invalid transaction amount.'
+
+    try:
+        with db_transaction(ISOLATION_LEVEL_SERIALIZABLE) as tx:
+            recipient_user = tx.fetch_account_username(username=recipient)
+            if recipient_user is None:
+                raise LookupError('Could not find recipient')
+
+            tx.transfer(
+                from_user_id=session.get_user_id(),
+                to_user_id=recipient_user.id,
+                amount=amount,
+                fee=amount * Decimal(0.10))
+
+        return 'Your business is appreciated.'
+
+    except LookupError as exception:
+        return 'error: ' + str(exception)
+
+    except IntegrityError as exception:
+        if isinstance(exception.__cause__, CheckViolation):
+            return 'error: Insufficient funds.'
+        else:
+            raise exception
+
+
+@router.post('/wallet/deposit')
+def wallet_deposit(
+    session: UserSession=Depends(UserSession.authenticated),
+    amount: Decimal=Form(...),
+):
+    if amount <= 0:
+        return 'error: Invalid transaction amount.'
+
+    with db_transaction(ISOLATION_LEVEL_SERIALIZABLE) as tx:
+        tx.deposit(user_id=session.get_user_id(), amount=amount)
+
+    return 'Your business is appreciated.'
+
+
+@router.post('/wallet/withdraw')
+def wallet_withdraw(
+    session: UserSession=Depends(UserSession.authenticated),
+    amount: Decimal=Form(...),
+):
+    if amount <= 0:
+        return 'error: Invalid transaction amount.'
+
+    try:
+        with db_transaction(ISOLATION_LEVEL_SERIALIZABLE) as tx:
+            tx.withdraw(user_id=session.get_user_id(), amount=amount)
+
+        return 'Annnnnd... It\'s gone.'
+
+    except IntegrityError as exception:
+        if isinstance(exception.__cause__, CheckViolation):
+            return 'error: Insufficient funds.'
+        else:
+            raise exception
-- 
cgit v1.2.3