Skip to main content

A Python utility for executing SQL files against databases with support for multiple statements, comments, and pretty-printed results

Project description

splurge-sql-runner

Splurge Python SQL Runner

A Python utility for executing SQL files against databases with support for multiple statements, comments, and pretty-printed results.

Features

  • Execute SQL files with multiple statements
  • Support for various database backends (SQLite, PostgreSQL, MySQL, etc.)
  • Automatic comment removal and statement parsing
  • Pretty-printed results with tabulated output
  • Batch processing of multiple files
  • Batch SQL execution with error handling
  • Clean CLI interface with comprehensive error handling
  • Security validation for database URLs and file operations
  • Configuration management with JSON-based config files
  • Configurable logging for CLI usage (level, format, console/file output)

Installation

pip install splurge-sql-runner

CLI Usage

The main interface is through the command-line tool:

Basic Usage

# Execute a single SQL file
python -m splurge_sql_runner -c "sqlite:///database.db" -f "script.sql"

# Execute multiple SQL files using a pattern
python -m splurge_sql_runner -c "sqlite:///database.db" -p "*.sql"

# With verbose output
python -m splurge_sql_runner -c "sqlite:///database.db" -f "script.sql" -v

# Using the installed script (after pip install)
splurge-sql-runner -c "sqlite:///database.db" -f "script.sql"

# Load with a JSON config file
python -m splurge_sql_runner --config config.json -c "sqlite:///database.db" -f "script.sql"

# Output as JSON (for scripting)
python -m splurge_sql_runner -c "sqlite:///database.db" -f "script.sql" --json

# Disable emoji (useful on limited consoles)
python -m splurge_sql_runner -c "sqlite:///database.db" -f "script.sql" --no-emoji

# Continue executing statements after an error (per-file)
python -m splurge_sql_runner -c "sqlite:///database.db" -f "script.sql" --continue-on-error

Command Line Options

  • -c, --connection: Database connection string (required)

    • SQLite: sqlite:///database.db
    • PostgreSQL: postgresql://user:pass@localhost/db
    • MySQL: mysql://user:pass@localhost/db
  • -f, --file: Single SQL file to execute

  • -p, --pattern: File pattern to match multiple SQL files (e.g., "*.sql")

  • -v, --verbose: Enable verbose output

  • --debug: Enable SQLAlchemy debug mode (SQLAlchemy echo)

  • --config FILE: Path to JSON config file. Values from the file are merged with defaults and overridden by any CLI arguments.

  • --json: Output results as JSON (machine-readable).

  • --no-emoji: Replace emoji glyphs with ASCII tags in output.

  • --continue-on-error: Continue processing remaining statements after an error (default is stop on first error).

Security validation cannot be disabled. If stricter defaults block your use case, adjust SecurityConfig in your JSON config (e.g., security.max_statements_per_file, security.allowed_file_extensions, or dangerous pattern lists) and rerun.

  • --max-statements: Maximum statements per file (default: 100)

Examples

# SQLite example
python -m splurge_sql_runner -c "sqlite:///test.db" -f "setup.sql"

# PostgreSQL example
python -m splurge_sql_runner -c "postgresql://user:pass@localhost/mydb" -p "migrations/*.sql"

# MySQL example with verbose output
python -m splurge_sql_runner -c "mysql://user:pass@localhost/mydb" -f "data.sql" -v

# Process all SQL files in current directory
python -m splurge_sql_runner -c "sqlite:///database.db" -p "*.sql"

# Adjust security via config (example)
# config.json
#{
#  "security": {
#    "max_statements_per_file": 500,
#    "allowed_file_extensions": [".sql", ".ddl"]
#  }
#}
python -m splurge_sql_runner -c "sqlite:///database.db" -f "script.sql"

Security Tuning

Security is enforced by default. If your workflow requires broader allowances, tune these fields in your JSON configuration file:

  • security.max_statements_per_file (int): Maximum SQL statements allowed per file. Increase if you run bulk scripts. CLI also accepts --max-statements to override per run.
  • security.validation.max_statement_length (int): Maximum size in characters for a single statement.
  • security.allowed_file_extensions (list[str]): Allowed extensions for SQL files (e.g., [".sql", ".ddl"]).
  • security.validation.dangerous_sql_patterns (list[str]): Substrings considered dangerous in SQL. Remove or modify with caution.
  • security.validation.dangerous_path_patterns (list[str]): Path substrings that are blocked for file safety.
  • security.validation.dangerous_url_patterns (list[str]): Connection URL substrings that are blocked.

