jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 15:17:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Downloader now supports downloading to subdirs

Browse Source
This commit is contained in:
Paulus Schoutsen 2014-03-24 18:00:57 -07:00
parent 82357b421f
commit 49e6386794
1 changed files with 32 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
homeassistant/components/downloader.py
View File

@ -15,6 +15,9 @@ DOMAIN = "downloader"
SERVICE_DOWNLOAD_FILE = "download_file" SERVICE_DOWNLOAD_FILE = "download_file"
ATTR_URL = "url"
ATTR_SUBDIR = "subdir"
# pylint: disable=too-many-branches # pylint: disable=too-many-branches
def setup(bus, download_path): def setup(bus, download_path):
@ -41,14 +44,22 @@ def setup(bus, download_path):
def download_file(service): def download_file(service):
""" Starts thread to download file specified in the url. """ """ Starts thread to download file specified in the url. """
if not 'url' in service.data: if not ATTR_URL in service.data:
logger.error("Service called but 'url' parameter not specified.") logger.error("Service called but 'url' parameter not specified.")
return return
def do_download(): def do_download():
""" Downloads the file. """ """ Downloads the file. """
try: try:
url = service.data['url'] url = service.data[ATTR_URL]
subdir = service.data.get(ATTR_SUBDIR)
if subdir:
subdir = util.sanitize_filename(subdir)
final_path = None
req = requests.get(url, stream=True) req = requests.get(url, stream=True)
if req.status_code == 200: if req.status_code == 200:
@ -71,8 +82,20 @@ def setup(bus, download_path):
# Remove stuff to ruin paths # Remove stuff to ruin paths
filename = util.sanitize_filename(filename) filename = util.sanitize_filename(filename)
path, ext = os.path.splitext(os.path.join(download_path, # Do we want to download to subdir, create if needed
filename)) if subdir:
subdir_path = os.path.join(download_path, subdir)
# Ensure subdir exist
if not os.path.isdir(subdir_path):
os.makedirs(subdir_path)
final_path = os.path.join(subdir_path, filename)
else:
final_path = os.path.join(download_path, filename)
path, ext = os.path.splitext(final_path)
# If file exist append a number. # If file exist append a number.
# We test filename, filename_2.. # We test filename, filename_2..
@ -81,7 +104,7 @@ def setup(bus, download_path):
while os.path.isfile(final_path): while os.path.isfile(final_path):
tries += 1 tries += 1
final_path = path + "_{}".format(tries) + ext final_path = "{}_{}.{}".format(path, tries, ext)
logger.info("{} -> {}".format( logger.info("{} -> {}".format(
url, final_path)) url, final_path))
@ -97,6 +120,10 @@ def setup(bus, download_path):
logger.exception("ConnectionError occured for {}". logger.exception("ConnectionError occured for {}".
format(url)) format(url))
# Remove file if we started downloading but failed
if final_path and os.path.isfile(final_path):
os.remove(final_path)
threading.Thread(target=do_download).start() threading.Thread(target=do_download).start()
bus.register_service(DOMAIN, SERVICE_DOWNLOAD_FILE, bus.register_service(DOMAIN, SERVICE_DOWNLOAD_FILE,