import conf import sqlalchemy from db import Transaction, Product, User from helpers import less from text_interface.helpermenus import Menu, Selector class TransferMenu(Menu): def __init__(self): Menu.__init__(self, 'Transfer credit between users', uses_db=True) def _execute(self): self.print_header() amount = self.input_int('Transfer amount> ', (1, 100000)) self.set_context('Transfering %d kr' % amount, display=False) user1 = self.input_user('From user> ') self.add_to_context(' from ' + user1.name) user2 = self.input_user('To user> ') self.add_to_context(' to ' + user2.name) t1 = Transaction(user1, amount, 'transfer to ' + user2.name) t2 = Transaction(user2, -amount, 'transfer from ' + user1.name) t1.perform_transaction() t2.perform_transaction() self.session.add(t1) self.session.add(t2) try: self.session.commit() 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' % (user2, user2.credit) except sqlalchemy.exc.SQLAlchemyError, e: print 'Could not perform transfer: %s' % e # self.pause() class ShowUserMenu(Menu): def __init__(self): Menu.__init__(self, 'Show user', uses_db=True) def _execute(self): self.print_header() user = self.input_user('User name, card number or RFID> ') print 'User name: %s' % user.name print 'Card number: %s' % user.card print 'RFID: %s' % user.rfid print 'Credit: %s kr' % user.credit selector = Selector('What do you want to know about %s?' % user.name, items=[('transactions', 'Recent transactions (List of last ' + str( conf.user_recent_transaction_limit) + ')'), ('products', 'Which products %s has bought, and how many' % user.name), ('transactions-all', 'Everything (List of all transactions)')]) what = selector.execute() if what == 'transactions': self.print_transactions(user, conf.user_recent_transaction_limit) elif what == 'products': self.print_purchased_products(user) elif what == 'transactions-all': self.print_all_transactions(user) else: print 'What what?' @staticmethod def print_all_transactions(user): num_trans = len(user.transactions) string = '%s\'s transactions (%d):\n' % (user.name, num_trans) for t in user.transactions[::-1]: string += ' * %s: %s %d kr, ' % \ (t.time.strftime('%Y-%m-%d %H:%M'), {True: 'in', False: 'out'}[t.amount < 0], abs(t.amount)) if t.purchase: string += 'purchase (' string += ', '.join(map(lambda e: e.product.name, t.purchase.entries)) string += ')' if t.penalty > 1: string += ' * %dx penalty applied' % t.penalty else: string += t.description string += '\n' less(string) @staticmethod def print_transactions(user, limit=10): num_trans = len(user.transactions) if num_trans <= limit: string = '%s\'s transactions (%d):\n' % (user.name, num_trans) else: string = '%s\'s transactions (%d, showing only last %d):\n' % (user.name, num_trans, limit) for t in user.transactions[-1:-limit - 1:-1]: string += ' * %s: %s %d kr, ' % \ (t.time.strftime('%Y-%m-%d %H:%M'), {True: 'in', False: 'out'}[t.amount < 0], abs(t.amount)) if t.purchase: string += 'purchase (' string += ', '.join(map(lambda e: e.product.name, t.purchase.entries)) string += ')' if t.penalty > 1: string += ' * %dx penalty applied' % t.penalty else: string += t.description string += '\n' less(string) @staticmethod def print_purchased_products(user): products = [] for ref in user.products: product = ref.product count = ref.count products.append((product, count)) num_products = len(products) if num_products == 0: print 'No products purchased yet' else: text = '' text += 'Products purchased:\n' for product, count in products: text += u'{0:<47} {1:>3}\n'.format(product.name, count) less(text) class UserListMenu(Menu): def __init__(self): Menu.__init__(self, 'User list', uses_db=True) def _execute(self): self.print_header() user_list = self.session.query(User).all() total_credit = self.session.query(sqlalchemy.func.sum(User.credit)).first()[0] line_format = '%-12s | %6s\n' hline = '---------------------\n' text = '' text += line_format % ('username', 'credit') text += hline for user in user_list: text += line_format % (user.name, user.credit) text += hline text += line_format % ('total credit', total_credit) less(text) class AdjustCreditMenu(Menu): # reimplements ChargeMenu; these should be combined to one def __init__(self): Menu.__init__(self, 'Adjust credit', uses_db=True) def _execute(self): self.print_header() user = self.input_user('User> ') print 'User %s\'s credit is %d kr' % (user.name, user.credit) self.set_context('Adjusting credit for user %s' % user.name, display=False) print '(Note on sign convention: Enter a positive amount here if you have' print 'added money to the PVVVV money box, a negative amount if you have' print 'taken money from it)' amount = self.input_int('Add amount> ', (-100000, 100000)) print '(The "log message" will show up in the transaction history in the' print '"Show user" menu. It is not necessary to enter a message, but it' print 'might be useful to help you remember why you adjusted the credit)' description = self.input_str('Log message> ', length_range=(0, 50)) if description == '': description = 'manually adjusted credit' transaction = Transaction(user, -amount, description) transaction.perform_transaction() self.session.add(transaction) try: self.session.commit() print 'User %s\'s credit is now %d kr' % (user.name, user.credit) except sqlalchemy.exc.SQLAlchemyError, e: print 'Could not store transaction: %s' % e # self.pause() class ProductListMenu(Menu): def __init__(self): Menu.__init__(self, 'Product list', uses_db=True) def _execute(self): self.print_header() text = '' product_list = self.session.query(Product).filter(Product.hidden.is_(False)).order_by(Product.stock.desc()) total_value = 0 for p in product_list: total_value += p.price * p.stock line_format = '%-15s | %5s | %-' + str(Product.name_length) + 's | %5s \n' text += line_format % ('bar code', 'price', 'name', 'stock') text += 78 * '-' + '\n' for p in product_list: text += line_format % (p.bar_code, p.price, p.name, p.stock) text += 78 * '-' + '\n' text += line_format % ('Total value', total_value, '', '',) less(text) class ProductSearchMenu(Menu): def __init__(self): Menu.__init__(self, 'Product search', uses_db=True) def _execute(self): self.print_header() self.set_context('Enter (part of) product name or bar code') product = self.input_product() print 'Result: %s, price: %d kr, bar code: %s, stock: %d, hidden: %s' % (product.name, product.price, product.bar_code, product.stock, ("Y" if product.hidden else "N")) # self.pause()