mirror of
https://github.com/home-assistant/core.git
synced 2025-07-07 21:37:07 +00:00
Service validation for downloader component.
This commit is contained in:
parent
652fe7e0f2
commit
003bd24976
@ -10,8 +10,10 @@ import re
|
||||
import threading
|
||||
|
||||
import requests
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.helpers import validate_config
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
from homeassistant.util import sanitize_filename
|
||||
|
||||
DOMAIN = "downloader"
|
||||
@ -21,6 +23,11 @@ SERVICE_DOWNLOAD_FILE = "download_file"
|
||||
ATTR_URL = "url"
|
||||
ATTR_SUBDIR = "subdir"
|
||||
|
||||
SERVICE_DOWNLOAD_FILE_SCHEMA = vol.Schema({
|
||||
vol.Required(ATTR_URL): vol.Url,
|
||||
vol.Optional(ATTR_SUBDIR): cv.string,
|
||||
})
|
||||
|
||||
CONF_DOWNLOAD_DIR = 'download_dir'
|
||||
|
||||
|
||||
@ -48,10 +55,6 @@ def setup(hass, config):
|
||||
|
||||
def download_file(service):
|
||||
"""Start thread to download file specified in the URL."""
|
||||
if ATTR_URL not in service.data:
|
||||
logger.error("Service called but 'url' parameter not specified.")
|
||||
return
|
||||
|
||||
def do_download():
|
||||
"""Download the file."""
|
||||
try:
|
||||
@ -127,7 +130,7 @@ def setup(hass, config):
|
||||
|
||||
threading.Thread(target=do_download).start()
|
||||
|
||||
hass.services.register(DOMAIN, SERVICE_DOWNLOAD_FILE,
|
||||
download_file)
|
||||
hass.services.register(DOMAIN, SERVICE_DOWNLOAD_FILE, download_file,
|
||||
schema=SERVICE_DOWNLOAD_FILE_SCHEMA)
|
||||
|
||||
return True
|
||||
|
Loading…
x
Reference in New Issue
Block a user