From 9c1c8d381bd387beda38f4416f0f7ee38d89b372 Mon Sep 17 00:00:00 2001 From: Petr Baudis Date: Sat, 29 Sep 2012 00:22:32 +0200 Subject: [PATCH] brmbar.Shop.receipt_to_credit(): New method for converting receipts to credit --- brmbar3/brmbar/Shop.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/brmbar3/brmbar/Shop.py b/brmbar3/brmbar/Shop.py index f57e4c6..bcb794b 100644 --- a/brmbar3/brmbar/Shop.py +++ b/brmbar3/brmbar/Shop.py @@ -75,6 +75,12 @@ class Shop: return cost + def receipt_to_credit(self, user, credit, description): + transaction = self._transaction(responsible = user, description = "Receipt: " + description) + self.profits.credit(transaction, credit, user.name) + user.credit(transaction, credit, "Credit from receipt: " + description) + self.db.commit() + def _transaction(self, responsible = None, description = None): with closing(self.db.cursor()) as cur: cur.execute("INSERT INTO transactions (responsible, description) VALUES (%s, %s) RETURNING id",