Example minimal config to relax limits:

{
  "security": {
    "max_statements_per_file": 500,
    "allowed_file_extensions": [".sql", ".ddl"],
    "validation": {
      "max_statement_length": 200000
    }
  }
}

Logging Behavior

Logging is configured via the logging section in your JSON config and is applied automatically on startup:

  • logging.level: One of DEBUG, INFO, WARNING, ERROR, CRITICAL.
  • logging.format: TEXT (human-readable) or JSON.
  • logging.enable_console: Whether to log to console.
  • logging.enable_file: Whether to log to a file.
  • logging.log_file/logging.log_dir: Exact file or directory to write logs. If only log_dir is provided, a default filename is used.
  • logging.backup_count: Number of daily-rotated log files to keep.

CLI will bootstrap minimal logging with default values and then reconfigure using your JSON config (if provided) before running.

Notes on paths and patterns

  • The --file path is expanded with ~ (home) and resolved to an absolute path before use.
  • The --pattern is expanded for ~ and matched with glob; matched files are resolved to absolute paths.

Programmatic Usage

Note: This library is primarily designed to be used via the CLI interface. The programmatic API is provided for advanced use cases and integration scenarios, but the CLI offers the most comprehensive features and best user experience.

Basic Usage

from splurge_sql_runner.database.database_client import DatabaseClient
from splurge_sql_runner.config.database_config import DatabaseConfig

client = DatabaseClient(DatabaseConfig(url="sqlite:///database.db"))

try:
    results = client.execute_batch("SELECT 1;")
    for r in results:
        print(r)
finally:
    client.close()

Advanced Usage

from splurge_sql_runner.config import AppConfig
from splurge_sql_runner.database import DatabaseClient

config = AppConfig.load("config.json")

client = DatabaseClient(config.database)

try:
    results = client.execute_batch("SELECT 1; INSERT INTO test VALUES (1);")
    for result in results:
        print(f"Statement type: {result['statement_type']}")
        if result['statement_type'] == 'fetch':
            print(f"Rows returned: {result['row_count']}")
finally:
    client.close()

Configuration

The library supports JSON-based configuration files for advanced usage:

{
    "database": {
        "url": "sqlite:///database.db",
        "connection": {
            "timeout": 30
        },
        "enable_debug": false
    },
    "security": {
        "enable_validation": true,
        "max_statements_per_file": 100
    },
    "logging": {
        "level": "INFO",
        "format": "TEXT",
        "enable_console": true,
        "enable_file": false
    }
}

SQL File Format

The tool supports SQL files with:

  • Multiple statements separated by semicolons
  • Single-line comments (-- comment)
  • Multi-line comments (/* comment */)
  • Comments within string literals are preserved

Example SQL file:

-- Create table
CREATE TABLE users (
    id INTEGER PRIMARY KEY,
    name TEXT NOT NULL
);

-- Insert data
INSERT INTO users (name) VALUES ('John');
INSERT INTO users (name) VALUES ('Jane');

-- Query data
SELECT * FROM users;

Output Format

The CLI provides formatted output showing:

  • File being processed
  • Each statement executed
  • Results in tabulated format for SELECT queries
  • Success/error status for each statement
  • Summary of files processed

Error Handling

  • Individual statement errors don't stop the entire batch
  • Failed statements are reported with error details
  • Database connections are properly cleaned up
  • Exit codes indicate success/failure
  • (Removed) Circuit breaker/retry error-recovery layers in favor of simple CLI errors
  • Security validation with configurable thresholds

License

MIT License - see LICENSE file for details.

Development

Installation for Development

# Clone the repository
git clone https://github.com/jim-schilling/splurge-sql-runner.git
cd splurge-sql-runner

# Install in development mode with all dependencies
pip install -e ".[dev]"

# Run tests
pytest -x -v

# Run linting
flake8 splurge_sql_runner/
black splurge_sql_runner/
mypy splurge_sql_runner/

Changelog

