Make HassIO more objectable

hassio_api/hassio/__main__.py

@@ -16,8 +16,10 @@ if __name__ == "__main__":
         exit(1)
 
     loop = asyncio.get_event_loop()
+    hassio = core.HassIO(loop)
+
     _LOGGER.info("Start Hassio task")
-    loop.create_task(core.run_hassio(loop))
+    loop.create_task(hassio.start())
 
     loop.run_forever()
     _LOGGER.info("Close Hassio")

hassio_api/hassio/core.py

@@ -14,53 +14,58 @@ from .docker.supervisor import DockerSupervisor
 _LOGGER = logging.getLogger(__name__)
 
 
-async def run_hassio(loop):
+class HassIO(object):
-    """Start HassIO."""
+    """Main object of hassio."""
-    websession = aiohttp.ClientSession(loop=loop)
+
-    dock = docker.DockerClient(
+    def __init__(self, loop):
+        """Initialize hassio object."""
+        self.loop = loop
+        self.config = bootstrap.initialize_system_data()
+        self.websession = aiohttp.ClientSession(loop=self.loop)
+        self.dock = docker.DockerClient(
             base_url="unix:/{}".format(SOCKET_DOCKER), version='auto')
 
-    # init system
+        # init basic docker container
-    config = bootstrap.initialize_system_data()
+        self.supervisor = DockerSupervisor(
-
+            self.config, self.loop, self.dock)
-    # init Supervisor Docker
+        self.homeassistant = DockerHomeAssistant(
-    docker_super = DockerSupervisor(config, loop, dock)
+            self.config, self.loop, self.dock)
-    await docker_super.attach()
-    _LOGGER.info(
-        "Attach to supervisor image %s tag %s", docker_super.image,
-        docker_super.tag)
-
-    # init HomeAssistant Docker
-    docker_hass = DockerHomeAssistant(
-        config, loop, dock, image=config.homeassistant_image,
-        tag=config.homeassistant_tag
-    )
 
         # init HostControll
-    host_controll = HostControll(loop)
+        self.host_controll = HostControll(self.loop)
-    host_info = await host_controll.info()
+
+    async def start(self):
+        """Start HassIO."""
+        await self.supervisor.attach()
+        _LOGGER.info(
+            "Attach to supervisor image %s tag %s", self.supervisor.image,
+            self.supervisor.tag)
+
+        host_info = await self.host_controll.info()
         if host_info:
             _LOGGER.info(
                 "Connected to host controll daemon. OS: %s Version: %s",
                 host_info.get('host'), host_info.get('version'))
 
         # first start of supervisor?
-    if config.homeassistant_tag is None:
+        if self.config.homeassistant_tag is None:
             _LOGGER.info("First start of supervisor, read version from github.")
 
             # read homeassistant tag and install it
             current = None
             while True:
-            current = await tools.fetch_current_versions(websession)
+                current = await tools.fetch_current_versions(self.websession)
                 if current and HOMEASSISTANT_TAG in current:
-                if await docker_hass.install(current[HOMEASSISTANT_TAG]):
+                    resp = await self.homeassistant.install(
+                        current[HOMEASSISTANT_TAG]):
+                    if resp:
                         break
                 _LOGGER.warning("Can't fetch info from github. Retry in 60.")
                 await asyncio.sleep(60, loop=loop)
 
-        config.homeassistant_tag = current[HOMEASSISTANT_TAG]
+            self.config.homeassistant_tag = current[HOMEASSISTANT_TAG]
         else:
             _LOGGER.info("HomeAssistant docker is exists.")
 
         # run HomeAssistant
-    await docker_hass.run()
+        await self.homeassistant.run()

hassio_api/hassio/docker/homeassistant.py

@@ -13,6 +13,13 @@ HASS_DOCKER_NAME = 'homeassistant'
 class DockerHomeAssistant(DockerBase):
     """Docker hassio wrapper for HomeAssistant."""
 
+    def __init__(self, config, loop, dock):
+        """Initialize docker homeassistant wrapper."""
+        super().__init__(
+            config, loop, dock, image=config.homeassistant_image,
+            tag=config.homeassistant_tag
+        )
+
     @property
     def docker_name(self):
         """Return name of docker container."""