Skip to content

Commit

Permalink
style: black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
jorwoods committed Dec 10, 2023
1 parent f42948a commit 76559d4
Show file tree
Hide file tree
Showing 4 changed files with 14 additions and 24 deletions.
11 changes: 7 additions & 4 deletions tableauserverclient/helpers/headers.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,18 +2,21 @@
from typing import Any, Generic, Mapping, Optional, TypeVar, Union
from urllib.parse import unquote_plus

T = TypeVar("T", )
T = TypeVar(
"T",
)


def fix_filename(params: Mapping[str, T]) -> Mapping[str, T]:
if "filename*" not in params:
return params

params = deepcopy(params)
filename = params["filename*"]
prefix = "UTF-8''"
if filename.startswith(prefix):
filename = filename[len(prefix):]
filename = filename[len(prefix) :]

params["filename"] = unquote_plus(filename)
del params["filename*"]
return params
return params
9 changes: 3 additions & 6 deletions test/test_datasource.py
Original file line number Diff line number Diff line change
Expand Up @@ -703,10 +703,7 @@ def test_bad_download_response(self) -> None:
self.baseurl + "/9dbd2263-16b5-46e1-9c43-a76bb8ab65fb/content",
headers={
"Content-Disposition": '''name="tableau_datasource"; filename*=UTF-8''"Sample datasource.tds"'''
}
)
file_path = self.server.datasources.download(
"9dbd2263-16b5-46e1-9c43-a76bb8ab65fb",
td
},
)
self.assertTrue(os.path.exists(file_path))
file_path = self.server.datasources.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", td)
self.assertTrue(os.path.exists(file_path))
9 changes: 2 additions & 7 deletions test/test_flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,12 +209,7 @@ def test_bad_download_response(self) -> None:
with requests_mock.mock() as m, tempfile.TemporaryDirectory() as td:
m.get(
self.baseurl + "/9dbd2263-16b5-46e1-9c43-a76bb8ab65fb/content",
headers={
"Content-Disposition": '''name="tableau_flow"; filename*=UTF-8''"Sample flow.tfl"'''
}
)
file_path = self.server.flows.download(
"9dbd2263-16b5-46e1-9c43-a76bb8ab65fb",
td
headers={"Content-Disposition": '''name="tableau_flow"; filename*=UTF-8''"Sample flow.tfl"'''},
)
file_path = self.server.flows.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", td)
self.assertTrue(os.path.exists(file_path))
9 changes: 2 additions & 7 deletions test/test_workbook.py
Original file line number Diff line number Diff line change
Expand Up @@ -937,12 +937,7 @@ def test_bad_download_response(self) -> None:
with requests_mock.mock() as m, tempfile.TemporaryDirectory() as td:
m.get(
self.baseurl + "/9dbd2263-16b5-46e1-9c43-a76bb8ab65fb/content",
headers={
"Content-Disposition": '''name="tableau_workbook"; filename*=UTF-8''"Sample workbook.twb"'''
}
)
file_path = self.server.workbooks.download(
"9dbd2263-16b5-46e1-9c43-a76bb8ab65fb",
td
headers={"Content-Disposition": '''name="tableau_workbook"; filename*=UTF-8''"Sample workbook.twb"'''},
)
file_path = self.server.workbooks.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", td)
self.assertTrue(os.path.exists(file_path))

0 comments on commit 76559d4

Please sign in to comment.