Improve importing experience

scripts/download_and_parse.py

@@ -5,10 +5,7 @@
 
 from dotenv import load_dotenv
 
-from tuya_vacuum.map.layout import Layout
-from tuya_vacuum.map.map import Map
-from tuya_vacuum.map.path import Path
-from tuya_vacuum.vacuum import Vacuum
+import tuya_vacuum
 
 logging.basicConfig(level=logging.DEBUG)
 
@@ -24,18 +21,18 @@
 
 def main():
     """Download, parse, and save the current real-time vacuum map."""
-    vacuum = Vacuum(ORIGIN, CLIENT_ID, CLIENT_SECRET, DEVICE_ID)
+    vacuum = tuya_vacuum.Vacuum(ORIGIN, CLIENT_ID, CLIENT_SECRET, DEVICE_ID)
 
     vacuum_map_data = vacuum.fetch_realtime_map_data()
     if (
         vacuum_map_data
         and vacuum_map_data["layout_data"]
         and vacuum_map_data["path_data"]
     ):
-        layout_map = Layout(vacuum_map_data["layout_data"])
-        path_map = Path(vacuum_map_data["path_data"])
-        map = Map(layout_map, path_map)
-        image = map.to_image()
+        layout_map = tuya_vacuum.map.Layout(vacuum_map_data["layout_data"])
+        path_map = tuya_vacuum.map.Path(vacuum_map_data["path_data"])
+        vacuum_map = tuya_vacuum.Map(layout_map, path_map)
+        image = vacuum_map.to_image()
         image.save("map.png")
     else:
         print("Err, what the hell")

tuya_vacuum/__init__.py

@@ -1 +1,6 @@
 """Module to get maps from tuya vacuums."""
+
+from .map import Map
+from .vacuum import Vacuum
+
+__all__ = ["Map", "Vacuum"]

tuya_vacuum/errors.py

@@ -0,0 +1,17 @@
+"""Package level errors"""
+
+
+class InvalidClientIDError(Exception):
+    """Invalid Client ID Error."""
+
+
+class InvalidClientSecretError(Exception):
+    """Invalid Client Secret Error."""
+
+
+class InvalidDeviceIDError(Exception):
+    """Invalid Device ID Error."""
+
+
+class CrossRegionAccessError(Exception):
+    """Cross Region Access Error."""

tuya_vacuum/map/__init__.py

@@ -0,0 +1,8 @@
+"""Map related classes."""
+
+from .layout import Layout
+from .map import Map
+from .path import Path
+from .room import Room
+
+__all__ = ["Layout", "Path", "Room", "Map"]

tuya_vacuum/tuya.py

@@ -9,6 +9,13 @@
 
 import httpx
 
+from tuya_vacuum.errors import (
+    CrossRegionAccessError,
+    InvalidClientIDError,
+    InvalidClientSecretError,
+    InvalidDeviceIDError,
+)
+
 _LOGGER = logging.getLogger(__name__)
 
 # The SHA256 hash of a empty request body
@@ -18,22 +25,6 @@
 ACCESS_TOKEN_ENDPOINT = "/v1.0/token?grant_type=1"
 
 
-class InvalidClientIDError(Exception):
-    """Invalid Client ID Error."""
-
-
-class InvalidClientSecretError(Exception):
-    """Invalid Client Secret Error."""
-
-
-class InvalidDeviceIDError(Exception):
-    """Invalid Device ID Error."""
-
-
-class CrossRegionAccessError(Exception):
-    """Cross Region Access Error."""
-
-
 class TuyaCloudAPI:
     """Handles communication with a Tuya Cloud API."""