def add_transaction(self): symbol = self.cbbSymbol.get() if (not symbol): self.set_status("Symbol is empty.") return buy_or_sell = self.cbbBuyOrSell.get() if (not buy_or_sell): self.set_status("Buy/Sell is empty.") return quantity = self.entryQuantity.get() if (not quantity): self.set_status("Quantity is empty.") return price = self.entryPrice.get() if (not price): self.set_status("Price is empty.") return # date = self.entryDate.get() # if (not date): # self.set_status("Date is empty.") # return stock_transaction = StockTransaction() stock_transaction.set_symbol(symbol) stock_transaction.set_buy_or_sell(buy_or_sell) stock_transaction.set_quantity(quantity) stock_transaction.set_price(price) # stock_transaction.set_date(date) stock_transaction.set_date(datetime.datetime.now()) stock_transaction_table = StockTransactionTable() stock_transaction_table.add_stock_transaction(stock_transaction) self.set_status("Added") return
def add_transaction(self, symbol, buy_or_sell, quantity, price): stock_transaction = StockTransaction() stock_transaction.set_symbol(symbol) stock_transaction.set_buy_or_sell(buy_or_sell) stock_transaction.set_quantity(quantity) stock_transaction.set_price(price) stock_transaction.set_date(datetime.datetime.now()) stock_transaction_table = StockTransactionTable() stock_transaction_table.add_stock_transaction(stock_transaction) return
def test_stock_transaction_sanity(self): stock_db_connection = get_default_db_connection() reset_table(stock_db_connection) stock_transaction_table = StockTransactionTable(stock_db_connection) stock_transaction = StockTransaction() stock_transaction.set_symbol("601398") stock_transaction.set_buy_or_sell("buy") stock_transaction.set_quantity(100) stock_transaction.set_price(4.51) stock_transaction.set_date(date(2015, 11, 10)) stock_transaction_table.add_stock_transaction(stock_transaction) stock_transaction = StockTransaction() stock_transaction.set_symbol("601857") stock_transaction.set_buy_or_sell("buy") stock_transaction.set_quantity(100) stock_transaction.set_price(4.51) stock_transaction.set_date(date(2015, 11, 10)) stock_transaction_table.add_stock_transaction(stock_transaction) stock_transaction = \ stock_transaction_table.get_stock_transaction_by_trans_id(1) stock_transaction.set_quantity("500") stock_transaction_table.update_stock_transaction(stock_transaction) stock_transaction = \ stock_transaction_table.get_stock_transaction_by_trans_id(1) stock_transaction_table.delete_stock_transaction(stock_transaction) return