示例#1
0
def breakquery(q):
    if len(q) > 1:
        raise functions.OperatorError(
            __name__.rsplit('.')[-1],
            "Ambiguous query column, result has more than one columns")
    st = ''
    for row in q[0].splitlines():
        strow = filterlinecomment(row)
        if strow == '':
            continue
        if st != '':
            st += '\n' + strow
        else:
            st += strow
        if apsw.complete(st):
            yield st
            st = ''

    if len(st) > 0 and not re.match(r'\s+$', st, re.DOTALL | re.UNICODE):
        if len(st) > 35:
            raise functions.OperatorError(
                __name__.rsplit('.')[-1],
                "Incomplete statement found : %s ... %s" % (st[:15], st[-15:]))
        else:
            raise functions.OperatorError(
                __name__.rsplit('.')[-1],
                "Incomplete statement found : %s" % (st, ))
示例#2
0
文件: flow.py 项目: HBPSP8Repo/madis
    def VTiter(self, *parsedArgs,**envars):
        largs, dictargs = self.full_parse(parsedArgs)

        if 'query' not in dictargs:
            raise functions.OperatorError(__name__.rsplit('.')[-1],"No query argument ")

        query=dictargs['query']
        connection=envars['db']
        
        yield (('query', 'text'),)
        cur=connection.cursor()
        execit=cur.execute(query, parse = False)

        st=''
        for row in execit:
            strow=filterlinecomment(' '.join(row))
            if strow=='':
                continue
            if st!='':
                st+='\n'+strow
            else:
                st+=strow
            if apsw.complete(st):
                yield [st]
                st=''
        if len(st)>0 and not re.match(r'\s+$', st, re.DOTALL| re.UNICODE):
            if len(st)>35:
                raise functions.OperatorError(__name__.rsplit('.')[-1],"Incomplete statement found : %s ... %s" %(st[:15],st[-15:]))
            else:
                raise functions.OperatorError(__name__.rsplit('.')[-1],"Incomplete statement found : %s" %(st))
def sqlstatement(iter):
    st=''
    for row in iter:
        strow=filterlinecomment(' '.join(row))
        if strow=='':
            continue
        if st!='':
            st+='\n'+strow
        else:
            st+=strow
        if apsw.complete(st):
            yield [st]
            st=''
    if len(st)>0 and not re.match(r'\s+$', st, re.DOTALL| re.UNICODE):
        if len(st)>35:
            raise functions.OperatorError(__name__.rsplit('.')[-1],"Incomplete statement found : %s ... %s" %(st[:15],st[-15:]))
        else:
            raise functions.OperatorError(__name__.rsplit('.')[-1],"Incomplete statement found : %s" %(st))
    return
示例#4
0
文件: exec.py 项目: HBPSP8Repo/madis
def breakquery(q):
    if len(q) > 1:
        raise functions.OperatorError(__name__.rsplit('.')[-1], "Ambiguous query column, result has more than one columns")
    st = ''
    for row in q[0].splitlines():
        strow = filterlinecomment(row)
        if strow == '':
            continue
        if st != '':
            st += '\n'+strow
        else:
            st += strow
        if apsw.complete(st):
            yield st
            st = ''

    if len(st) > 0 and not re.match(r'\s+$', st, re.DOTALL| re.UNICODE):
        if len(st) > 35:
            raise functions.OperatorError(__name__.rsplit('.')[-1], "Incomplete statement found : %s ... %s" % (st[:15], st[-15:]))
        else:
            raise functions.OperatorError(__name__.rsplit('.')[-1], "Incomplete statement found : %s" % (st,))
示例#5
0
    def VTiter(self, *parsedArgs, **envars):
        largs, dictargs = self.full_parse(parsedArgs)

        if 'query' not in dictargs:
            raise functions.OperatorError(
                __name__.rsplit('.')[-1], "No query argument ")

        query = dictargs['query']
        connection = envars['db']

        yield (('query', 'text'), )
        cur = connection.cursor()
        execit = cur.execute(query, parse=False)

        st = ''
        for row in execit:
            strow = filterlinecomment(' '.join(row))
            if strow == '':
                continue
            if st != '':
                st += '\n' + strow
            else:
                st += strow
            if apsw.complete(st):
                yield [st]
                st = ''
        if len(st) > 0 and not re.match(r'\s+$', st, re.DOTALL | re.UNICODE):
            if len(st) > 35:
                raise functions.OperatorError(
                    __name__.rsplit('.')[-1],
                    "Incomplete statement found : %s ... %s" %
                    (st[:15], st[-15:]))
            else:
                raise functions.OperatorError(
                    __name__.rsplit('.')[-1],
                    "Incomplete statement found : %s" % (st))
