chore: update wetransfer script from upstream

This commit is contained in:
Ahmed Zetao Yang
2021-03-31 09:05:16 +08:00
committed by borgmanJeremy
parent 71d595fbce
commit b361d34dec

View File

@@ -156,14 +156,17 @@ def _file_name_and_size(file: str) -> dict:
def _prepare_session() -> requests.Session: def _prepare_session() -> requests.Session:
"""Prepare a wetransfer.com session. """Prepare a wetransfer.com session.
Return a requests session that will always pass the initial X-CSRF-Token: Return a requests session that will always pass the required headers
and with cookies properly populated that can be used for wetransfer and with cookies properly populated that can be used for wetransfer
requests. requests.
""" """
s = requests.Session() s = requests.Session()
r = s.get('https://wetransfer.com/') r = s.get('https://wetransfer.com/')
m = re.search('name="csrf-token" content="([^"]+)"', r.text) m = re.search('name="csrf-token" content="([^"]+)"', r.text)
s.headers.update({'X-CSRF-Token': m.group(1)}) s.headers.update({
'x-csrf-token': m.group(1),
'x-requested-with': 'XMLHttpRequest',
})
return s return s