#12 - stored function for cash deposit transactions

brmbar3/brmbar/Shop.py

@@ -87,10 +87,17 @@ class Shop:
         return cost
 
     def add_credit(self, credit, user):
-        transaction = self._transaction(responsible = user, description = "BrmBar credit replenishment for " + user.name)
-        self.cash.debit(transaction, credit, user.name)
-        user.credit(transaction, credit, "Credit replenishment")
+        trn = self.db.execute_and_fetch(
+            "SELECT public.add_credit(%s,  %s, %s, %s)",
+            [self.cash.id, credit, user.id, user.name]
+        )[0]
         self.db.commit()
+        return trn # unused
+
+        #transaction = self._transaction(responsible = user, description = "BrmBar credit replenishment for " + user.name)
+        #self.cash.debit(transaction, credit, user.name)
+        #user.credit(transaction, credit, "Credit replenishment")
+        #self.db.commit()
 
     def withdraw_credit(self, credit, user):
         transaction = self._transaction(responsible = user, description = "BrmBar credit withdrawal for " + user.name)

brmbar3/schema/0012-shop-add-credit.sql

@@ -0,0 +1,110 @@
+--
+-- 0012-shop-add-credit.sql
+--
+-- #12 - stored function for cash deposit transactions
+--
+-- ISC License
+--
+-- Copyright 2023-2025 Brmlab, z.s.
+-- TMA <tma+hs@jikos.cz>
+--
+-- Permission to use, copy, modify, and/or distribute this software
+-- for any purpose with or without fee is hereby granted, provided
+-- that the above copyright notice and this permission notice appear
+-- in all copies.
+--
+-- THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
+-- WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
+-- WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
+-- AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR
+-- CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
+-- OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
+-- NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+-- CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+--
+
+-- To require fully-qualified names
+SELECT pg_catalog.set_config('search_path', '', false);
+
+DO $upgrade_block$
+BEGIN
+
+IF brmbar_privileged.has_exact_schema_version(11) THEN
+
+CREATE OR REPLACE FUNCTION public.add_credit(
+	i_cash_account_id public.accounts.id%TYPE,
+	i_credit NUMERIC,
+	i_user_id public.accounts.id%TYPE,
+	i_user_name TEXT
+) RETURNS VOID
+LANGUAGE plpgsql
+AS $$
+DECLARE
+	v_transaction_id public.transactions.id%TYPE;
+BEGIN
+	-- Create a new transaction
+	v_transaction_id := brmbar_privileged.create_transaction(i_user_id, 'BrmBar credit replenishment for ' || i_user_name);
+	-- Debit cash (credit replenishment)
+	INSERT INTO public.transaction_splits (transaction, side, account, amount, memo)
+	VALUES (v_transaction_id, 'debit', i_cash_account_id, i_credit, i_user_name);
+	-- Credit the user
+	INSERT INTO public.transaction_splits (transaction, side, account, amount, memo)
+	VALUES (v_transaction_id, 'credit', i_user_id, i_credit, 'Credit replenishment');
+END;
+$$;
+
+
+
+CREATE OR REPLACE FUNCTION public.undo_sale_of_item(
+	i_item_id public.accounts.id%TYPE,
+	i_amount INTEGER,
+	i_user_id public.accounts.id%TYPE,
+	i_target_currency_id public.currencies.id%TYPE,
+	i_description TEXT
+) RETURNS NUMERIC
+LANGUAGE plpgsql
+AS $$
+DECLARE
+	v_buy_rate NUMERIC;
+	v_sell_rate NUMERIC;
+	v_cost NUMERIC;
+	v_profit NUMERIC;
+	v_transaction_id public.transactions.id%TYPE;
+BEGIN
+	-- Get the buy and sell rates from the stored functions
+	v_buy_rate := public.find_buy_rate(i_item_id, i_target_currency_id);
+	v_sell_rate := public.find_sell_rate(i_item_id, i_target_currency_id);
+
+	-- Calculate cost and profit
+	v_cost := i_amount * v_sell_rate;
+	v_profit := i_amount * (v_sell_rate - v_buy_rate);
+
+	-- Create a new transaction
+	v_transaction_id := brmbar_privileged.create_transaction(i_user_id, i_description);
+
+	-- the item (decrease stock)
+	INSERT INTO public.transaction_splits (transaction, side, account, amount, memo)
+	VALUES (i_transaction_id, 'debit', i_item_id, i_amount,
+	(SELECT "name" || ' (sale undo)' FROM public.accounts WHERE id = i_user_id));
+
+	-- the user
+	INSERT INTO public.transaction_splits (transaction, side, account, amount, memo)
+	VALUES (i_transaction_id, 'credit', i_user_id, v_cost,
+	(SELECT "name" || ' (sale undo)' FROM public.accounts WHERE id = i_item_id));
+
+	-- the profit
+	INSERT INTO public.transaction_splits (transaction, side, account, amount, memo)
+	VALUES (i_transaction_id, 'credit', (SELECT account_id FROM accounts WHERE name = 'BrmBar Profits'), v_profit, (SELECT 'Margin repaid on ' || "name" FROM public.accounts WHERE id = i_item_id));
+
+	-- Return the cost
+	RETURN v_cost;
+END;
+$$;
+
+PERFORM brmbar_privileged.upgrade_schema_version_to(12);
+END IF;
+
+END;
+$upgrade_block$;
+
+-- vim: set ft=plsql :