[PECO-2051] Add custom auth headers into cloud fetch request (databricks#460)

jackyhu-db · web-flow · commit d3cb62c4c4c0 · 2024-10-25T09:41:40.000-07:00
Signed-off-by: Jacky Hu &lt;jacky.hu@databricks.com&gt;
diff --git a/src/databricks/sql/cloudfetch/downloader.py b/src/databricks/sql/cloudfetch/downloader.py
@@ -100,6 +100,7 @@ def run(self) -> DownloadedFile:
                 self.link.fileLink,
                 timeout=self.settings.download_timeout,
                 verify=self._ssl_options.tls_verify,
+                headers=self.link.httpHeaders
                 # TODO: Pass cert from `self._ssl_options`
             )
             response.raise_for_status()

Original file line number	Diff line number	Diff line change
`@@ -100,6 +100,7 @@ def run(self) -> DownloadedFile:`
`100`	`100`	`self.link.fileLink,`
`101`	`101`	`timeout=self.settings.download_timeout,`
`102`	`102`	`verify=self._ssl_options.tls_verify,`
	`103`	`+ headers=self.link.httpHeaders`
`103`	`104`	# TODO: Pass cert from `self._ssl_options`
`104`	`105`	`)`
`105`	`106`	`response.raise_for_status()`