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
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
