Support .wizrc to set default command line args

Author: ddrscott

.gitignore

@@ -5,3 +5,4 @@ dist/
 *.pyc
 __pycache__/
 .DS_Store
+.wizrc

.wizrc.example

@@ -0,0 +1,6 @@
+# Example .wizrc configuration
+files: &files
+  exclude: ["*.md", "*.log"]
+  
+prompt:
+  <<: *files

README.md

@@ -30,6 +30,7 @@ export ANTHROPIC_API_KEY=your_api_key_here
 - 🖼️ Includes image attachment capabilities for visual context
 - 🚫 Exclude files via regex patterns to fine-tune your magical selection
 - 📋 Preview files that would be included before spending API tokens
+- 📝 Supports `.wizrc` configuration file for persistent command-line arguments
 
 ## 🧙 Installation
 
@@ -70,9 +71,38 @@ For permanent enchantment, add this to your shell profile (.bashrc, .zshrc, etc.
 
 > 🔴 **Note**: The `ANTHROPIC_API_KEY` environment variable is **required**. The tool will not function without it properly set. If you encounter errors about unauthorized access or missing API keys, please verify this variable is correctly set.
 
+## 📝 Configuration with .wizrc
+
+You can store command configurations in a `.wizrc` YAML file in your project directory:
+
+```yaml
+# Example .wizrc file
+files:
+  exclude: ["*test*"]
+  image: ["screenshot.png"]
+  max_tokens: 80000
+
+# Example .wizrc configuration
+files: &files
+  exclude: ["*.log", "build", "dist"]
+  
+prompt:
+  <<: *files
+```
+
+The tool will automatically load this configuration and apply it to corresponding commands. The example above shows:
+- Setting exclusion patterns to ignore specific files 
+- Using YAML anchors (`&files`) to define common configurations
+- Inheriting settings between commands with YAML merge keys (`<<: *files`)
+- Configuring command-specific options like max_tokens
+
+Each top-level key in the YAML corresponds to a subcommand, with nested options matching the command's options. All configuration is completely optional, and command-line arguments will override settings in `.wizrc` when provided.
+
+The `.wizrc` file is automatically excluded from git via `.gitignore`.
+
 ## 🪄 Usage
 
-Wizard Prompt CLI offers three main incantations:
+Wizard Prompt CLI offers three main incantations and supports configuration via a `.wizrc` file:
 
 ### 📝 Prompt Command
 
@@ -297,4 +327,4 @@ echo $env:ANTHROPIC_API_KEY
 
 <p align="center">
   <i>✨ If this magical tool helped you, consider giving it a star! ✨</i>
-</p>
+</p>

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wizard-prompt-cli"
-version = "0.3.1"
+version = "0.4.0"
 description = "A CLI tool for building and running magical LLM prompts"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -10,6 +10,7 @@ authors = [
 dependencies = [
     "anthropic>=0.49.0",
     "click>=8.1.8",
+    "pyyaml>=6.0.2",
     "rich>=13.9.4",
 ]
 

tests/test_cli.py

@@ -1,7 +1,8 @@
 import pytest
+import os
 from unittest.mock import patch, MagicMock, mock_open
 from click.testing import CliRunner
-from wiz import cli, files, apply
+from wiz import cli, files, apply, load_wizrc_config
 
 def test_files_command(runner):
     """Test the 'files' command."""
@@ -70,4 +71,72 @@ def test_prompt_command(runner):
             assert result.exit_code == 0
             # Verify reply was called with the question
             mock_reply.assert_called_once()
-            assert 'How can I improve this code?' in mock_reply.call_args[0][0]
+            assert 'How can I improve this code?' in mock_reply.call_args[0][0]
+
+def test_load_wizrc_config():
+    """Test loading configuration from .wizrc file."""
+    wizrc_yaml = """
+    prompt:
+      exclude: ["*.log", "*.tmp"]
+      max_tokens: 80000
+      file: ["test.py", "main.py"]
+    files:
+      exclude: ["*.md"]
+    """
+    expected_config = {
+        'prompt': {
+            'exclude': ['*.log', '*.tmp'],
+            'max_tokens': 80000,
+            'file': ['test.py', 'main.py']
+        },
+        'files': {
+            'exclude': ['*.md']
+        }
+    }
+    
+    with patch('os.path.exists', return_value=True):
+        with patch('builtins.open', mock_open(read_data=wizrc_yaml)):
+            with patch('wiz.console.print'):  # Suppress console output during test
+                config = load_wizrc_config()
+                assert config == expected_config
+
+def test_load_wizrc_empty_file():
+    """Test loading from empty .wizrc file."""
+    with patch('os.path.exists', return_value=True):
+        with patch('builtins.open', mock_open(read_data='')):
+            with patch('wiz.console.print'):  # Suppress console output during test
+                config = load_wizrc_config()
+                assert config == {}
+
+def test_load_wizrc_file_not_exists():
+    """Test when .wizrc file doesn't exist."""
+    with patch('os.path.exists', return_value=False):
+        config = load_wizrc_config()
+        assert config == {}
+
+def test_cli_with_wizrc_config(runner):
+    """Test CLI with .wizrc configuration."""
+    
+    # Test that configuration from .wizrc is applied
+    with patch('wiz.load_wizrc_config') as mock_load_config:
+        # Mock the config that would be loaded from .wizrc
+        mock_load_config.return_value = {
+            'prompt': {
+                'exclude': ['*.log'],
+                'max_tokens': 80000,
+                'file': ['test.py']
+            }
+        }
+        
+        with patch('wiz.reply') as mock_reply:
+            mock_reply.return_value = "Response content"
+            
+            # Mock file operations
+            with patch('builtins.open', mock_open()):
+                # Run the command
+                with patch('wiz.console.print'):  # Suppress console output
+                    result = runner.invoke(cli, ['prompt', 'How can I improve this code?'])
+                    
+                    assert result.exit_code == 0
+                    # Verify config was loaded
+                    mock_load_config.assert_called_once()