Converted krypton file suffix and kdf context into constants

Author: aabmets

quantcrypt/internal/cipher/krypton_kem.py

@@ -15,7 +15,7 @@
 from quantcrypt.internal.kdf.argon2_kdf import Argon2
 from quantcrypt.internal.pqa.base_kem import BaseKEM
 from quantcrypt.internal.chunksize import ChunkSize
-from quantcrypt.internal import utils
+from quantcrypt.internal import utils, constants as const
 
 
 __all__ = ["KryptonKEM"]
@@ -44,7 +44,7 @@ def __init__(
 			self,
 			kem_class: Type[BaseKEM],
 			kdf_params: KDFParams = None,
-			context: bytes = b"quantcrypt",
+			context: bytes = const.KDFContext,
 			callback: Optional[Callable] = None,
 			chunk_size: ChunkSize.Atd = None
 	) -> None:
@@ -111,7 +111,7 @@ def encrypt(
 			raise FileNotFoundError(_in_file)
 
 		if output_file is None:
-			_out_file = _in_file.with_suffix(".kptn")
+			_out_file = _in_file.with_suffix(const.KryptonFileSuffix)
 		else:
 			_out_file = utils.resolve_relpath(output_file)
 

quantcrypt/internal/cli/annotations.py

@@ -9,8 +9,9 @@
 #   SPDX-License-Identifier: MIT
 #
 
-from typing import Annotated
 from typer import Option
+from typing import Annotated
+from quantcrypt.internal import constants as const
 
 
 __all__ = [
@@ -80,7 +81,8 @@
 EncOutFile = Annotated[str, Option(
 	'--out-file', '-o', show_default=False, help=' '.join([
 		"Path to the output file where the encrypted data will be written to, optional.",
-		"Defaults to the Current Working Directory, using the data file name with the .kptn suffix."
+		"Defaults to the Current Working Directory, using the data file name "
+		f"with the {const.KryptonFileSuffix} suffix."
 	])
 )]
 
@@ -94,7 +96,7 @@
 DecOutFile = Annotated[str, Option(
 	'--out-file', '-o', show_default=False, help=' '.join([
 		"Path to the output file where the decrypted data will be written to, optional.",
-		"Defaults to the Current Working Directory, using the original filename of the ",
+		"Defaults to the Current Working Directory, using the original filename of the",
 		"plaintext file that was stored into the ciphertext file."
 	])
 )]

quantcrypt/internal/constants.py

@@ -118,6 +118,8 @@ def armor_names(cls, pqa_type: PQAType | None = None) -> list[str]:
             if not pqa_type or spec.type == pqa_type]
 
 
+KDFContext = b"quantcrypt"
+KryptonFileSuffix = ".kptn"
 SupportedVariants = [PQAVariant.REF]
 AMDArches = ["x86_64", "amd64", "x86-64", "x64", "intel64"]
 ARMArches = ["arm_8", "arm64", "aarch64", "armv8", "armv8-a"]