Change StorageType to be StrEnum class (#798)

NabuCasa · Feb 6, 2025 · 75ff500 · 75ff500
1 parent f0aa92f
commit 75ff500
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/hass_nabucasa/files.py b/hass_nabucasa/files.py
@@ -3,8 +3,9 @@
 from __future__ import annotations
 
 from collections.abc import AsyncIterator, Callable, Coroutine
+from enum import StrEnum
 import logging
-from typing import TYPE_CHECKING, Any, Literal, TypedDict
+from typing import TYPE_CHECKING, Any, TypedDict
 
 from aiohttp import (
     ClientResponseError,
@@ -18,7 +19,11 @@
 
 _FILE_TRANSFER_TIMEOUT = 43200.0  # 43200s == 12h
 
-type StorageType = Literal["backup"]
+
+class StorageType(StrEnum):
+    """Storage types."""
+
+    BACKUP = "backup"
 
 
 class FilesError(CloudApiError):