SNOW-26262: split_statement returns a single SQL statement and a flag indicating PUT or GET command

Author: smtakeda

connection.py

@@ -340,12 +340,12 @@ def execute_string(self, sql_text,
                 sql_text, str) else sql_text)
         else:
             stream = StringIO(sql_text)
-        for sql in split_statements(
+        for sql, is_put_or_get in split_statements(
                 stream, remove_comments=remove_comments):
             cur = self.cursor()
             if return_cursors:
                 ret.append(cur)
-            cur.execute(sql)
+            cur.execute(sql, _is_put_get=is_put_or_get)
         return ret
 
     def execute_stream(self, stream,
@@ -354,10 +354,10 @@ def execute_stream(self, stream,
         Executes a stream of SQL statements.
         This is a non-standard convenient method.
         """
-        for sql in split_statements(
+        for sql, is_put_or_get in split_statements(
                 stream, remove_comments=remove_comments):
             cur = self.cursor()
-            cur.execute(sql)
+            cur.execute(sql, _is_put_get=is_put_or_get)
             yield cur
 
     def __set_error_attributes(self):

cursor.py

@@ -75,6 +75,7 @@ def __init__(self, connection):
         self._total_rowcount = -1
         self._sequence_counter = -1
         self._request_id = None
+        self._is_file_transfer = False
 
         self._timestamp_output_format = None
         self._timestamp_ltz_output_format = None
@@ -245,7 +246,7 @@ def is_file_transfer(self):
         """
         Is PUT or GET command?
         """
-        return self._is_file_transfer is not None
+        return hasattr(self, '_is_file_transfer') and self._is_file_transfer
 
     def callproc(self, procname, args=()):
         u"""
@@ -458,8 +459,8 @@ def execute(self, command, params=None, timeout=None,
                 self._connection._role = data[u'finalRoleName']
 
             # self.logger.debug(ret)
-            self.logger.debug(u"PUT OR GET: %s", self._is_file_transfer)
-            if self._is_file_transfer:
+            self.logger.debug(u"PUT OR GET: %s", self.is_file_transfer)
+            if self.is_file_transfer:
                 sf_file_transfer_agent = SnowflakeFileTransferAgent(
                     self, query, ret,
                     put_callback=_put_callback,