souleyez 2.43.29__py3-none-any.whl → 3.0.0__py3-none-any.whl

souleyez/security.py

@@ -27,21 +27,26 @@ def unlock_credentials_if_needed():
         return True
 
     # Need to unlock
-    click.echo(click.style("🔒 Credentials are encrypted.", fg='yellow'))
+    click.echo(click.style("🔒 Credentials are encrypted.", fg="yellow"))
 
     max_attempts = 3
     for attempt in range(max_attempts):
         try:
             password = getpass.getpass("Enter master password: ")
             if crypto.unlock(password):
-                click.echo(click.style("✅ Unlocked successfully!", fg='green'))
+                click.echo(click.style("✅ Unlocked successfully!", fg="green"))
                 return True
             else:
                 remaining = max_attempts - attempt - 1
                 if remaining > 0:
-                    click.echo(click.style(f"❌ Incorrect password. {remaining} attempts remaining.", fg='red'))
+                    click.echo(
+                        click.style(
+                            f"❌ Incorrect password. {remaining} attempts remaining.",
+                            fg="red",
+                        )
+                    )
                 else:
-                    click.echo(click.style("❌ Access denied.", fg='red'))
+                    click.echo(click.style("❌ Access denied.", fg="red"))
         except KeyboardInterrupt:
             click.echo("\n❌ Cancelled by user.")
             return False
@@ -74,11 +79,22 @@ def require_password(f):
     Returns:
         The wrapped function that requires authentication
     """
+
     @functools.wraps(f)
     def wrapper(*args, **kwargs):
         if not unlock_credentials_if_needed():
-            click.echo(click.style("\n⚠️  Authentication required to access this command.", fg='yellow'))
-            click.echo(click.style("   This command accesses sensitive data and requires master password.", fg='yellow'))
+            click.echo(
+                click.style(
+                    "\n⚠️  Authentication required to access this command.", fg="yellow"
+                )
+            )
+            click.echo(
+                click.style(
+                    "   This command accesses sensitive data and requires master password.",
+                    fg="yellow",
+                )
+            )
             sys.exit(1)
         return f(*args, **kwargs)
+
     return wrapper