get rid of two-level relative imports

Author: rnovatorov

enapter/mqtt/client.py

@@ -7,17 +7,18 @@
 
 import aiomqtt
 
-from .. import async_, mdns
+import enapter
+
 from .device_channel import DeviceChannel
 
 LOGGER = logging.getLogger(__name__)
 
 
-class Client(async_.Routine):
+class Client(enapter.async_.Routine):
     def __init__(self, config):
         self._logger = self._new_logger(config)
         self._config = config
-        self._mdns_resolver = mdns.Resolver()
+        self._mdns_resolver = enapter.mdns.Resolver()
         self._tls_context = self._new_tls_context(config)
         self._client = None
         self._client_ready = asyncio.Event()
@@ -40,7 +41,7 @@ async def publish(self, *args, **kwargs):
         client = await self._wait_client()
         await client.publish(*args, **kwargs)
 
-    @async_.generator
+    @enapter.async_.generator
     async def subscribe(self, topic):
         while True:
             client = await self._wait_client()

enapter/mqtt/device_channel.py

@@ -3,7 +3,8 @@
 import logging
 import time
 
-from .. import async_
+import enapter
+
 from .command import CommandRequest
 
 LOGGER = logging.getLogger(__name__)
@@ -29,7 +30,7 @@ def _new_logger(hardware_id, channel_id):
         extra = {"hardware_id": hardware_id, "channel_id": channel_id}
         return logging.LoggerAdapter(LOGGER, extra=extra)
 
-    @async_.generator
+    @enapter.async_.generator
     async def subscribe_to_command_requests(self):
         async with self._subscribe("v1/command/requests") as messages:
             async for msg in messages:

enapter/vucm/app.py

@@ -1,20 +1,21 @@
 import asyncio
 
-from .. import async_, log, mqtt
+import enapter
+
 from .config import Config
 from .ucm import UCM
 
 
 async def run(device_factory):
-    log.configure(level=log.LEVEL or "info")
+    enapter.log.configure(level=enapter.log.LEVEL or "info")
 
     config = Config.from_env()
 
     async with App(config=config, device_factory=device_factory) as app:
         await app.join()
 
 
-class App(async_.Routine):
+class App(enapter.async_.Routine):
     def __init__(self, config, device_factory):
         self._config = config
         self._device_factory = device_factory
@@ -23,7 +24,7 @@ async def _run(self):
         tasks = set()
 
         mqtt_client = await self._stack.enter_async_context(
-            mqtt.Client(config=self._config.mqtt)
+            enapter.mqtt.Client(config=self._config.mqtt)
         )
         tasks.add(mqtt_client.task())
 

enapter/vucm/config.py

@@ -2,7 +2,7 @@
 import json
 import os
 
-from .. import mqtt
+import enapter
 
 
 class Config:
@@ -23,7 +23,7 @@ def from_env(cls, prefix="ENAPTER_VUCM_", env=os.environ):
         hardware_id = os.environ[prefix + "HARDWARE_ID"]
         channel_id = os.environ[prefix + "CHANNEL_ID"]
 
-        mqtt_config = mqtt.Config.from_env(prefix=prefix, env=env)
+        mqtt_config = enapter.mqtt.Config.from_env(prefix=prefix, env=env)
 
         start_ucm = os.environ.get(prefix + "START_UCM", "1") != "0"
 
@@ -38,7 +38,7 @@ def from_env(cls, prefix="ENAPTER_VUCM_", env=os.environ):
     def from_blob(cls, blob):
         payload = json.loads(base64.b64decode(blob))
 
-        mqtt_config = mqtt.Config(
+        mqtt_config = enapter.mqtt.Config(
             host=payload["mqtt_host"],
             port=int(payload["mqtt_port"]),
             tls_ca_cert=payload["mqtt_ca"],

enapter/vucm/device.py

@@ -4,11 +4,12 @@
 import traceback
 from typing import Optional, Set
 
-from .. import async_, mqtt, types
+import enapter
+
 from .logger import Logger
 
 
-class Device(async_.Routine):
+class Device(enapter.async_.Routine):
     def __init__(
         self,
         channel,
@@ -28,7 +29,9 @@ def __init__(
         self.log = Logger(channel=channel)
         self.alerts: Set[str] = set()
 
-    async def send_telemetry(self, telemetry: Optional[types.JSON] = None) -> None:
+    async def send_telemetry(
+        self, telemetry: Optional[enapter.types.JSON] = None
+    ) -> None:
         if telemetry is None:
             telemetry = {}
         else:
@@ -38,7 +41,9 @@ async def send_telemetry(self, telemetry: Optional[types.JSON] = None) -> None:
 
         await self.__channel.publish_telemetry(telemetry)
 
-    async def send_properties(self, properties: Optional[types.JSON] = None) -> None:
+    async def send_properties(
+        self, properties: Optional[enapter.types.JSON] = None
+    ) -> None:
         if properties is None:
             properties = {}
         else:
@@ -104,9 +109,11 @@ async def __execute_command(self, req):
         try:
             cmd = getattr(self, self.__cmd_prefix + req.name)
         except AttributeError:
-            return mqtt.CommandState.ERROR, {"reason": "unknown command"}
+            return enapter.mqtt.CommandState.ERROR, {"reason": "unknown command"}
 
         try:
-            return mqtt.CommandState.COMPLETED, await cmd(**req.args)
+            return enapter.mqtt.CommandState.COMPLETED, await cmd(**req.args)
         except:
-            return mqtt.CommandState.ERROR, {"traceback": traceback.format_exc()}
+            return enapter.mqtt.CommandState.ERROR, {
+                "traceback": traceback.format_exc()
+            }

enapter/vucm/logger.py

@@ -1,6 +1,6 @@
 import logging
 
-from .. import mqtt
+import enapter
 
 LOGGER = logging.getLogger(__name__)
 
@@ -17,22 +17,30 @@ def _new_logger(hardware_id, channel_id):
 
     async def debug(self, msg: str, persist: bool = False):
         self._logger.debug(msg)
-        await self.log(msg, severity=mqtt.DeviceLogSeverity.DEBUG, persist=persist)
+        await self.log(
+            msg, severity=enapter.mqtt.DeviceLogSeverity.DEBUG, persist=persist
+        )
 
     async def info(self, msg: str, persist: bool = False):
         self._logger.info(msg)
-        await self.log(msg, severity=mqtt.DeviceLogSeverity.INFO, persist=persist)
+        await self.log(
+            msg, severity=enapter.mqtt.DeviceLogSeverity.INFO, persist=persist
+        )
 
     async def warning(self, msg: str, persist: bool = False):
         self._logger.warning(msg)
-        await self.log(msg, severity=mqtt.DeviceLogSeverity.WARNING, persist=persist)
+        await self.log(
+            msg, severity=enapter.mqtt.DeviceLogSeverity.WARNING, persist=persist
+        )
 
     async def error(self, msg: str, persist: bool = False):
         self._logger.error(msg)
-        await self.log(msg, severity=mqtt.DeviceLogSeverity.ERROR, persist=persist)
+        await self.log(
+            msg, severity=enapter.mqtt.DeviceLogSeverity.ERROR, persist=persist
+        )
 
     async def log(
-        self, msg: str, severity: mqtt.DeviceLogSeverity, persist: bool = False
+        self, msg: str, severity: enapter.mqtt.DeviceLogSeverity, persist: bool = False
     ):
         await self._channel.publish_logs(msg=msg, severity=severity, persist=persist)