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 delete_stock_cash(self): index = self.lstboxStockTransaction.curselection() if len(index) == 0: return list_box_string = self.lstboxStockTransaction.get(index[0]) list_box_string_list = list_box_string.split(",") id_string = list_box_string_list[0] trans_id = int(id_string[3:]) stock_transaction_table = StockTransactionTable() stock_transaction = \ stock_transaction_table.get_stock_transaction_by_trans_id(trans_id) stock_transaction_table.delete_stock_transaction(stock_transaction) self.refresh_list_box() return