예제 #1
0
class UI(object):
    def __init__(self):
        self.driver = webdriver.Chrome()
        self.trader = Trader(self.driver, 3)
        self.trader.login()
        self.stockselector = None
        self.stockpredictor = None
        self.portfolio = {}

    def get_portfolio(self):
        self.trader.open_portfolio_page()
        return self.trader.get_portfolio()

    def trade(self, ticker, action, amount):
        self.trader.open_trading_page()
        self.trader.trade(self, ticker, action, amount)

    def __del__(self):
        self.trader.logout()
        self.driver.close()