feat: add Python linting support with pylint and black

bin/actions/lint.py

@@ -8,54 +8,62 @@ from pathlib import Path
 
 # Import helper functions
 sys.path.append(os.path.join(os.path.dirname(os.path.dirname(__file__))))
-from helpers.functions import printfe, ensure_dependencies, command_exists
+from helpers.functions import printfe, command_exists
 
 DOTFILES_ROOT = os.path.expanduser("~/.dotfiles")
 
+
 def lint_ansible(fix=False):
     """Run ansible-lint on Ansible files"""
     ansible_dir = os.path.join(DOTFILES_ROOT, "config/ansible")
-    
+
     if not os.path.isdir(ansible_dir):
         printfe("yellow", "No ansible directory found at config/ansible")
         return 0
-    
+
     # Find all YAML files in the ansible directory
     yaml_files = []
     for ext in [".yml", ".yaml"]:
         yaml_files.extend(list(Path(ansible_dir).glob(f"**/*{ext}")))
-    
+
     if not yaml_files:
         printfe("yellow", "No Ansible files found in config/ansible to lint")
         return 0
-    
+
     if not command_exists("ansible-lint"):
-        printfe("red", "ansible-lint is not installed. Please install it with pip or your package manager.")
+        printfe(
+            "red",
+            "ansible-lint is not installed. Please install it with pip or your package manager.",
+        )
         return 1
-    
+
     printfe("blue", f"Running ansible-lint{' with auto-fix' if fix else ''}...")
     files_to_lint = [str(f) for f in yaml_files]
-    
+
     command = ["ansible-lint"]
     if fix:
         command.append("--fix")
     command.extend(files_to_lint)
-    
+
     result = subprocess.run(command)
     return result.returncode
 
+
 def lint_nix():
     """Run nixfmt on Nix files"""
     nix_files = list(Path(DOTFILES_ROOT).glob("**/*.nix"))
-    
+
     if not nix_files:
         printfe("yellow", "No Nix files found to lint")
         return 0
-    
+
     if not command_exists("nixfmt"):
-        printfe("red", "nixfmt is not installed. Please install it with nix-env or your package manager.")
+        printfe(
+            "red",
+            "nixfmt is not installed. Please install it with nix-env or your package manager.",
+        )
         return 1
-    
+
     printfe("blue", "Running nixfmt...")
     exit_code = 0
     for nix_file in nix_files:
@@ -63,33 +71,92 @@ def lint_nix():
         result = subprocess.run(["nixfmt", str(nix_file)])
         if result.returncode != 0:
             exit_code = 1
-    
+
     return exit_code
 
+
+def lint_python(fix=False):
+    """Run pylint and black on Python files"""
+    python_files = list(Path(DOTFILES_ROOT).glob("**/*.py"))
+
+    if not python_files:
+        printfe("yellow", "No Python files found to lint")
+        return 0
+
+    exit_code = 0
+
+    # Check for pylint
+    if command_exists("pylint"):
+        printfe("blue", "Running pylint...")
+        files_to_lint = [str(f) for f in python_files]
+        result = subprocess.run(["pylint"] + files_to_lint)
+        if result.returncode != 0:
+            exit_code = 1
+    else:
+        printfe("yellow", "pylint is not installed. Skipping Python linting.")
+
+    # Check for black
+    if command_exists("black"):
+        printfe(
+            "blue", f"Running black{'--check' if not fix else ''} on Python files..."
+        )
+        black_args = ["black"]
+        if not fix:
+            black_args.append("--check")
+        black_args.extend([str(f) for f in python_files])
+
+        result = subprocess.run(black_args)
+        if result.returncode != 0:
+            exit_code = 1
+    else:
+        printfe("yellow", "black is not installed. Skipping Python formatting.")
+
+    if not command_exists("pylint") and not command_exists("black"):
+        printfe(
+            "red",
+            "Neither pylint nor black is installed. Install them with pip: pip install pylint black",
+        )
+        return 1
+
+    return exit_code
+
+
 def main():
     parser = argparse.ArgumentParser(description="Run linters on dotfiles")
     parser.add_argument("--ansible", action="store_true", help="Run only ansible-lint")
     parser.add_argument("--nix", action="store_true", help="Run only nixfmt")
-    parser.add_argument("--fix", action="store_true", help="Auto-fix issues where possible (for ansible-lint)")
+    parser.add_argument(
+        "--python", action="store_true", help="Run only Python linters (pylint, black)"
+    )
+    parser.add_argument(
+        "--fix", action="store_true", help="Auto-fix issues where possible"
+    )
     args = parser.parse_args()
-    
-    # If no specific linter is specified, run both
-    run_ansible = args.ansible or not (args.ansible or args.nix)
-    run_nix = args.nix or not (args.ansible or args.nix)
-    
+
+    # If no specific linter is specified, run all
+    run_ansible = args.ansible or not (args.ansible or args.nix or args.python)
+    run_nix = args.nix or not (args.ansible or args.nix or args.python)
+    run_python = args.python or not (args.ansible or args.nix or args.python)
+
     exit_code = 0
-    
+
     if run_ansible:
         ansible_result = lint_ansible(fix=args.fix)
         if ansible_result != 0:
             exit_code = ansible_result
-            
+
     if run_nix:
         nix_result = lint_nix()
         if nix_result != 0:
             exit_code = nix_result
-    
+
+    if run_python:
+        python_result = lint_python(fix=args.fix)
+        if python_result != 0:
+            exit_code = python_result
+
     return exit_code
 
+
 if __name__ == "__main__":
     sys.exit(main())