annotating Enum.value

see also: python/typing#535

Author: Fallen-Breath

prime_backup/cli/return_codes.py

@@ -1,5 +1,6 @@
 import enum
 import sys
+from typing import TYPE_CHECKING
 
 from typing_extensions import NoReturn
 
@@ -12,5 +13,8 @@ class ErrorReturnCodes(enum.Enum):
 	backup_file_not_found = 5
 	missing_dependency = 6
 
+	if TYPE_CHECKING:
+		value: int
+
 	def sys_exit(self) -> NoReturn:
 		sys.exit(self.value)

prime_backup/compressors.py

@@ -2,7 +2,7 @@
 import dataclasses
 import enum
 from abc import abstractmethod, ABC
-from typing import BinaryIO, Union, ContextManager, Tuple, Callable, Literal, Generator
+from typing import BinaryIO, Union, ContextManager, Tuple, Callable, Literal, Generator, TYPE_CHECKING, Type
 
 from typing_extensions import Protocol, override
 
@@ -222,5 +222,8 @@ class CompressMethod(enum.Enum):
 	zstd = ZstdCompressor
 	lz4 = Lz4Compressor
 
+	if TYPE_CHECKING:
+		value: Type[Compressor]
+
 	def __repr__(self) -> str:
 		return '{}({!r})'.format(self.__class__.__name__, self.name)

prime_backup/types/backup_tags.py

@@ -1,5 +1,5 @@
 import enum
-from typing import Optional, Any, Type, List
+from typing import Optional, Any, Type, List, TYPE_CHECKING
 
 from mcdreforged.api.all import RText, RTextBase, RColor
 
@@ -31,6 +31,9 @@ class BackupTagName(enum.Enum):
 	protected = BackupTagValue(bool, 'P', RColor.dark_green)
 	scheduled = BackupTagValue(bool, 'S', RColor.dark_blue)
 
+	if TYPE_CHECKING:
+		value: BackupTagValue
+
 	@classmethod
 	def bool_tags(cls) -> List['BackupTagName']:
 		return [

prime_backup/types/chunk_method.py

@@ -1,8 +1,8 @@
 import enum
 from pathlib import Path
-from typing import Optional, IO
+from typing import Optional, IO, TYPE_CHECKING
 
-from prime_backup.types.chunker_factory import CDCChunkerFactory, FixedSizeChunkerFactory
+from prime_backup.types.chunker_factory import ChunkerFactory, CDCChunkerFactory, FixedSizeChunkerFactory
 from prime_backup.utils.chunker import Chunker
 from prime_backup.utils.path_like import PathLike
 
@@ -18,6 +18,9 @@ class ChunkMethod(enum.Enum):
 	fixed_32k = FixedSizeChunkerFactory(32 * 1024)
 	fixed_128k = FixedSizeChunkerFactory(128 * 1024)
 
+	if TYPE_CHECKING:
+		value: ChunkerFactory
+
 	@classmethod
 	def get_for_file(cls, file_path: PathLike, file_size: int) -> Optional['ChunkMethod']:
 		"""

prime_backup/types/hash_method.py

@@ -1,7 +1,7 @@
 import dataclasses
 import enum
 import importlib
-from typing import Protocol, Union
+from typing import Protocol, Union, TYPE_CHECKING
 
 
 class Hasher(Protocol):
@@ -30,6 +30,9 @@ class HashMethod(enum.Enum):
 	sha256 = _HashMethodItem('hashlib.sha256', 64)
 	blake3 = _HashMethodItem('blake3.blake3', 64)
 
+	if TYPE_CHECKING:
+		value: _HashMethodItem
+
 
 def __verify_hex_length():
 	for hash_method in HashMethod:

prime_backup/types/standalone_backup_format.py

@@ -1,7 +1,7 @@
 import dataclasses
 import enum
 import os
-from typing import Optional, List
+from typing import Optional, List, Union, TYPE_CHECKING
 
 from prime_backup.types.tar_format import TarFormat
 from prime_backup.utils.path_like import PathLike
@@ -24,12 +24,16 @@ class StandaloneBackupFormat(enum.Enum):
 	tar_zst = TarFormat.zstd
 	zip = ZipFormat('.zip')
 
+	if TYPE_CHECKING:
+		value: Union[TarFormat, ZipFormat]
+
 	@property
 	def __all_file_extensions(self) -> List[str]:
-		if isinstance(self.value, TarFormat):
-			return self.value.value.all_extensions
-		elif isinstance(self.value, ZipFormat):
-			return self.value.all_extensions
+		format_value = self.value
+		if isinstance(format_value, TarFormat):
+			return format_value.value.all_extensions
+		elif isinstance(format_value, ZipFormat):
+			return format_value.all_extensions
 		else:
 			raise ValueError(self.value)
 

prime_backup/types/tar_format.py

@@ -1,6 +1,6 @@
 import dataclasses
 import enum
-from typing import Tuple, List, Literal, cast
+from typing import Tuple, List, Literal, cast, TYPE_CHECKING
 
 from prime_backup.compressors import CompressMethod
 
@@ -35,6 +35,9 @@ class TarFormat(enum.Enum):
 	lzma = _TarFormatItem('.tar.xz', ('.txz',), ':xz', CompressMethod.plain)
 	zstd = _TarFormatItem('.tar.zst', ('.tar.zstd', '.tzst', '.tzstd'), ':', CompressMethod.zstd)
 
+	if TYPE_CHECKING:
+		value: _TarFormatItem
+
 
 def __validate_tar_formats():
 	for tf in TarFormat: