125 lines
5.2 KiB
Python
125 lines
5.2 KiB
Python
from math import ceil
|
|
|
|
import sqlalchemy
|
|
from db import Product, User, Transaction, PurchaseEntry, Purchase
|
|
from text_interface.helpermenus import Menu
|
|
|
|
|
|
class AddStockMenu(Menu):
|
|
def __init__(self):
|
|
Menu.__init__(self, 'Add stock and adjust credit', uses_db=True)
|
|
self.help_text = '''
|
|
Enter what you have bought for PVVVV here, along with your user name and how
|
|
much money you're due in credits for the purchase when prompted.\n'''
|
|
self.users = []
|
|
self.users = []
|
|
self.products = {}
|
|
self.price = 0
|
|
|
|
def _execute(self):
|
|
questions = {
|
|
(False, False): 'Enter user id or a string of the form "<number> <product>"',
|
|
(False, True): 'Enter user id or more strings of the form "<number> <product>"',
|
|
(True, False): 'Enter a string of the form "<number> <product>"',
|
|
(True, True): 'Enter more strings of the form "<number> <product>", or an empty line to confirm'
|
|
}
|
|
|
|
self.users = []
|
|
self.products = {}
|
|
self.price = 0
|
|
|
|
while True:
|
|
self.print_info()
|
|
self.printc(questions[bool(self.users), bool(len(self.products))])
|
|
thing_price = 0
|
|
|
|
# Read in a 'thing' (product or user):
|
|
line = self.input_multiple(add_nonexisting=('user', 'product'), empty_input_permitted=True,
|
|
find_hidden_products=False)
|
|
|
|
if line:
|
|
(thing, amount) = line
|
|
|
|
if isinstance(thing, Product):
|
|
self.printc("%d of %s registered" % (amount, thing.name))
|
|
thing_price = self.input_int('What did you pay a piece? ', (1, 100000),
|
|
default=thing.price) * amount
|
|
self.price += thing_price
|
|
|
|
# once we get something in the
|
|
# purchase, we want to protect the
|
|
# user from accidentally killing it
|
|
self.exit_confirm_msg = 'Abort transaction?'
|
|
else:
|
|
if not self.complete_input():
|
|
if self.confirm('Not enough information entered. Abort transaction?', default=True):
|
|
return False
|
|
continue
|
|
break
|
|
|
|
# Add the thing to the pending adjustments:
|
|
self.add_thing_to_pending(thing, amount, thing_price)
|
|
|
|
self.perform_transaction()
|
|
|
|
def complete_input(self):
|
|
return bool(self.users) and len(self.products) and self.price
|
|
|
|
def print_info(self):
|
|
print (6 + Product.name_length) * '-'
|
|
if self.price:
|
|
print "Amount to be credited: {{0:>{0}d}}".format(Product.name_length-17).format(self.price)
|
|
if self.users:
|
|
print "Users to credit:"
|
|
for user in self.users:
|
|
print " %s" % str(user.name)
|
|
print u"\n{{0:s}}{{1:>{0}s}}".format(Product.name_length-1).format("Product", "Amount")
|
|
print (6 + Product.name_length) * '-'
|
|
if len(self.products):
|
|
for product in self.products.keys():
|
|
print u'{{0:<{0}}}{{1:>6d}}'.format(Product.name_length).format(product.name, self.products[product][0])
|
|
print (6 + Product.name_length) * '-'
|
|
|
|
def add_thing_to_pending(self, thing, amount, price):
|
|
if isinstance(thing, User):
|
|
self.users.append(thing)
|
|
elif thing in self.products.keys():
|
|
print 'Already added this product, adding amounts'
|
|
self.products[thing][0] += amount
|
|
self.products[thing][1] += price
|
|
else:
|
|
self.products[thing] = [amount, price]
|
|
|
|
def perform_transaction(self):
|
|
description = self.input_str('Log message> ', length_range=(0, 50))
|
|
if description == '':
|
|
description = 'Purchased products for PVVVV, adjusted credit ' + str(self.price)
|
|
for product in self.products:
|
|
value = max(product.stock, 0) * product.price + self.products[product][1]
|
|
old_price = product.price
|
|
old_hidden = product.hidden
|
|
product.price = int(ceil(float(value) / (max(product.stock, 0) + self.products[product][0])))
|
|
product.stock += self.products[product][0]
|
|
product.hidden = False
|
|
print "New stock for %s: %d" % (product.name, product.stock), \
|
|
("- New price: " + str(product.price) if old_price != product.price else ""), \
|
|
("- Removed hidden status" if old_hidden != product.hidden else "")
|
|
|
|
purchase = Purchase()
|
|
for user in self.users:
|
|
Transaction(user, purchase=purchase, amount=-self.price, description=description)
|
|
for product in self.products:
|
|
PurchaseEntry(purchase, product, -self.products[product][0])
|
|
|
|
purchase.perform_soft_purchase(-self.price, round_up=False)
|
|
self.session.add(purchase)
|
|
|
|
try:
|
|
self.session.commit()
|
|
print "Success! Transaction performed:"
|
|
# self.print_info()
|
|
for user in self.users:
|
|
print "User %s's credit is now %i" % (user.name, user.credit)
|
|
except sqlalchemy.exc.SQLAlchemyError, e:
|
|
print 'Could not perform transaction: %s' % e
|