forked from brmlab/brmbar-github
#6: new currency stored function
This commit is contained in:
parent
8f42145bee
commit
9235607d4c
3 changed files with 54 additions and 3 deletions
|
@ -42,7 +42,7 @@ class Account:
|
||||||
""" Constructor for new account """
|
""" Constructor for new account """
|
||||||
# id = db.execute_and_fetch("INSERT INTO accounts (name, currency, acctype) VALUES (%s, %s, %s) RETURNING id", [name, currency.id, acctype])
|
# id = db.execute_and_fetch("INSERT INTO accounts (name, currency, acctype) VALUES (%s, %s, %s) RETURNING id", [name, currency.id, acctype])
|
||||||
id = db.execute_and_fetch("SELECT public.create_account(%s, %s, %s)", [name, currency.id, acctype])
|
id = db.execute_and_fetch("SELECT public.create_account(%s, %s, %s)", [name, currency.id, acctype])
|
||||||
id = id[0]
|
# id = id[0]
|
||||||
return cls(db, name = name, id = id, currency = currency, acctype = acctype)
|
return cls(db, name = name, id = id, currency = currency, acctype = acctype)
|
||||||
|
|
||||||
def balance(self):
|
def balance(self):
|
||||||
|
|
|
@ -31,8 +31,9 @@ class Currency:
|
||||||
@classmethod
|
@classmethod
|
||||||
def create(cls, db, name):
|
def create(cls, db, name):
|
||||||
""" Constructor for new currency """
|
""" Constructor for new currency """
|
||||||
id = db.execute_and_fetch("INSERT INTO currencies (name) VALUES (%s) RETURNING id", [name])
|
# id = db.execute_and_fetch("INSERT INTO currencies (name) VALUES (%s) RETURNING id", [name])
|
||||||
id = id[0]
|
id = db.execute_and_fetch("SELECT public.create_currency(%s)", [name])
|
||||||
|
# id = id[0]
|
||||||
return cls(db, name = name, id = id)
|
return cls(db, name = name, id = id)
|
||||||
|
|
||||||
def rates(self, other):
|
def rates(self, other):
|
||||||
|
|
50
brmbar3/schema/0006-new-currency.sql
Normal file
50
brmbar3/schema/0006-new-currency.sql
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
--
|
||||||
|
-- 0006-new-currency.sql
|
||||||
|
--
|
||||||
|
-- #6 - stored procedure for creating new currency
|
||||||
|
--
|
||||||
|
-- ISC License
|
||||||
|
--
|
||||||
|
-- Copyright 2023-2025 Brmlab, z.s.
|
||||||
|
-- Dominik Pantůček <dominik.pantucek@trustica.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.
|
||||||
|
--
|
||||||
|
|
||||||
|
-- Require fully-qualified names
|
||||||
|
SELECT pg_catalog.set_config('search_path', '', false);
|
||||||
|
|
||||||
|
DO $upgrade_block$
|
||||||
|
BEGIN
|
||||||
|
|
||||||
|
IF brmbar_privileged.has_exact_schema_version(5) THEN
|
||||||
|
|
||||||
|
CREATE OR REPLACE FUNCTION public.create_currency(
|
||||||
|
IN i_name public.currencies.name%TYPE
|
||||||
|
) RETURNS INTEGER LANGUAGE plpgsql AS $$
|
||||||
|
DECLARE
|
||||||
|
r_id INTEGER;
|
||||||
|
BEGIN
|
||||||
|
INSERT INTO public.currencies (name)
|
||||||
|
VALUES (i_name) RETURNING id INTO r_id;
|
||||||
|
RETURN r_id;
|
||||||
|
END
|
||||||
|
$$;
|
||||||
|
|
||||||
|
PERFORM brmbar_privileged.upgrade_schema_version_to(6);
|
||||||
|
END IF;
|
||||||
|
|
||||||
|
END;
|
||||||
|
$upgrade_block$;
|
Loading…
Add table
Add a link
Reference in a new issue