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

added comments for accounting functions

Browse source
This commit is contained in:
tomaae 2020-04-08 17:52:58 +02:00
parent 3b9e14b371
commit f059149f59
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
custom_components/mikrotik_router/mikrotik_controller.py
View file

@ -831,6 +831,9 @@ class MikrotikControllerData:
if self.data["host"][uid]["available"]:
self.data["host"][uid]["last-seen"] = utcnow()
# ---------------------------
# _address_part_of_local_network
# ---------------------------
def _address_part_of_local_network(self, address):
address = ip_address(address)
for vals in self.data["dhcp-network"].values():
@ -838,12 +841,18 @@ class MikrotikControllerData:
return True
return False
# ---------------------------
# _get_accounting_uid_by_ip
# ---------------------------
def _get_accounting_uid_by_ip(self, requested_ip):
for mac, vals in self.data['accounting'].items():
if vals.get('address') is requested_ip:
return mac
return None
# ---------------------------
# get_accounting
# ---------------------------
def get_accounting(self):
"""Get Accounting data from Mikrotik"""
# Check if accounting and account-local-traffic is enabled
@ -893,7 +902,8 @@ class MikrotikControllerData:
destination_ip = str(item.get('dst-address')).strip()
bits_count = int(str(item.get('bytes')).strip()) * 8
if self._address_part_of_local_network(source_ip) and self._address_part_of_local_network(destination_ip):
if self._address_part_of_local_network(source_ip) and \
self._address_part_of_local_network(destination_ip):
# LAN TX/RX
if source_ip in tmp_accounting_values:
tmp_accounting_values[source_ip]['lan-tx'] += bits_count