/lib/sql.py - Diff - BIEN 3 - NCEAS Projects

« Previous | Next »

Revision 3010

sql.py: mk_insert_select(): INSERT IGNORE: Implemented simple cursor loop that just inserts each row, without ignoring duplicate keys

         '''
         table = sql_gen.remove_table_rename(sql_gen.as_Table(table))
         if cols == []: cols = None # no cols (all defaults) = unknown col names
         if cols != None:
             cols = [sql_gen.to_name_only_col(v, table).to_str(db) for v in cols]
         if cols != None: cols = [sql_gen.to_name_only_col(c, table) for c in cols]
         if select_query == None: select_query = 'DEFAULT VALUES'
         if returning != None: returning = sql_gen.as_Col(returning, table)
-...
         def mk_insert(select_query):
             query = first_line
             if cols != None: query += '\n('+', '.join(cols)+')'
             if cols != None:
                 query += '\n('+(', '.join((c.to_str(db) for c in cols)))+')'
             query += '\n'+select_query
             if returning != None:
-...
             embeddable = True # must use function
             lang = 'plpgsql'
             assert cols != None
             row_cols = [sql_gen.Col(c.name, 'row') for c in cols]
             row_query = mk_insert(sql_gen.Values(row_cols).to_str(db))
             query = '''\
     DECLARE
         row record;
         cur CURSOR FOR
     '''+select_query+'''
+    ;
     BEGIN
     END;
         OPEN cur;
         FOUND := true; -- for initial check
         WHILE FOUND LOOP
             FETCH FROM cur INTO row;
     '''+row_query+'''
+    ;
         END LOOP;
         CLOSE cur;
     END;\
     '''
         else: query = mk_insert(select_query)

Also available in: Unified diff