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

changed librouteros library name to avoid conflict with loaded library #2

Browse source
This commit is contained in:
tomaae 2019-12-08 18:10:09 +01:00
parent 10898768e3
commit 0896d0c991
8 changed files with 42 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

10
custom_components/mikrotik_router/librouteros/__init__.py → custom_components/mikrotik_router/librouteros_custom/__init__.py
View file

@ -3,17 +3,17 @@
from socket import create_connection
from collections import ChainMap
from librouteros.exceptions import (
from .exceptions import (
ConnectionClosed,
FatalError,
)
from librouteros.connections import SocketTransport
from librouteros.protocol import ApiProtocol
from librouteros.login import (
from .connections import SocketTransport
from .protocol import ApiProtocol
from .login import (
plain,
token,
)
from librouteros.api import Api
from .api import Api
DEFAULTS = {
'timeout': 10,

6
custom_components/mikrotik_router/librouteros/api.py → custom_components/mikrotik_router/librouteros_custom/api.py
View file

@ -2,12 +2,12 @@
from posixpath import join as pjoin
from librouteros.exceptions import TrapError, MultiTrapError
from librouteros.protocol import (
from .exceptions import TrapError, MultiTrapError
from .protocol import (
compose_word,
parse_word,
)
from librouteros.query import Query
from .query import Query
class Api:

2
custom_components/mikrotik_router/librouteros/connections.py → custom_components/mikrotik_router/librouteros_custom/connections.py
View file

@ -1,6 +1,6 @@
# -*- coding: UTF-8 -*-
from librouteros.exceptions import ConnectionClosed
from .exceptions import ConnectionClosed
class SocketTransport:

0
custom_components/mikrotik_router/librouteros/exceptions.py → custom_components/mikrotik_router/librouteros_custom/exceptions.py
View file

0
custom_components/mikrotik_router/librouteros/login.py → custom_components/mikrotik_router/librouteros_custom/login.py
View file

2
custom_components/mikrotik_router/librouteros/protocol.py → custom_components/mikrotik_router/librouteros_custom/protocol.py
View file

@ -3,7 +3,7 @@
from struct import pack, unpack
from logging import getLogger, NullHandler
from librouteros.exceptions import (
from .exceptions import (
ProtocolError,
FatalError,
)

2
custom_components/mikrotik_router/librouteros/query.py → custom_components/mikrotik_router/librouteros_custom/query.py
View file

@ -1,6 +1,6 @@
# -*- coding: UTF-8 -*-
from itertools import chain
from librouteros.protocol import (
from .protocol import (
cast_to_api,
)

62
custom_components/mikrotik_router/mikrotikapi.py
View file

@ -11,12 +11,12 @@ from .const import (
DEFAULT_ENCODING,
)
MODULE_PATH = os.path.join(os.path.dirname(__file__), "librouteros", "__init__.py")
MODULE_NAME = "librouteros"
MODULE_PATH = os.path.join(os.path.dirname(__file__), "librouteros_custom", "__init__.py")
MODULE_NAME = "librouteros_custom"
spec = importlib.util.spec_from_file_location(MODULE_NAME, MODULE_PATH)
librouteros = importlib.util.module_from_spec(spec)
sys.modules[spec.name] = librouteros
spec.loader.exec_module(librouteros)
librouteros_custom = importlib.util.module_from_spec(spec)
sys.modules[spec.name] = librouteros_custom
spec.loader.exec_module(librouteros_custom)
_LOGGER = logging.getLogger(__name__)
@ -69,13 +69,13 @@ class MikrotikAPI:
kwargs["ssl_wrapper"] = self._ssl_wrapper
try:
self._connection = librouteros.connect(self._host, self._username, self._password, **kwargs)
self._connection = librouteros_custom.connect(self._host, self._username, self._password, **kwargs)
except (
librouteros.exceptions.TrapError,
librouteros.exceptions.MultiTrapError,
librouteros.exceptions.ConnectionClosed,
librouteros.exceptions.ProtocolError,
librouteros.exceptions.FatalError,
librouteros_custom.exceptions.TrapError,
librouteros_custom.exceptions.MultiTrapError,
librouteros_custom.exceptions.ConnectionClosed,
librouteros_custom.exceptions.ProtocolError,
librouteros_custom.exceptions.FatalError,
ssl.SSLError
) as api_error:
_LOGGER.error("Mikrotik %s: %s", self._host, api_error)
@ -121,16 +121,16 @@ class MikrotikAPI:
try:
response = self._connection.path(path)
tuple(response)
except librouteros.exceptions.ConnectionClosed:
except librouteros_custom.exceptions.ConnectionClosed:
_LOGGER.error("Mikrotik %s connection closed", self._host)
self._connected = False
self._connection = None
return None
except (
librouteros.exceptions.TrapError,
librouteros.exceptions.MultiTrapError,
librouteros.exceptions.ProtocolError,
librouteros.exceptions.FatalError
librouteros_custom.exceptions.TrapError,
librouteros_custom.exceptions.MultiTrapError,
librouteros_custom.exceptions.ProtocolError,
librouteros_custom.exceptions.FatalError
) as api_error:
_LOGGER.error("Mikrotik %s connection error %s", self._host, api_error)
return None
@ -166,16 +166,16 @@ class MikrotikAPI:
try:
response.update(**params)
except librouteros.exceptions.ConnectionClosed:
except librouteros_custom.exceptions.ConnectionClosed:
_LOGGER.error("Mikrotik %s connection closed", self._host)
self._connected = False
self._connection = None
return None
except (
librouteros.exceptions.TrapError,
librouteros.exceptions.MultiTrapError,
librouteros.exceptions.ProtocolError,
librouteros.exceptions.FatalError
librouteros_custom.exceptions.TrapError,
librouteros_custom.exceptions.MultiTrapError,
librouteros_custom.exceptions.ProtocolError,
librouteros_custom.exceptions.FatalError
) as api_error:
_LOGGER.error("Mikrotik %s connection error %s", self._host, api_error)
return None
@ -210,16 +210,16 @@ class MikrotikAPI:
entry_found = True
try:
run = response('run', **{'.id': tmp['.id']})
except librouteros.exceptions.ConnectionClosed:
except librouteros_custom.exceptions.ConnectionClosed:
_LOGGER.error("Mikrotik %s connection closed", self._host)
self._connected = False
self._connection = None
return None
except (
librouteros.exceptions.TrapError,
librouteros.exceptions.MultiTrapError,
librouteros.exceptions.ProtocolError,
librouteros.exceptions.FatalError
librouteros_custom.exceptions.TrapError,
librouteros_custom.exceptions.MultiTrapError,
librouteros_custom.exceptions.ProtocolError,
librouteros_custom.exceptions.FatalError
) as api_error:
_LOGGER.error("Mikrotik %s connection error %s", self._host, api_error)
return None
@ -249,16 +249,16 @@ class MikrotikAPI:
args = {'interface': interfaces, 'once': True}
try:
traffic = response('monitor-traffic', **args)
except librouteros.exceptions.ConnectionClosed:
except librouteros_custom.exceptions.ConnectionClosed:
_LOGGER.error("Mikrotik %s connection closed", self._host)
self._connected = False
self._connection = None
return None
except (
librouteros.exceptions.TrapError,
librouteros.exceptions.MultiTrapError,
librouteros.exceptions.ProtocolError,
librouteros.exceptions.FatalError
librouteros_custom.exceptions.TrapError,
librouteros_custom.exceptions.MultiTrapError,
librouteros_custom.exceptions.ProtocolError,
librouteros_custom.exceptions.FatalError
) as api_error:
_LOGGER.error("Mikrotik %s connection error %s", self._host, api_error)
return None