mirror of
https://github.com/KevinMidboe/escape-snapchat.git
synced 2025-10-29 09:30:21 +00:00
Replaced urllib.request.urlretrieve w/ streaming contents to a local file using shuttil
This commit is contained in:
@@ -7,6 +7,7 @@ import json
|
|||||||
import requests
|
import requests
|
||||||
import datetime
|
import datetime
|
||||||
import os
|
import os
|
||||||
|
import shutil
|
||||||
|
|
||||||
# Helper Functions
|
# Helper Functions
|
||||||
|
|
||||||
@@ -22,6 +23,13 @@ def get_ext(memory_media_type):
|
|||||||
ext = "jpg"
|
ext = "jpg"
|
||||||
return ext
|
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
|
overwrite = False # Overwrites existing useful memory metadata file
|
||||||
|
|
||||||
# Import Memories Data
|
# Import Memories Data
|
||||||
@@ -77,7 +85,7 @@ for memory in memories:
|
|||||||
|
|
||||||
print("Downloading File {}:".format(filename), end=" ")
|
print("Downloading File {}:".format(filename), end=" ")
|
||||||
try:
|
try:
|
||||||
urllib.request.urlretrieve(memory["url"], filename)
|
download_file_from_url(memory["url"], filename)
|
||||||
modtime = datetime.datetime.timestamp(memory_datetime)
|
modtime = datetime.datetime.timestamp(memory_datetime)
|
||||||
os.utime(filename, (modtime, modtime))
|
os.utime(filename, (modtime, modtime))
|
||||||
print("Success")
|
print("Success")
|
||||||
|
|||||||
Reference in New Issue
Block a user