2025.3.1 (08-20-2025)

  • Test Coverage: Improved test coverage to 85% target across core modules
    • sql_helper.py: Reached 85% coverage with comprehensive edge case testing
    • database_client.py: Improved from ~71% to 77% coverage with additional test scenarios
    • cli.py: Reached 84% coverage with enhanced CLI functionality testing
  • Test Quality: Added behavior-driven tests focusing on public APIs and real functionality
    • Enhanced CTE (Common Table Expressions) parsing edge cases
    • Added DCL (Data Control Language) statement type detection
    • Improved error handling and rollback behavior testing
    • Added config file handling and security guidance output tests
    • Enhanced pattern matching and multi-file processing scenarios
  • Code Quality: Moved all import statements to top of modules where possible
    • Cleaned up inline imports in test files (test_cli.py, conftest.py, test_logging_performance.py)
    • Removed duplicate test functions that were accidentally created
    • Maintained appropriate inline imports for test setup methods where needed
  • Documentation: Created comprehensive test improvement plan in plans/improve-code-coverage.md
  • Testing: Verified all examples work correctly with enhanced test suite
    • Interactive demo functionality confirmed working
    • CLI automation tests passing
    • Database deployment script execution verified
    • Pattern matching and JSON output features tested

2025.3.0 (08-11-2025)

  • Documentation: Updated Programmatic Usage section to clarify that the library is primarily designed for CLI usage
  • Documentation: Added note explaining that programmatic API is for advanced use cases and integration scenarios
  • Documentation: Emphasized that CLI offers the most comprehensive features and best user experience
  • Breaking Changes: Unified engine abstraction replaced by DatabaseClient
  • New: Centralized configuration constants in splurge_sql_runner.config.constants
  • Improved: Security validation now uses centralized SecurityConfig from splurge_sql_runner.config.security_config
  • Code Quality: Eliminated code duplication across the codebase
  • Breaking Changes: Environment variables now use SPLURGE_SQL_RUNNER_ prefix instead of JPY_
    • JPY_DB_URLSPLURGE_SQL_RUNNER_DB_URL
    • JPY_DB_TIMEOUTSPLURGE_SQL_RUNNER_DB_TIMEOUT
    • JPY_SECURITY_ENABLEDSPLURGE_SQL_RUNNER_SECURITY_ENABLED
    • JPY_MAX_FILE_SIZE_MBSPLURGE_SQL_RUNNER_MAX_FILE_SIZE_MB
    • JPY_MAX_STATEMENTS_PER_FILESPLURGE_SQL_RUNNER_MAX_STATEMENTS_PER_FILE
    • JPY_VERBOSESPLURGE_SQL_RUNNER_VERBOSE
    • JPY_LOG_LEVELSPLURGE_SQL_RUNNER_LOG_LEVEL
    • JPY_LOG_FORMATSPLURGE_SQL_RUNNER_LOG_FORMAT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

splurge_sql_runner-2025.3.1.tar.gz (70.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

splurge_sql_runner-2025.3.1-py3-none-any.whl (45.4 kB view details)

Uploaded Python 3

File details

Details for the file splurge_sql_runner-2025.3.1.tar.gz.

File metadata

  • Download URL: splurge_sql_runner-2025.3.1.tar.gz
  • Upload date:
  • Size: 70.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.10

File hashes

Hashes for splurge_sql_runner-2025.3.1.tar.gz
Algorithm Hash digest
SHA256 b895c021d259698dd8f3c46973d34011cf26b4d0fbc533e501eb1e7783ab4720
MD5 40205d1c63acb7a30332986ba47a6651
BLAKE2b-256 bbdc8c3dedb289a8613540d3c7e05655340e2603f6114a2191fb7e92b7e6343b

See more details on using hashes here.

File details

Details for the file splurge_sql_runner-2025.3.1-py3-none-any.whl.

File metadata

File hashes

Hashes for splurge_sql_runner-2025.3.1-py3-none-any.whl
Algorithm Hash digest
SHA256 8bf731578886d3cf16c852a2512f59b53fb222eeb468e979efd861556d488f73
MD5 dd6f04bd50cb59d166bc0838dd66dc09
BLAKE2b-256 6c454fc1063ec8445985ed0a8c6580128ac917f4e040537c80e4a985d3b768e7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page