Merge branch 'master' of https://github.com/sunyinqi0508/AQuery2

reconstruct/ast.py

@@ -296,7 +296,7 @@ class projection(ast_node):
         if 'outfile' in node:
             self.outfile = outfile(self, node['outfile'], sql = self.sql)
             if not self.has_postproc:
-                self.sql += self.outfile.sql
+                self.sql = self.outfile.sql
         else:
             self.outfile = None
         
@@ -398,10 +398,10 @@ class projection(ast_node):
             else:
                 self.context.emitc(f'{self.out_table.contextname_cpp}->printall(" ","\\n", nullptr, nullptr, {self.limit});')
         
-        if self.outfile:
-            self.outfile.finalize()
+        if self.outfile and self.has_postproc:
+                self.outfile.finalize()
 
-        if 'into' in node:
+        if 'into' in node: 
             self.context.emitc(select_into(self, node['into']).ccode)
             self.has_postproc = True
         if not self.distinct:
@@ -417,7 +417,7 @@ class projection(ast_node):
                 self.context.postproc_end(self.postproc_fname)
             else:
                 self.context.ccode = ''
-                if self.limit != 0:
+                if self.limit != 0 and not self.outfile:
                     self.context.direct_output()
         
 class select_distinct(projection):
@@ -1186,8 +1186,8 @@ class outfile(ast_node):
     name="_outfile"
     def __init__(self, parent, node, context = None, *, sql = None):
         self.node = node
-        super().__init__(parent, node, context)
         self.sql = sql if sql else ''
+        super().__init__(parent, node, context)
         
     def init(self, _):
         assert(isinstance(self.parent, projection))
@@ -1207,10 +1207,10 @@ class outfile(ast_node):
     def produce_monetdb(self, node):
         filename = node['loc']['literal'] if 'loc' in node else node['literal']
         import os
-        p = os.path.abspath('.').replace('\\', '/') + '/' + filename
-        self.sql = f'COPY {self.sql} INTO "{p}"'
-        d = '\t'
-        e = '\n'
+        p =  os.path.abspath('.').replace('\\', '/') + '/' + filename
+        self.sql = f'COPY {self.parent.sql} INTO \'{p}\''
+        d = ','
+        e = '\\n'
         if 'term' in node:
             d = node['term']['literal']
         self.sql += f' delimiters \'{d}\', \'{e}\''

reconstruct/storage.py

@@ -253,6 +253,10 @@ class Context:
     def direct_output(self):
         self.queries.append('O')
     
+    def abandon_postproc(self):
+        self.ccode = ''
+        self.finalize_query()
+    
     def finalize_udf(self):
         if self.udf is not None:
             return (Context.udf_head