示例#6
0
    if flowname == None:
        parser.print_help()
        sys.exit(1)

    try:
        f = open(flowname, 'r')
    except Exception, e:
        exitwitherror("Error in opening SQL flow: " + str(e))

    statement = f.readline()
    if not statement:
        sys.exit()

    while True:
        while not apsw.complete(statement):
            line = f.readline()
            statement += line
            if not line:
                if statement.rstrip('\r\n') != '':
                    sys.stderr.write("Incomplete query:" + statement)
                f.close()
                sys.exit()
        try:
            for row in Connection.cursor().execute(statement):
                if len(row) > 1:
                    print(json.dumps(row, separators=(',', ':'), ensure_ascii=False).encode('utf_8', 'replace'))
                else:
                    print(unicode(row[0]).encode('utf_8', 'replace'))
            statement = ''
        except Exception, e:
示例#7
0
文件: mexec.py 项目: johnfouf/iis
    if flowname == None:
        parser.print_help()
        sys.exit(1)

    try:
        f = io.open(flowname, mode='r', encoding='utf-8')
    except Exception, e:
        exitwitherror("Error in opening SQL flow: " + str(e))


    statement = f.readline()
    if not statement:
        sys.exit()

    while True:
        while not apsw.complete(statement):
            line = f.readline()
            statement += line
            if not line:
                 if statement.rstrip('\r\n')!='':
                    sys.stderr.write("Incomplete query:"+statement)
                 f.close()
                 sys.exit()
        try :
            for row in Connection.cursor().execute(statement):
                if len(row) > 1:
                    print(json.dumps(row, separators=(',',':'), ensure_ascii=False).encode('utf_8', 'replace'))
                else:
                    print(unicode(row[0]).encode('utf_8', 'replace'))
            statement = ''
        except Exception, e:
示例#8
0
def main():
    desc = """Use this program to run madSQL queries on data coming from standard input. You may provide a database to run your queries. Results are streamed to standard output.
    """
    parser = OptionParser(description=desc,
                          usage="usage: %prog [options] [dbname] flowname",
                          version="%prog 1.0")
    parser.add_option("-f", "--flow", help="flow file to execute")
    parser.add_option("-d", "--db", help="db to connect")
    parser.add_option("-w",
                      "--dbw",
                      help="db to connect (open in create mode)")

    (options, args) = parser.parse_args()

    dbname = ''
    flags = apsw.SQLITE_OPEN_READWRITE | apsw.SQLITE_OPEN_CREATE
    try:
        dbname = args[0]
        flags = apsw.SQLITE_OPEN_READWRITE
    except:
        pass

    if options.db != None:
        dbname = options.db
        flags = apsw.SQLITE_OPEN_READWRITE

    if options.dbw != None:
        dbname = options.dbw

    try:
        Connection = madis.functions.Connection(dbname, flags)

    except Exception as e:
        exitwitherror("Error in opening DB: " + str(dbname) +
                      "\nThe error was: " + str(e))

    flowname = None
    try:
        flowname = args[1]
    except:
        pass

    if options.flow is not None:
        flowname = options.flow

    if flowname is None:
        parser.print_help()
        sys.exit(1)

    try:
        f = open(flowname, 'r')
    except Exception as e:
        exitwitherror("Error in opening SQL flow: " + str(e))

    statement = f.readline()
    if not statement:
        sys.exit()

    while True:
        while not apsw.complete(statement):
            line = f.readline()
            statement += line
            if not line:
                if statement.rstrip('\r\n') != '':
                    sys.stderr.write("Incomplete query:" + statement)
                f.close()
                sys.exit()
        try:
            for row in Connection.cursor().execute(statement):
                if len(row) > 1:
                    print(
                        json.dumps(row,
                                   separators=(',', ':'),
                                   ensure_ascii=False).encode(
                                       'utf_8', 'replace'))
                else:
                    print(str(row[0]).encode('utf_8', 'replace'))
            statement = ''
        except Exception as e:
            exitwitherror("Error when executing query: \n" + statement +
                          "\nThe error was: " + str(e))