fiftyfive 0.3.0__tar.gz → 0.4.0__tar.gz

{fiftyfive-0.3.0 → fiftyfive-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiftyfive
-Version: 0.3.0
+Version: 0.4.0
 Summary: Python 3 package to retrieve EV charger data from 50five
 Project-URL: Homepage, https://github.com/Crazy-Duck/fiftyfive
 Project-URL: Bug Tracker, https://github.com/Crazy-Duck/fiftyfive/issues
@@ -29,14 +29,14 @@ pip3 install fiftyfive
 import asyncio
 
 from aiohttp import ClientSession
-from fiftyfive import Api, Market, NetworkOverview
+from fiftyfive import Api, CustomerType, Market, NetworkOverview
 
 
 async def main():
     """Main module"""
 
     async with ClientSession() as session:
-        api = Api(session, "email", "password", Market.BELUX, was_shell=True)
+        api = Api(session, "email", "password", Market.BELUX, CustomerType.FORMER_SHELL)
 
         result = await api.make_requests([NetworkOverview()])
         print(result)

{fiftyfive-0.3.0 → fiftyfive-0.4.0}/README.md

@@ -14,14 +14,14 @@ pip3 install fiftyfive
 import asyncio
 
 from aiohttp import ClientSession
-from fiftyfive import Api, Market, NetworkOverview
+from fiftyfive import Api, CustomerType, Market, NetworkOverview
 
 
 async def main():
     """Main module"""
 
     async with ClientSession() as session:
-        api = Api(session, "email", "password", Market.BELUX, was_shell=True)
+        api = Api(session, "email", "password", Market.BELUX, CustomerType.FORMER_SHELL)
 
         result = await api.make_requests([NetworkOverview()])
         print(result)

{fiftyfive-0.3.0 → fiftyfive-0.4.0}/example.py

@@ -9,7 +9,7 @@ from sys import stdout
 from aiohttp import ClientSession
 from dotenv import load_dotenv
 
-from fiftyfive import Api, Market, NetworkOverview, Overview, Channel
+from fiftyfive import Api, CustomerType, Market, NetworkOverview, Overview
 
 load_dotenv()
 
@@ -18,7 +18,7 @@ async def main():
     """Main module"""
 
     async with ClientSession() as session:
-        api = Api(session, getenv("50FIVE_EMAIL"), getenv("50FIVE_PASSWORD"), Market.BELUX)
+        api = Api(session, getenv("50FIVE_EMAIL"), getenv("50FIVE_PASSWORD"), Market.BELUX, CustomerType.FORMER_SHELL)
 
         networks = await api.make_requests([NetworkOverview()])
         networks[0] += networks[0]

{fiftyfive-0.3.0 → fiftyfive-0.4.0}/fiftyfive/api.py

@@ -19,17 +19,29 @@ class Market(StrEnum):
     UK = "uk"
 
 
+class CustomerType(StrEnum):
+    """Enum type for customers."""
+
+    FIFTYFIVE = "fiftyfive"
+    FORMER_SHELL = "shell"
+
+
 class Api:
     """Class to make API requests"""
 
     def __init__(
-        self, session: ClientSession, email: str, password: str, market: Market, was_shell=True
+        self,
+        session: ClientSession,
+        email: str,
+        password: str,
+        market: Market,
+        customer_type=CustomerType.FORMER_SHELL,
     ):
         """Initilaize the session."""
         self.session = session
         self.email = email
         self.password = password
-        self.url = f"https://50five{'-s' if was_shell else ''}{market}.evc-net.com"
+        self.url = f"https://50five{'-s' if customer_type == CustomerType.FORMER_SHELL else ''}{market}.evc-net.com"
         self.api = f"{self.url}/api/ajax"
         self.logger = logging.getLogger("50five")
 

{fiftyfive-0.3.0 → fiftyfive-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fiftyfive"
-version = "0.3.0"
+version = "0.4.0"
 description = "Python 3 package to retrieve EV charger data from 50five"
 dependencies = [
     "aiohttp",