From 56b08a6ddb491c2fb673129693f8be2ea4eb60a9 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Sat, 13 Apr 2019 16:50:19 -0600 Subject: [PATCH] Ensure OpenUV service checks permissions (#22668) * Create decorator to check service permissions * Ensure OpenUV service has proper user permissions * Reverting strange changes --- homeassistant/components/openuv/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/homeassistant/components/openuv/__init__.py b/homeassistant/components/openuv/__init__.py index 8e8401bbeac..63d2744cd4d 100644 --- a/homeassistant/components/openuv/__init__.py +++ b/homeassistant/components/openuv/__init__.py @@ -11,6 +11,7 @@ from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.helpers import aiohttp_client, config_validation as cv from homeassistant.helpers.dispatcher import async_dispatcher_send from homeassistant.helpers.entity import Entity +from homeassistant.helpers.service import verify_domain_control from .config_flow import configured_instances from .const import DOMAIN @@ -130,6 +131,8 @@ async def async_setup_entry(hass, config_entry): from pyopenuv import Client from pyopenuv.errors import OpenUvError + _verify_domain_control = verify_domain_control(hass, DOMAIN) + try: websession = aiohttp_client.async_get_clientsession(hass) openuv = OpenUV( @@ -155,6 +158,7 @@ async def async_setup_entry(hass, config_entry): hass.config_entries.async_forward_entry_setup( config_entry, component)) + @_verify_domain_control async def update_data(service): """Refresh OpenUV data.""" _LOGGER.debug('Refreshing OpenUV data')