From fadd3167fd74ff567c2a0fea99c275128815a889 Mon Sep 17 00:00:00 2001
From: David Wahlstrom <david.wahlstrom@gmail.com>
Date: Mon, 22 Nov 2021 06:51:59 -0700
Subject: [PATCH] Fix teslacharge module (#809)

Tesla made changes to their API, which broke some pip modules. As such,
we need to switch to a new module and update some calls to properly
track the changes in module behaviour.

Signed-off-by: David Wahlstrom <david.wahlstrom@gmail.com>
---
 docs/conf.py              |  2 +-
 i3pystatus/teslacharge.py | 16 ++++++++--------
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/docs/conf.py b/docs/conf.py
index d332071..9354843 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -68,7 +68,7 @@ MOCK_MODULES = [
     "requests.adapters",
     "exchangelib",
     "soco",
-    "tesla_api",
+    "teslapy"
     "yfinance"
 ]
 
diff --git a/i3pystatus/teslacharge.py b/i3pystatus/teslacharge.py
index d06153e..e633755 100644
--- a/i3pystatus/teslacharge.py
+++ b/i3pystatus/teslacharge.py
@@ -1,4 +1,4 @@
-from tesla_api import TeslaApiClient
+import teslapy
 from i3pystatus import IntervalModule
 
 
@@ -7,7 +7,7 @@ class TeslaCharge(IntervalModule):
     Displays the current charge/range of your Tesla vehicle. There is a ton of
     data that could be displayed, so read this module to see the full list of
     datapoints that can be displayed.
-    Requires: tesla_api
+    Requires: teslapy
     """
 
     settings = (
@@ -32,21 +32,21 @@ class TeslaCharge(IntervalModule):
 
     def run(self):
         # Setup Tesla API client
-        tclient = TeslaApiClient(self.email, self.password)
-        vehicles = tclient.list_vehicles()
+        tclient = teslapy.Tesla(self.email, self.password)
+        vehicles = tclient.vehicle_list()
         # Currently, only one vehicle is supported. It would be nice to be
         # able to click through multipe vehicles.
         thisvehicle = vehicles[0]
-        display_name = thisvehicle.display_name
+        display_name = thisvehicle['display_name']
 
         # If vehicle is offline, do not bother grabbing info
-        if thisvehicle.state != 'online':
+        if thisvehicle['state'] != 'online':
             self.output = {
-                "full_text": "%s: %s" % (display_name, thisvehicle.state),
+                "full_text": "%s: %s" % (display_name, thisvehicle['state']),
                 "color": self.offline_color
             }
         else:
-            charge_info = vehicles[0].charge.get_state()
+            charge_info = vehicles[0].get_vehicle_data()['charge_state']
 
             # Miles or meters?
             if self.units != "miles":