def update_portfolio(user, keyword, url, investment, ip):
portfolio = TallstreetPortfolio.get_invested(user, url, keyword)
if not(portfolio):
portfolio = TallstreetPortfolio(parent=user, key_name='url%skeyword%s' % (url.key(), keyword.key()))
portfolio.url = url
portfolio.keyword = keyword
portfolio.user = user
portfolio.money = investment
if investment == 0:
return 0
change = investment
else:
change = investment - portfolio.money
if change == 0:
return 0
portfolio.money = investment
if user.money_outstanding - change < 0:
raise forms.ValidationError("No Money left in account.")
user.money_outstanding -= change
if investment == 0:
portfolio.delete()
portfolio = None
else:
portfolio.put()
transaction = TallstreetTransaction(parent=user, change=change, new_amount=investment, account_balance=user.money, account_balance_outstanding=user.money_outstanding, portfolio=portfolio, user=user, ip=ip)
if change > 0:
transaction.description = "Bought in [%s] under %s" % (url.key().id_or_name(), keyword.tag)
else:
transaction.description = "Sold in [%s] under %s" % (url.key().id_or_name(), keyword.tag)
transaction.put()
user.put()
return change
def update_portfolio_gain(user, url, keyword, gain): portfolio = TallstreetPortfolio.get_invested(user, url, keyword) change = long(round(portfolio.money * (1.0 * gain / 100))) if change == 0: return 0 portfolio.money = portfolio.money + change user.money += change if portfolio.money <= 0: portfolio.delete() portfolio = None else: portfolio.put() transaction = TallstreetTransaction(parent=user, change=change, new_amount=portfolio.money, account_balance=user.money, account_balance_outstanding=user.money_outstanding, portfolio=portfolio, user=user) if change > 0: transaction.description = "Gained in [%s] under %s" % (url.key().id_or_name(), keyword.tag) else: transaction.description = "Lost in [%s] under %s" % (url.key().id_or_name(), keyword.tag) transaction.put() user.put() return change
