From 002bf9b180e95d56b6bb99f2eae2ee0035debe08 Mon Sep 17 00:00:00 2001 From: Lavish Saluja Date: Sat, 25 Jan 2020 06:23:34 +0530 Subject: [PATCH] Used altchars parameter in decode function There is a special field for alternate characters to '+' and '\' in b64 decode function. [https://www.base64encode.net/python-base64-b64encode](Link to the documentation) --- websiteScreenshot.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/websiteScreenshot.py b/websiteScreenshot.py index 7997cac..4e4d717 100644 --- a/websiteScreenshot.py +++ b/websiteScreenshot.py @@ -28,12 +28,9 @@ print("Invalid JSON encountered.") sys.exit() -# Google has a weird way of encoding the Base64 data -screenshot_encoded = screenshot_encoded.replace("_", "/") -screenshot_encoded = screenshot_encoded.replace("-", "+") # Decode the Base64 data -screenshot_decoded = base64.b64decode(screenshot_encoded) +screenshot_decoded = base64.b64decode(screenshot_encoded, altchars= '-_', validate=False) # Save the file