Ensure file and folder creation

Author: AfonsoFGarcia

custom_components/bluecon/ConfigFolderNotificationInfoStorage.py

@@ -4,11 +4,17 @@
 from typing import Any
 from .const import DOMAIN
 import json
+from pathlib import Path
 
 class ConfigFolderNotificationInfoStorage(INotificationInfoStorage):
     def __init__(self, hass: HomeAssistant, entry: ConfigEntry):
-        self.__credentialsFileName = hass.config.path(f'.{DOMAIN}', entry.entry_id, 'credentials.json')
-        self.__persistentIdsFileName = hass.config.path(f'.{DOMAIN}', entry.entry_id, 'persistent_ids.txt')
+        self.__credentialsFileName = Path(hass.config.path(f'.{DOMAIN}', entry.entry_id, 'credentials.json'))
+        self.__persistentIdsFileName = Path(hass.config.path(f'.{DOMAIN}', entry.entry_id, 'persistent_ids.txt'))
+
+        self.__credentialsFileName.mkdir(parents=True, exist_ok=True)
+        self.__credentialsFileName.touch(exist_ok=True)
+        self.__persistentIdsFileName.mkdir(parents=True, exist_ok=True)
+        self.__persistentIdsFileName.touch(exist_ok=True)
 
     def retrieveCredentials(self) -> dict[str, dict[str, Any]] | None:
         try:
@@ -18,7 +24,7 @@ def retrieveCredentials(self) -> dict[str, dict[str, Any]] | None:
             return None
 
     def storeCredentials(self, credentials: dict[str, dict[str, Any]]):
-        with open(self.__credentialsFileName, "w+") as f:
+        with open(self.__credentialsFileName, "w") as f:
             json.dump(credentials, f)
 
     def retrievePersistentIds(self) -> list[str] | None:
@@ -29,5 +35,5 @@ def retrievePersistentIds(self) -> list[str] | None:
             return None
 
     def storePersistentId(self, persistentId: str):
-        with open(self.__persistentIdsFileName, "a+") as f:
+        with open(self.__persistentIdsFileName, "a") as f:
             f.write(persistentId + "\n")

custom_components/bluecon/ConfigFolderOAuthTokenStorage.py

@@ -2,15 +2,19 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from .const import DOMAIN
+from pathlib import Path
 
 class ConfigFolderOAuthTokenStorage(IOAuthTokenStorage):
     def __init__(self, hass: HomeAssistant, entry: ConfigEntry):
-        self.__oAuthTokenFileName = hass.config.path(f'.{DOMAIN}', entry.entry_id, 'oauth_token.json')
+        self.__oAuthTokenFileName = Path(hass.config.path(f'.{DOMAIN}', entry.entry_id, 'oauth_token.json'))
+
+        self.__oAuthTokenFileName.mkdir(parents=True, exist_ok=True)
+        self.__oAuthTokenFileName.touch(exist_ok=True)
 
     def retrieveOAuthToken(self) -> OAuthToken:
         with open(self.__oAuthTokenFileName, "r") as f:
             return OAuthToken.fromJson(f)
 
     def storeOAuthToken(self, oAuthToken: OAuthToken):
-        with open(self.__oAuthTokenFileName, "w+") as f:
+        with open(self.__oAuthTokenFileName, "w") as f:
             f.write(oAuthToken.toJson())

custom_components/bluecon/manifest.json

@@ -1,7 +1,7 @@
 {
     "domain": "bluecon",
     "name": "Fermax Blue",
-    "version": "0.1.12",
+    "version": "0.1.13",
     "documentation": "https://hass-bluecon.afonsogarcia.dev/",
     "issue_tracker": "https://github.com/AfonsoFGarcia/hass-bluecon/issues",
     "requirements": ["bluecon==0.1.2", "aiohttp", "oscrypto", "protobuf", "http-ece"],