Exemplo n.º 1
0
class SmarttConsole(Cmd):
    ##########################################################################
    ### Configurations ###
    ######################
    prompt = "smartt> "

    def preloop(self):
        self.smartt_client = SmarttClient()
    ##########################################################################

    def splitArgs(self, arg):
        extracted_values = re.findall('("([^"]*)")|(\S+)', arg)
        values = [(value[2] if value[0] == "" else value[1])
                  for value in extracted_values]
        return values

    def printValue(self, value):
        if isinstance(value, dict):
            for (name, value) in value.iteritems():
                print "%s: %s" % (name, value)
        elif isinstance(value, list):
            index = 0
            for element in value:
                print str(index) + ":"
                self.printValue(element)
                index += 1
                print ""
        else:
            print value

    def printResponse(self, response):
        print BLUECOLORCODE

        self.printValue(response)

        print ENDOFCOLORCODE

    ##########################################################################
    ### Smartt Functions ###
    ########################
    def do_login(self, arg):
        splitted_args = self.splitArgs(arg)
        if len(splitted_args) == 0:
            print "Login not specified"
            return

        username = splitted_args[0]

        print "Logging in as '%s'" % username
        password = getpass.getpass()

        self.printResponse(self.smartt_client.login(username, password))

    def do_logged(self, arg):
        self.printResponse(self.smartt_client.logged())

    def do_logout(self, arg):
        self.printResponse(self.smartt_client.logout())

    def do_get_client(self, arg):
        attributes = self.splitArgs(arg)
        self.printResponse(self.smartt_client.getClient(attributes))

    def do_get_time(self, arg):
        self.printResponse(self.smartt_client.getTime())

    def do_get_stock(self, arg):
        splitted_args = self.splitArgs(arg)

        stock_code = splitted_args[0]
        market_name = splitted_args[1]
        attributes = splitted_args[2:]

        self.printResponse(
            self.smartt_client.getStock(stock_code, market_name, attributes))

    def do_send_order(self, arg):
        splitted_args = self.splitArgs(arg)

        print "Not implemented"

    def do_cancel_order(self, arg):
        splitted_args = self.splitArgs(arg)

        order_id = splitted_args[0]

        self.printResponse(self.smartt_client.cancelOrder(order_id))

    def do_change_order(self, arg):
        splitted_args = self.splitArgs(arg)

        print "Not implemented"

    def do_send_stop_order(self, arg):
        splitted_args = self.splitArgs(arg)

        print "Not implemented"

    def do_cancel_stop_order(self, arg):
        splitted_args = self.splitArgs(arg)

        stop_order_id = splitted_args[0]

        self.printResponse(self.smartt_client.cancelStopOrder(stop_order_id))

    def do_get_orders(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getOrders())

    def do_get_orders_events(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getOrdersEvents())

    def do_get_stop_orders(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getStopOrders())

    def do_get_stop_orders_events(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getStopOrdersEvents())

    def do_get_trades(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getTrades())

    def do_get_portfolio(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getPortfolio())

    def do_get_available_limits(self, arg):
        splitted_args = self.splitArgs(arg)

        self.printResponse(self.smartt_client.getAvailableLimits())
    ##########################################################################

    ##########################################################################
    ### Lower level Smartt messaging ###
    ####################################
    def do_message(self, arg):
        message = self.splitArgs(arg)
        self.smartt_client.sendMessage(message)
        print self.smartt_client.receiveMessage()

    def do_query(self, arg):
        self.do_message(arg)

    def do_rawmessage(self, arg):
        self.smartt_client.sendRawMessage(arg)
        print self.smartt_client.receiveRawMessage()

    def do_rawquery(self, arg):
        self.do_rawmessage(arg)

    ##########################################################################
    ### Quitting commands ###
    #########################
    def do_EOF(self, arg):
        print ""
        return True

    def do_quit(self, arg):
        return True

    def do_exit(self, arg):
        return True
    ##########################################################################

    ##########################################################################
    ### Wrappers and overloaded functions ###
    #########################################
    def onecmd(self, line):
        ### Wraps commands and catches exceptions
        try:
            return Cmd.onecmd(self, line)
        except SmarttClientException as e:
            print REDCOLORCODE + str(e) + ENDOFCOLORCODE
            return False

    def emptyline(self):
        ### Empty lines do nothing - default is to run last command
        pass

    def default(self, line):
        ### Default message - unknown command
        print(REDCOLORCODE
              + "Unknown command: '{0}' - you sure you typed that right?"
              .format(line) + ENDOFCOLORCODE)
Exemplo n.º 2
0
 def preloop(self):
     self.smartt_client = SmarttClient()
Exemplo n.º 3
0
 def preloop(self):
     self.smartt_client = SmarttClient(use_ssl=False)