style: reformat all files with black 24.10.0

Author: apitanga

src/nakimi/core/config.py

@@ -65,9 +65,7 @@ def _load_config(self):
         self._yubikey_enabled = self._get_bool(
             "NAKIMI_YUBIKEY_ENABLED", config_values.get("yubikey_enabled"), False
         )
-        self._yubikey_slot = self._get_str(
-            "NAKIMI_YUBIKEY_SLOT", config_values.get("yubikey_slot"), "1"
-        )
+        self._yubikey_slot = self._get_str("NAKIMI_YUBIKEY_SLOT", config_values.get("yubikey_slot"), "1")
         self._yubikey_require_touch = self._get_bool(
             "NAKIMI_YUBIKEY_REQUIRE_TOUCH", config_values.get("yubikey_require_touch"), True
         )

src/nakimi/core/plugin.py

@@ -115,9 +115,7 @@ def __init__(self, secrets_data: Optional[Dict[str, Any]] = None):
         """
         self.secrets_data = secrets_data or {}
         self._plugins: Dict[str, Plugin] = {}
-        self._commands: Dict[str, tuple[str, PluginCommand]] = (
-            {}
-        )  # full_name -> (plugin_name, command)
+        self._commands: Dict[str, tuple[str, PluginCommand]] = {}  # full_name -> (plugin_name, command)
 
     def register_plugin(self, plugin_class: type[Plugin], plugin_secrets: Optional[Dict] = None):
         """
@@ -183,11 +181,7 @@ def discover_plugins(self):
                     # Find the Plugin subclass
                     for attr_name in dir(module):
                         attr = getattr(module, attr_name)
-                        if (
-                            isinstance(attr, type)
-                            and issubclass(attr, Plugin)
-                            and attr is not Plugin
-                        ):
+                        if isinstance(attr, type) and issubclass(attr, Plugin) and attr is not Plugin:
                             self.register_plugin(attr)
                             break
 

src/nakimi/core/vault.py

@@ -291,9 +291,7 @@ def _check_age_installed(self):
         try:
             subprocess.run(["age", "--version"], capture_output=True, check=True)
         except (subprocess.CalledProcessError, FileNotFoundError):
-            raise VaultCryptoError(
-                "age is not installed. Install it from https://age-encryption.org"
-            )
+            raise VaultCryptoError("age is not installed. Install it from https://age-encryption.org")
 
     def generate_key(self) -> str:
         """Generate a new age key pair"""

src/nakimi/core/yubikey.py

@@ -107,8 +107,7 @@ def _get_yubikey_recipient(self) -> str:
             raise YubiKeyError("No recipient found in age-plugin-yubikey output")
         except FileNotFoundError:
             raise YubiKeyError(
-                "age-plugin-yubikey not found. "
-                "Install it from https://github.com/str4d/age-plugin-yubikey"
+                "age-plugin-yubikey not found. " "Install it from https://github.com/str4d/age-plugin-yubikey"
             )
         except subprocess.CalledProcessError as e:
             raise YubiKeyError(f"Failed to get recipient: {e.stderr}")
@@ -131,8 +130,7 @@ def _get_yubikey_identity(self) -> str:
             return result.stdout.strip()
         except FileNotFoundError:
             raise YubiKeyError(
-                "age-plugin-yubikey not found. "
-                "Install it from https://github.com/str4d/age-plugin-yubikey"
+                "age-plugin-yubikey not found. " "Install it from https://github.com/str4d/age-plugin-yubikey"
             )
         except subprocess.CalledProcessError as e:
             raise YubiKeyError(f"Failed to get identity: {e.stderr}")
@@ -341,9 +339,7 @@ def verify_pin(self, pin: str) -> bool:
             return False
 
         try:
-            subprocess.run(
-                ["ykman", "piv", "verify-pin", pin], capture_output=True, text=True, check=True
-            )
+            subprocess.run(["ykman", "piv", "verify-pin", pin], capture_output=True, text=True, check=True)
             return True
         except subprocess.CalledProcessError:
             return False

src/nakimi/plugins/gmail/client.py

@@ -274,10 +274,7 @@ def search(self, query: str, max_results: int = 10) -> List[Dict[str, Any]]:
             self._ensure_valid_token()
 
             results = self._execute_with_retry(
-                self.service.users()
-                .messages()
-                .list(userId="me", q=query, maxResults=max_results)
-                .execute
+                self.service.users().messages().list(userId="me", q=query, maxResults=max_results).execute
             )
 
             if not results:
@@ -320,9 +317,7 @@ def list_labels(self) -> List[Dict[str, Any]]:
         """List Gmail labels"""
         try:
             self._ensure_valid_token()
-            results = self._execute_with_retry(
-                self.service.users().labels().list(userId="me").execute
-            )
+            results = self._execute_with_retry(self.service.users().labels().list(userId="me").execute)
             return results.get("labels", []) if results else []
         except GmailAuthError:
             return []

src/nakimi/plugins/gmail/plugin.py

@@ -74,9 +74,7 @@ def get_commands(self) -> List[PluginCommand]:
                     ("limit", "Maximum results", False),
                 ],
             ),
-            PluginCommand(
-                name="labels", description="List Gmail labels", handler=self.cmd_labels, args=[]
-            ),
+            PluginCommand(name="labels", description="List Gmail labels", handler=self.cmd_labels, args=[]),
             PluginCommand(
                 name="profile", description="Show Gmail profile", handler=self.cmd_profile, args=[]
             ),

tests/unit/test_plugin.py

