def run(self, username): self.session = get_session() user = self.session.query(User).filter_by(username=username).first() if not user: raise UserNotFoundError(username) transaction = self.session.query(Transaction).filter_by( username=user.username, operation='SELL', committed=False ).first() if not transaction: raise NoSellTransactionError(username) if (datetime.now() - transaction.creation_time) > config.TRANSACTION_TIMEOUT: self.session.delete(transaction) self.session.commit() raise ExpiredSellTransactionError(username) price = transaction.stock_value * transaction.quantity user.account_balance += price # update the StockPurchase for this stock symbol stock = self.session.query(StockPurchase).filter_by( user=user, stock_symbol=transaction.stock_symbol ).one() stock.quantity = stock.quantity - transaction.quantity transaction.committed = True self.session.commit() xml.log_transaction('COMMIT_SELL', transaction, status_message='success') return xml.ResultResponse('success')
def run(self, username, stock_symbol, amount): self.session = get_session() user = self.session.query(User).filter_by(username=username).first() if not user: raise UserNotFoundError(username) xml.log_error('BUY', 'username not found') # Check for existing uncommitted transaction if self.session.query(Transaction).filter_by( user=user, operation='BUY', committed=False).count() > 0: raise BuyTransactionActiveError() xml.log_error('BUY', 'Outstanding Buy Exists') # Getting stock quote quote_client = get_quote_client() quote = quote_client.get_quote(stock_symbol, username) # Work out quantity of stock to buy, fail if not enough for one stock amount = Money.from_string(amount) quantity = amount_to_quantity(quote, amount) if quantity == 0: raise InsufficientFundsError() price = quote * quantity if user.account_balance < price: raise InsufficientFundsError() transaction = Transaction(user=user, quantity=quantity, operation='BUY', stock_symbol=stock_symbol, stock_value=quote, committed=False) self.session.add(transaction) self.session.commit() xml.log_transaction('BUY', transaction) return xml.QuoteResponse(quantity=quantity, price=price)
def run(self, username, stock_symbol, money_amount): # see if user exists self.session = get_session() user = self.session.query(User).filter_by(username=username).first() if not user: raise UserNotFoundError(username) # Check for existing uncommitted transaction if self.session.query(Transaction).filter_by( user=user, operation='SELL', committed=False).count() > 0: raise SellTransactionActiveError() #set up client to get quote quote_client = get_quote_client() quoted_stock_value = quote_client.get_quote(stock_symbol, username) # Work out quantity of stock to sell, fail if not enough for one stock money_amount = Money.from_string(money_amount) quantity_to_sell = amount_to_quantity(quoted_stock_value, money_amount) if quantity_to_sell == 0: raise InsufficientFundsError() # see if the user owns the requested stock and has enough for request records = self.session.query(StockPurchase).filter_by( username=user.username, stock_symbol=stock_symbol).all() if(len(records) > 1): raise UnknownCommandError('Multiple StockPurchase for user %s: %d', username, len(records)) if len(records) != 1 or records[0].quantity < quantity_to_sell: raise InsufficientStockError() price = quoted_stock_value * quantity_to_sell # make transaction self.trans = Transaction(username=user.username, stock_symbol=stock_symbol, operation='SELL', committed=False, quantity=quantity_to_sell, stock_value=quoted_stock_value) # commit transaction after all actions for atomicity self.session.add(self.trans) self.session.commit() xml.log_transaction('SELL', self.trans, status_message='success') return xml.QuoteResponse(quantity=quantity_to_sell, price=price)
def run(self, username): self.session = get_session() user = self.session.query(User).filter_by(username=username).first() if not user: xml.log_error('CANCEL_BUY', 'username not found') raise UserNotFoundError(username) transaction = self.session.query(Transaction).filter_by( username=user.username, operation='BUY', committed=False ).first() if not transaction: raise NoBuyTransactionError(username) self.session.delete(transaction) self.session.commit() xml.log_transaction('CANCEL_BUY', transaction, status_message='success') return xml.ResultResponse('success')
def run(self, username): self.session = get_session() user = self.session.query(User).filter_by(username=username).first() if not user: xml.log_error('COMMIT_BUY', 'username not found') raise UserNotFoundError(username) transaction = self.session.query(Transaction).filter_by( username=user.username, operation='BUY', committed=False ).first() if not transaction: xml.log_error('COMMIT_BUY', 'no buy tranaction found') raise NoBuyTransactionError(username) if (datetime.now() - transaction.creation_time) > config.TRANSACTION_TIMEOUT: self.session.delete(transaction) self.session.commit() raise ExpiredBuyTransactionError(username) price = transaction.stock_value * transaction.quantity user.account_balance -= price transaction.committed = True # create or update the StockPurchase for this stock symbol stock = self.session.query(StockPurchase).filter_by( user=user, stock_symbol=transaction.stock_symbol ).first() if not stock: stock = StockPurchase(user=user, stock_symbol=transaction.stock_symbol, quantity=transaction.quantity) else: stock.quantity = stock.quantity + transaction.quantity self.session.commit() xml.log_transaction('COMMIT_BUY', transaction, status_message='success') return xml.ResultResponse('success')