kernel-demo/jupyter-mysql-kernel/parser.py

"""
	jupyter-mysql-kernel
	author:rabin
"""

class MysqlParser(object):

    def __init__(self, response, commands=False):
        self.response = response
        self.result = []
        self.is_array = False
        self.is_error = False
        if not commands:
            self.parse_response()
        else:
            self.parse_commands()

    def parse_commands(self):
        # get each section of the command response
        sections = self.response.split('*6\r\n')
        for section in sections:
            parts = section.split('\r\n')
            # only the second one is the command name
            # no need to parse for now
            if parts[1] is not None and parts[1].__len__() > 0:
                self.result.append(parts[1])

    def parse_response(self):
        # get each line of the response
        parts = self.response.split('\r\n')

        if parts[0].startswith('*'):
            self.is_array = True

        for part in parts:
            if part != '':
                value = self.parse_part(part)
                if value is not None:
                    self.result.append(value)

    def parse_part(self, part):
        if part[0] == '*':
            # array count
            return None
        elif part[0] in ['-', '+', ':']:
            if part[0] == '-':
                self.is_error = True
                #error or string or integer
                return part[1:]
            elif part[0] == '+':
                return part[1:]
            elif part[0] == ':':
                return int(part[1:])
        elif part[0] == '$':
            if part[1:] == '-1':
                # handle nil
                return 'nil'
            else:
                # ignore the byte count
                return None
        else:
            # values returned after the type specifier
            return part

    def html(self):
        out = None
        res = self.get_result()
        if self.is_error:
            out = "<p style='color:red'>" + res + '</p>'
        else:
            out = res
        return out

    def text(self):
        return self.get_result()

    def get_result(self):
        if self.result.__len__() > 1:
            out = []
            for x in self.result:
                if isinstance(x, int):
                    out.append(str(x))
                else:
                    out.append(x)
            return out
        elif self.result.__len__() > 0:
            if type(self.result[0] == int):
                return str(self.result[0])
            else:
                return self.result[0]
        else:
            self.is_error = True
            return 'Error executing command. There was no result.'