From 33a812ae767f2f654a999652fcfd6d1538804477 Mon Sep 17 00:00:00 2001 From: Luke Date: Wed, 8 Feb 2023 14:03:27 -0500 Subject: [PATCH] fix: made accounts private --- src/southern_company_api/parser.py | 10 +++++++--- tests/test_parser.py | 1 + 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/src/southern_company_api/parser.py b/src/southern_company_api/parser.py index dc88fec..d1bbbd6 100644 --- a/src/southern_company_api/parser.py +++ b/src/southern_company_api/parser.py @@ -44,7 +44,11 @@ def __init__(self, username: str, password: str): self.jwt: typing.Optional[str] = None self.sc: typing.Optional[str] = None self.request_token: typing.Optional[str] = None - self.accounts: typing.Optional[List[Account]] = None + self._accounts: typing.Optional[List[Account]] = None + + @property + def accounts(self) -> typing.Optional[List[Account]]: + return self._accounts async def connect(self) -> None: """ @@ -53,7 +57,7 @@ async def connect(self) -> None: self.request_token = await get_request_verification_token() self.sc = await self._get_sc_web_token() self.jwt = await self.get_jwt() - self.accounts = await self.get_accounts() + self._accounts = await self.get_accounts() async def authenticate(self) -> bool: """Determines if you can authenticate with Southern Company with given login""" @@ -202,5 +206,5 @@ async def get_accounts(self) -> List[Account]: company=COMPANY_MAP.get(account["Company"], Company.GPC), ) ) - self.accounts = accounts + self._accounts = accounts return accounts diff --git a/tests/test_parser.py b/tests/test_parser.py index bcba69e..5d3aa3e 100644 --- a/tests/test_parser.py +++ b/tests/test_parser.py @@ -123,3 +123,4 @@ async def test_ga_power_get_accounts(): sca.jwt = "sample" response_token: typing.List[Account] = await sca.get_accounts() assert response_token[0].name == "Home Energy" + assert sca.accounts == response_token