brmbar.Shop.withdraw_credit(): New method analogous to add_credit()

This commit is contained in:
Petr Baudis 2012-09-24 04:39:30 +02:00
parent f261fe5ea5
commit eda6eaf9e4

View file

@ -57,6 +57,12 @@ class Shop:
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)
self.cash.credit(transaction, credit, user.name)
user.debit(transaction, credit, "Credit withdrawal")
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",