dycw-utilities 0.116.3__tar.gz → 0.116.4__tar.gz

{dycw_utilities-0.116.3 → dycw_utilities-0.116.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.116.3
+Version: 0.116.4
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.116.3 → dycw_utilities-0.116.4}/pyproject.toml

@@ -92,7 +92,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.116.3"
+version = "0.116.4"
 
 [project.optional-dependencies]
 test = [
@@ -334,7 +334,7 @@ zzz-test-zoneinfo = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.116.3"
+current_version = "0.116.4"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.116.3 → dycw_utilities-0.116.4}/src/tests/test_cryptography.py

@@ -31,5 +31,8 @@ class TestGetFernet:
         assert isinstance(fernet, Fernet)
 
     def test_error(self) -> None:
-        with temp_environ({_ENV_VAR: None}), raises(GetFernetError):
+        with (
+            temp_environ({_ENV_VAR: None}),
+            raises(GetFernetError, match="Environment variable 'FERNET_KEY' is None"),
+        ):
             _ = get_fernet()

{dycw_utilities-0.116.3 → dycw_utilities-0.116.4}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.116.3"
+__version__ = "0.116.4"

{dycw_utilities-0.116.3 → dycw_utilities-0.116.4}/src/utilities/cryptography.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
+from dataclasses import dataclass
 from os import getenv
+from typing import override
 
 from cryptography.fernet import Fernet
 
@@ -17,15 +19,23 @@ def decrypt(text: bytes, /, *, env_var: str = _ENV_VAR) -> str:
     return get_fernet(env_var=env_var).decrypt(text).decode()
 
 
+##
+
+
 def get_fernet(*, env_var: str = _ENV_VAR) -> Fernet:
     """Get the Fernet key."""
     if (key := getenv(env_var)) is None:
-        msg = f"{env_var!r} is None"
-        raise GetFernetError(msg)
+        raise GetFernetError(env_var=env_var)
     return Fernet(key.encode())
 
 
-class GetFernetError(Exception): ...
+@dataclass(kw_only=True, slots=True)
+class GetFernetError(Exception):
+    env_var: str
+
+    @override
+    def __str__(self) -> str:
+        return f"Environment variable {self.env_var!r} is None"
 
 
 __all__ = ["GetFernetError", "decrypt", "encrypt", "get_fernet"]