mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-07-24 04:24:29 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

flake8 complaint

Browse source
This commit is contained in:
tomaae 2019-12-02 03:19:07 +01:00
parent bb0db7f64a
commit 3b5acc2393
5 changed files with 56 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

30
custom_components/mikrotik_router/device_tracker.py
View file

@ -2,13 +2,16 @@
import logging
from homeassistant.core import callback
from homeassistant.helpers import entity_registry
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.components.device_tracker import DOMAIN as DEVICE_TRACKER_DOMAIN
from homeassistant.components.device_tracker.config_entry import ScannerEntity
from homeassistant.components.device_tracker.const import SOURCE_TYPE_ROUTER
from .const import DEFAULT_NAME, DOMAIN, DATA_CLIENT, ATTRIBUTION
from .const import (
DOMAIN,
DATA_CLIENT,
ATTRIBUTION,
)
from homeassistant.const import (
CONF_NAME,
ATTR_ATTRIBUTION,
@ -16,8 +19,6 @@ from homeassistant.const import (
_LOGGER = logging.getLogger(__name__)
ATTRIBUTION = "Data provided by Mikrotik"
DEVICE_ATTRIBUTES = [
"running",
"enabled",
@ -34,6 +35,7 @@ DEVICE_ATTRIBUTES = [
"default-name",
]
#---------------------------
# async_setup_entry
#---------------------------
@ -41,16 +43,11 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
"""Set up device tracker for Mikrotik Router component."""
name = config_entry.data[CONF_NAME]
mikrotik_controller = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id]
tracked = {}
registry = await entity_registry.async_get_registry(hass)
@callback
def update_controller():
"""Update the values of the controller."""
update_items(name, mikrotik_controller, async_add_entities, tracked)
update_items(name, mikrotik_controller, async_add_entities)
mikrotik_controller.listeners.append(
async_dispatcher_connect(hass, mikrotik_controller.signal_update, update_controller)
@ -59,15 +56,16 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
update_controller()
return
#---------------------------
# update_items
#---------------------------
@callback
def update_items(name, mikrotik_controller, async_add_entities, tracked):
def update_items(name, mikrotik_controller, async_add_entities):
"""Update tracked device state from the controller."""
tracked = {}
new_tracked = []
sensors = []
for uid in mikrotik_controller.data['interface']:
if mikrotik_controller.data['interface'][uid]['type'] == "ether":
item_id = name + "-" + mikrotik_controller.data['interface'][uid]['default-name']
@ -84,6 +82,7 @@ def update_items(name, mikrotik_controller, async_add_entities, tracked):
return
#---------------------------
# MikrotikControllerPortDeviceTracker
#---------------------------
@ -114,7 +113,6 @@ class MikrotikControllerPortDeviceTracker(ScannerEntity):
"""Synchronize state with controller."""
#await self.mikrotik_controller.async_update()
return
@property
def is_connected(self):
@ -158,7 +156,7 @@ class MikrotikControllerPortDeviceTracker(ScannerEntity):
"connections": {(CONNECTION_NETWORK_MAC, self.mikrotik_controller.data['interface'][self._uid]['port-mac-address'])},
"manufacturer": self.mikrotik_controller.data['resource']['platform'],
"model": "Port",
"name": self.mikrotik_controller.data['interface'][self._uid]['default-name'] ,
"name": self.mikrotik_controller.data['interface'][self._uid]['default-name'],
}
return info
@ -171,4 +169,4 @@ class MikrotikControllerPortDeviceTracker(ScannerEntity):
if variable in self.mikrotik_controller.data['interface'][self._uid]:
attributes[variable] = self.mikrotik_controller.data['interface'][self._uid][variable]
return attributes
return attributes