Cleanup

Author: Bre77

tesla_fleet_api/tesla/__init__.py

@@ -3,9 +3,23 @@
 from tesla_fleet_api.tesla.fleet import TeslaFleetApi
 from tesla_fleet_api.tesla.bluetooth import TeslaBluetooth
 from tesla_fleet_api.tesla.oauth import TeslaFleetOAuth
+from tesla_fleet_api.tesla.charging import Charging
+from tesla_fleet_api.tesla.energysite import EnergySites
+from tesla_fleet_api.tesla.partner import Partner
+from tesla_fleet_api.tesla.user import User
+from tesla_fleet_api.tesla.vehicle import Vehicles, VehiclesBluetooth, VehicleFleet, VehicleSigned, VehicleBluetooth
 
 __all__ = [
     "TeslaFleetApi",
     "TeslaBluetooth",
     "TeslaFleetOAuth",
+    "Charging",
+    "EnergySites",
+    "Partner",
+    "User",
+    "Vehicles",
+    "VehiclesBluetooth",
+    "VehicleFleet",
+    "VehicleSigned",
+    "VehicleBluetooth"
 ]

tesla_fleet_api/tesla/vehicle/__init__.py

@@ -1,12 +1,13 @@
 """Tesla Fleet API classes."""
 
-from tesla_fleet_api.tesla.vehicle.vehicles import Vehicles
+from tesla_fleet_api.tesla.vehicle.vehicles import Vehicles, VehiclesBluetooth
 from tesla_fleet_api.tesla.vehicle.fleet import VehicleFleet
 from tesla_fleet_api.tesla.vehicle.bluetooth import VehicleBluetooth
 from tesla_fleet_api.tesla.vehicle.signed import VehicleSigned
 
 __all__ = [
     "Vehicles",
+    "VehiclesBluetooth",
     "VehicleFleet",
     "VehicleBluetooth",
     "VehicleSigned",

tesla_fleet_api/tesla/vehicle/vehicles.py

@@ -1,16 +1,12 @@
 from __future__ import annotations
-import asyncio
 from typing import TYPE_CHECKING
-from bleak import BleakClient
 from bleak.backends.device import BLEDevice
-from bleak_retry_connector import establish_connection
 from cryptography.hazmat.primitives.asymmetric import ec
 
 from tesla_fleet_api.tesla.vehicle.signed import VehicleSigned
-from tesla_fleet_api.tesla.vehicle.bluetooth import NAME_UUID, VehicleBluetooth
+from tesla_fleet_api.tesla.vehicle.bluetooth import VehicleBluetooth
 from tesla_fleet_api.tesla.vehicle.fleet import VehicleFleet
 from tesla_fleet_api.tesla.vehicle.vehicle import Vehicle
-from tesla_fleet_api.const import LOGGER
 
 if TYPE_CHECKING:
     from tesla_fleet_api.tesla.fleet import TeslaFleetApi

tesla_fleet_api/teslemetry/__init__.py

@@ -1,5 +1,14 @@
 from tesla_fleet_api.teslemetry.teslemetry import Teslemetry
+from tesla_fleet_api.tesla.charging import Charging
+from tesla_fleet_api.tesla.energysite import EnergySites
+from tesla_fleet_api.tesla.user import User
+from tesla_fleet_api.teslemetry.vehicles import TeslemetryVehicles as Vehicles, TeslemetryVehicle as Vehicle
 
 __all__ = [
     "Teslemetry",
+    "Charging",
+    "EnergySites",
+    "User",
+    "Vehicles",
+    "Vehicle",
 ]

tesla_fleet_api/tessie/__init__.py

@@ -1,5 +1,14 @@
 from tesla_fleet_api.tessie.tessie import Tessie
+from tesla_fleet_api.tesla.charging import Charging
+from tesla_fleet_api.tesla.energysite import EnergySites
+from tesla_fleet_api.tesla.user import User
+from tesla_fleet_api.tessie.vehicles import TessieVehicles as Vehicles, TessieVehicle as Vehicle
 
 __all__ = [
     "Tessie",
+    "Charging",
+    "EnergySites",
+    "User",
+    "Vehicles",
+    "Vehicle",
 ]