mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-08 02:19:35 +00:00
04514a9f5c
* Create hassio network layer / allow linking * rename docker * fix lint * fix lint p2 * Set network options * First version of network code * Finish network layer * Remove old api_endpoint stuff * Add DNS forwarding * Fix DNS recorder * Fix lint p1 * Fix lint p2 * Fix lint p3 * Fix spell * Fix ipam struct * Fix ip to str * Fix ip to str v2 * Fix spell * Fix hass on host * Fix host attach to network * Cleanup network code * Fix lint & add debug * fix link * Remove log * Fix network * fix reattach of supervisor * set options * Fix containers * Fix remapping & add a test * Fix dict bug * Fix prop * Test with run container * Fix problem
74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
"""Internal network manager for HassIO."""
|
|
import logging
|
|
|
|
import docker
|
|
|
|
from ..const import DOCKER_NETWORK_MASK, DOCKER_NETWORK, DOCKER_NETWORK_RANGE
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class DockerNetwork(object):
|
|
"""Internal HassIO Network."""
|
|
|
|
def __init__(self, dock):
|
|
"""Initialize internal hassio network."""
|
|
self.docker = dock
|
|
self.network = self._get_network()
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return name of network."""
|
|
return DOCKER_NETWORK
|
|
|
|
@property
|
|
def containers(self):
|
|
"""Return of connected containers from network."""
|
|
return self.network.containers
|
|
|
|
@property
|
|
def gateway(self):
|
|
"""Return gateway of the network."""
|
|
return DOCKER_NETWORK_MASK[1]
|
|
|
|
@property
|
|
def supervisor(self):
|
|
"""Return supervisor of the network."""
|
|
return DOCKER_NETWORK_MASK[2]
|
|
|
|
def _get_network(self):
|
|
"""Get HassIO network."""
|
|
try:
|
|
return self.docker.networks.get(DOCKER_NETWORK)
|
|
except docker.errors.NotFound:
|
|
_LOGGER.info("Can't find HassIO network, create new network")
|
|
|
|
ipam_pool = docker.types.IPAMPool(
|
|
subnet=str(DOCKER_NETWORK_MASK),
|
|
gateway=str(self.gateway),
|
|
iprange=str(DOCKER_NETWORK_RANGE)
|
|
)
|
|
|
|
ipam_config = docker.types.IPAMConfig(pool_configs=[ipam_pool])
|
|
|
|
return self.docker.networks.create(
|
|
DOCKER_NETWORK, driver='bridge', ipam=ipam_config, options={
|
|
"com.docker.network.bridge.name": DOCKER_NETWORK,
|
|
})
|
|
|
|
def attach_container(self, container, alias=None, ipv4=None):
|
|
"""Attach container to hassio network.
|
|
|
|
Need run inside executor.
|
|
"""
|
|
ipv4 = str(ipv4) if ipv4 else ""
|
|
|
|
try:
|
|
self.network.connect(container, aliases=alias, ipv4_address=ipv4)
|
|
except docker.errors.APIError as err:
|
|
_LOGGER.error("Can't link container to hassio-net -> %s", err)
|
|
return False
|
|
|
|
self.network.reload()
|
|
return True
|