Merge pull request #243 from jschlyter/minor_fixes

Minor fixes before release
pypolestar · Nov 18, 2024 · a6ace38 · a6ace38
2 parents fdf5a76 + aeae625
commit a6ace38
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 7 deletions.
diff --git a/Makefile b/Makefile
@@ -7,8 +7,8 @@ lint:
 	ruff check $(SOURCE)
 
 reformat:
-	ruff check --select I --fix $(SOURCE)
-	ruff format $(SOURCE)
+	ruff check --select I --fix $(SOURCE) tests
+	ruff format $(SOURCE) tests
 
 test:
-	PYTHONPATH=$(SOURCE) $(PYTEST) -vv tests
+	PYTHONPATH=. $(PYTEST) -vv tests
diff --git a/custom_components/polestar_api/pypolestar/polestar.py b/custom_components/polestar_api/pypolestar/polestar.py
@@ -292,8 +292,8 @@ async def _query_graph_ql(
                 self.logger.debug("GraphQL TransportQueryError: %s", str(exc))
                 if (
                     exc.errors
-                    and len(exc.errors)
-                    and exc.errors[0]["extensions"]["code"] == "UNAUTHENTICATED"
+                    and exc.errors[0].get("extensions", {}).get("code")
+                    == "UNAUTHENTICATED"
                 ):
                     self.latest_call_code = 401
                     raise PolestarNotAuthorizedException(

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -3,7 +3,8 @@
 from pathlib import Path
 
 import pytest
-from polestar_api.pypolestar.models import (
+
+from custom_components.polestar_api.pypolestar.models import (
     CarBatteryData,
     CarInformationData,
     CarOdometerData,

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1,7 +1,8 @@
 from datetime import date, datetime
 
 import pytest
-from polestar_api.pypolestar.utils import (
+
+from custom_components.polestar_api.pypolestar.utils import (
     get_field_name_date,
     get_field_name_datetime,
     get_field_name_float,