diff --git a/memories-downloader.py b/memories-downloader.py index cfec814..ae8c7b6 100644 --- a/memories-downloader.py +++ b/memories-downloader.py @@ -7,6 +7,7 @@ import json import requests import datetime import os +import shutil # Helper Functions @@ -22,6 +23,13 @@ def get_ext(memory_media_type): ext = "jpg" return ext +def download_file_from_url(url, filename): + with requests.get(url, stream=True) as r: + with open(filename, 'wb') as f: + shutil.copyfileobj(r.raw, f) + + return filename + overwrite = False # Overwrites existing useful memory metadata file # Import Memories Data @@ -77,7 +85,7 @@ for memory in memories: print("Downloading File {}:".format(filename), end=" ") try: - urllib.request.urlretrieve(memory["url"], filename) + download_file_from_url(memory["url"], filename) modtime = datetime.datetime.timestamp(memory_datetime) os.utime(filename, (modtime, modtime)) print("Success")