@@ -146,9 +146,7 @@ def _validate_secrets(self):
             def get_commands(self):
                 return [
                     PluginCommand("cmd1", "Command 1", lambda: "result1", []),
-                    PluginCommand(
-                        "cmd2", "Command 2", lambda x: f"result{x}", [("x", "Argument", True)]
-                    ),
+                    PluginCommand("cmd2", "Command 2", lambda x: f"result{x}", [("x", "Argument", True)]),
                 ]
 
         manager = PluginManager()
@@ -330,9 +328,7 @@ def _validate_secrets(self):
 
             def get_commands(self):
                 return [
-                    PluginCommand(
-                        "echo", "Echo input", self.cmd_echo, [("text", "Text to echo", True)]
-                    ),
+                    PluginCommand("echo", "Echo input", self.cmd_echo, [("text", "Text to echo", True)]),
                     PluginCommand(
                         "add",
                         "Add numbers",
@@ -379,11 +375,7 @@ def _validate_secrets(self):
                 pass
 
             def get_commands(self):
-                return [
-                    PluginCommand(
-                        "echo", "Echo input", self.cmd_echo, [("text", "Text to echo", True)]
-                    )
-                ]
+                return [PluginCommand("echo", "Echo input", self.cmd_echo, [("text", "Text to echo", True)])]
 
             def cmd_echo(self, text):
                 return f"Echo: {text}"

tests/unit/test_vault.py

@@ -258,9 +258,7 @@ def test_secure_delete_with_shred(self, mock_run, mock_is_ram_disk, temp_dir):
         mock_run.return_value = mock_result
 
         secure_delete(test_file)
-        mock_run.assert_called_once_with(
-            ["shred", "-u", str(test_file)], capture_output=True, check=True
-        )
+        mock_run.assert_called_once_with(["shred", "-u", str(test_file)], capture_output=True, check=True)
 
     @patch("nakimi.core.vault.is_ram_disk")
     @patch("subprocess.run", side_effect=FileNotFoundError())

tests/unit/test_yubikey.py

@@ -40,9 +40,7 @@ def test_check_ykman_installed_found(self):
 
             result = yk._check_ykman_installed()
             assert result is True
-            mock_run.assert_called_once_with(
-                ["ykman", "--version"], capture_output=True, check=True
-            )
+            mock_run.assert_called_once_with(["ykman", "--version"], capture_output=True, check=True)
             # Second call should use cached value
             result2 = yk._check_ykman_installed()
             assert result2 is True
@@ -149,9 +147,7 @@ def test_encrypt_age_key(self):
         # Mock is_available to return True
         with patch.object(yk, "is_available", return_value=True):
             with patch.object(yk, "_check_age_plugin_installed", return_value=True):
-                with patch.object(
-                    yk, "_get_yubikey_recipient", return_value="age1yubikey1testrecipient"
-                ):
+                with patch.object(yk, "_get_yubikey_recipient", return_value="age1yubikey1testrecipient"):
                     with patch("subprocess.run") as mock_run:
                         mock_result = Mock()
                         mock_result.stdout = b"ENCRYPTED_DATA"
@@ -176,9 +172,7 @@ def test_decrypt_age_key(self):
         # Mock is_available to return True
         with patch.object(yk, "is_available", return_value=True):
             with patch.object(yk, "_check_age_plugin_installed", return_value=True):
-                with patch.object(
-                    yk, "_get_yubikey_identity", return_value="AGE-SECRET-KEY-1TESTIDENTITY"
-                ):
+                with patch.object(yk, "_get_yubikey_identity", return_value="AGE-SECRET-KEY-1TESTIDENTITY"):
                     with patch("tempfile.NamedTemporaryFile") as mock_temp:
                         mock_file = MagicMock()
                         mock_file.name = "/tmp/mock.age"
@@ -193,9 +187,7 @@ def test_decrypt_age_key(self):
                             with patch("os.unlink") as mock_unlink:
                                 decrypted = yk.decrypt_age_key(b"ENCRYPTED_DATA")
                                 assert decrypted == "DECRYPTED_DATA"
-                                mock_file.write.assert_called_once_with(
-                                    "AGE-SECRET-KEY-1TESTIDENTITY"
-                                )
+                                mock_file.write.assert_called_once_with("AGE-SECRET-KEY-1TESTIDENTITY")
                                 mock_run.assert_called_once_with(
                                     ["age", "-d", "-i", "/tmp/mock.age"],
                                     input=b"ENCRYPTED_DATA",
@@ -246,9 +238,7 @@ def test_verify_pin_failure(self):
         with patch.object(yk, "is_available", return_value=True):
             with patch(
                 "subprocess.run",
-                side_effect=subprocess.CalledProcessError(
-                    1, "ykman piv verify-pin", stderr="Wrong PIN"
-                ),
+                side_effect=subprocess.CalledProcessError(1, "ykman piv verify-pin", stderr="Wrong PIN"),
             ):
                 result = yk.verify_pin("wrong")
                 assert result is False
@@ -294,9 +284,7 @@ def test_change_pin_failure(self):
         with patch.object(yk, "is_available", return_value=True):
             with patch(
                 "subprocess.run",
-                side_effect=subprocess.CalledProcessError(
-                    1, "ykman piv change-pin", stderr="Wrong old PIN"
-                ),
+                side_effect=subprocess.CalledProcessError(1, "ykman piv change-pin", stderr="Wrong old PIN"),
             ):
                 result = yk.change_pin("wrong", "new")
                 assert result is False