Added message to 'transfer between users'

This commit is contained in:
Christoffer Viken 2017-10-01 05:46:52 +00:00
parent 3ef2323276
commit 17c2d05294
1 changed files with 5 additions and 2 deletions

View File

@ -18,11 +18,13 @@ class TransferMenu(Menu):
self.add_to_context(' from ' + user1.name) self.add_to_context(' from ' + user1.name)
user2 = self.input_user('To user> ') user2 = self.input_user('To user> ')
self.add_to_context(' to ' + user2.name) self.add_to_context(' to ' + user2.name)
comment = self.input_str('Comment> ')
self.add_to_context(' (comment) ' + user2.name)
t1 = Transaction(user1, amount, t1 = Transaction(user1, amount,
'transfer to ' + user2.name) 'transfer to ' + user2.name + ' "' + comment + '"')
t2 = Transaction(user2, -amount, t2 = Transaction(user2, -amount,
'transfer from ' + user1.name) 'transfer from ' + user1.name + ' "' + comment + '"')
t1.perform_transaction() t1.perform_transaction()
t2.perform_transaction() t2.perform_transaction()
self.session.add(t1) self.session.add(t1)
@ -32,6 +34,7 @@ class TransferMenu(Menu):
print 'Transfered %d kr from %s to %s' % (amount, user1, user2) print 'Transfered %d kr from %s to %s' % (amount, user1, user2)
print 'User %s\'s credit is now %d kr' % (user1, user1.credit) print 'User %s\'s credit is now %d kr' % (user1, user1.credit)
print 'User %s\'s credit is now %d kr' % (user2, user2.credit) print 'User %s\'s credit is now %d kr' % (user2, user2.credit)
print 'Comment: $s' % comment
except sqlalchemy.exc.SQLAlchemyError, e: except sqlalchemy.exc.SQLAlchemyError, e:
print 'Could not perform transfer: %s' % e print 'Could not perform transfer: %s' % e
# self.pause() # self.pause()