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

Changes after review

Browse Source
This commit is contained in:
Steven Looman 2018-09-03 21:16:45 +02:00
parent 85c40d29ee
commit f4a54e2483
1 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
homeassistant/components/igd/__init__.py
View File

@ -51,7 +51,7 @@ CONFIG_SCHEMA = vol.Schema({
async def _async_create_igd_device(hass: HomeAssistantType,
ssdp_description: str):
"""."""
"""Create IGD device."""
# build requester
from async_upnp_client.aiohttp import AiohttpSessionRequester
session = async_get_clientsession(hass)
@ -94,30 +94,36 @@ async def _async_add_port_mapping(hass: HomeAssistantType,
if local_ip == '127.0.0.1':
_LOGGER.warning('Could not create port mapping, our IP is 127.0.0.1')
return False
return
local_ip = IPv4Address(local_ip)
# create port mapping
from async_upnp_client import UpnpError
port = hass.http.server_port
_LOGGER.debug('Creating port mapping %s:%s:%s (TCP)', port, local_ip, port)
await igd_device.async_add_port_mapping(remote_host=None,
external_port=port,
protocol='TCP',
internal_port=port,
internal_client=local_ip,
enabled=True,
description="Home Assistant",
lease_duration=None)
return True
try:
await igd_device.async_add_port_mapping(remote_host=None,
external_port=port,
protocol='TCP',
internal_port=port,
internal_client=local_ip,
enabled=True,
description="Home Assistant",
lease_duration=None)
except (asyncio.TimeoutError, aiohttp.ClientError, UpnpError):
_LOGGER.warning('Could not add port mapping')
async def _async_delete_port_mapping(hass: HomeAssistantType, igd_device):
"""Remove a port mapping."""
from async_upnp_client import UpnpError
port = hass.http.server_port
await igd_device.async_delete_port_mapping(remote_host=None,
external_port=port,
protocol='TCP')
try:
await igd_device.async_delete_port_mapping(remote_host=None,
external_port=port,
protocol='TCP')
except (asyncio.TimeoutError, aiohttp.ClientError, UpnpError):
_LOGGER.warning('Could not delete port mapping')
# config