Metadata-Version: 2.4
Name: elexon-bmrs
Version: 0.3.0
Summary: Python client library for the Elexon BMRS (Balancing Mechanism Reporting Service) API
Author-email: Benjamin Watts <benjamin.watts@example.com>
Maintainer-email: Benjamin Watts <benjamin.watts@example.com>
Project-URL: Homepage, https://github.com/BenjaminWatts/balancing
Project-URL: Documentation, https://benjaminwatts.github.io/balancing/
Project-URL: Repository, https://github.com/BenjaminWatts/balancing
Project-URL: Issues, https://github.com/BenjaminWatts/balancing/issues
Keywords: elexon,bmrs,energy,electricity,uk,api,client,balancing-mechanism,power-market,energy-data
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Natural Language :: English
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Classifier: Topic :: Scientific/Engineering
Classifier: Typing :: Typed
Requires-Python: >=3.8
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: requests>=2.31.0
Requires-Dist: python-dateutil>=2.8.2
Requires-Dist: pydantic>=2.0.0
Provides-Extra: dev
Requires-Dist: pytest>=7.4.0; extra == "dev"
Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
Requires-Dist: black>=23.7.0; extra == "dev"
Requires-Dist: flake8>=6.1.0; extra == "dev"
Requires-Dist: mypy>=1.5.0; extra == "dev"
Requires-Dist: isort>=5.12.0; extra == "dev"
Dynamic: license-file

# Elexon BMRS Python Client

A Python client library for accessing the Elexon BMRS (Balancing Mechanism Reporting Service) API. This library provides easy access to UK electricity market data including generation, demand, pricing, and system information.

[![Python Version](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/)
[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)

## Features

- 🎯 **99% Type Coverage** - 284/287 endpoints return fully typed Pydantic models
- 🔌 **287 API endpoints** - Complete coverage of all BMRS data
- 🔑 **API key optional** (but recommended for higher rate limits)
- 📊 **280 Pydantic models** - Auto-generated with comprehensive validation
- 🧩 **39 field mixins** - Eliminate code duplication (~364+ lines saved)
- ⚡ **Full IDE autocomplete** - Type-safe access to all response fields
- 🛡️ Built-in error handling with Pydantic validation
- 📝 **Complete type hints** - Works with mypy, pyright, and IDE type checking
- 🤖 Auto-generated from OpenAPI specification - Always up-to-date
- 🧪 Comprehensive test coverage
- ⚠️ **Clear warnings** for 3 untyped endpoints (2 XML, 1 deprecated)
- 📚 **Complete documentation** with examples for all endpoints

## Installation

### From PyPI

```bash
pip install elexon-bmrs
```

### From Source

```bash
# Clone the repository
git clone https://github.com/benjaminwatts/elexon-bmrs.git
cd elexon-bmrs

# Install in development mode
pip install -e .

# Or install with dev dependencies
pip install -e ".[dev]"
```

### Verify Installation

```bash
python -c "from elexon_bmrs import BMRSClient; print('✓ Installation successful!')"
```

## API Key Setup

### Do I Need an API Key?

The API key is **optional but strongly recommended** by Elexon. While the API can function without a key, using one provides significant benefits:

✅ **Higher rate limits** - Avoid hitting rate limits during data retrieval  
✅ **Better performance** - Improved reliability and response times  
✅ **Usage tracking** - Monitor your API usage and get support  
✅ **Production ready** - Essential for any production application

### Getting Your API Key

1. Visit the [Elexon Portal](https://www.elexonportal.co.uk/)
2. Register for a free account
3. Navigate to the API section to generate your key
4. Copy your API key and keep it secure

⚠️ **Note:** If you initialize the client without an API key, you'll see a warning message. This is expected behavior.

## Quick Start

### Basic Usage (with API Key - Recommended)

```python
from elexon_bmrs import BMRSClient
from datetime import date

# Initialize the client with your API key (recommended)
client = BMRSClient(api_key="your-api-key-here")

# Get generation data by fuel type (returns Dict[str, Any])
generation_data = client.get_generation_by_fuel_type(
    from_date="2024-01-01",
    to_date="2024-01-02"
)

print(generation_data)
```

### Using Without an API Key (Not Recommended)

```python
from elexon_bmrs import BMRSClient

# You can use the API without a key, but you'll see a warning
client = BMRSClient()  # ⚠️ Warning will be logged

# API calls work but with lower rate limits
demand_data = client.get_system_demand(
    from_date="2024-01-01",
    to_date="2024-01-02"
)
```

### Type-Safe Usage (Recommended - v0.3.0+)

The SDK now provides **full type safety** with Pydantic models for 275/287 endpoints!

```python
from elexon_bmrs import BMRSClient
from elexon_bmrs.generated_models import (
    DynamicData_ResponseWithMetadata,
    AbucDatasetRow_DatasetResponse
)

# Initialize client
client = BMRSClient(api_key="your-api-key-here")

# ✨ Fully typed responses with automatic Pydantic parsing!
result: DynamicData_ResponseWithMetadata = client.get_balancing_dynamic(
    bmUnit="2__CARR-1",
    snapshotAt="2024-01-01T12:00:00Z"
)

# Full IDE autocomplete on all fields! 🎉
for item in result.data:
    print(f"Dataset: {item.dataset}")
    print(f"BMU: {item.bmUnit}")
    print(f"Value: {item.value}")
    print(f"Time: {item.time}")
    # ↑ IDE knows all available fields - autocomplete everywhere!

# List responses are also typed
demand_list = client.get_demand_outturn_summary(
    from_="2024-01-01",
    to_="2024-01-02"
)
# Returns: List[RollingSystemDemand] - each item is a typed model

for demand in demand_list:
    print(f"Date: {demand.settlement_date}, Demand: {demand.demand} MW")
    # ↑ Full type safety and validation!
```

### Context Manager

```python
from elexon_bmrs import BMRSClient

# Use as a context manager to ensure proper cleanup
with BMRSClient(api_key="your-api-key-here") as client:
    # Get system demand data
    demand = client.get_system_demand(
        from_date="2024-01-01",
        to_date="2024-01-02"
    )
    print(demand)
```

## API Documentation

### Authentication

The API key is optional but **strongly recommended** by Elexon. 

#### With API Key (Recommended)
```python
client = BMRSClient(api_key="your-api-key-here")
```

#### Without API Key (Lower Rate Limits)
```python
# A warning will be logged recommending you use an API key
client = BMRSClient()
```

Get your free API key at: [Elexon Portal](https://www.elexonportal.co.uk/)

### Available Methods

The client provides **287 methods** for accessing BMRS data. All methods are auto-generated from the OpenAPI specification with full type hints and documentation.

#### Verify All Endpoints

```python
from elexon_bmrs import BMRSClient

client = BMRSClient()
methods = [m for m in dir(client) if m.startswith('get_')]
print(f"Total endpoints: {len(methods)}")  # 287

# Get help for any method
help(client.get_balancing_dynamic)
```

#### Complete Documentation

- **[All Endpoints](https://benjaminwatts.github.io/balancing/api/all-endpoints)** - Complete endpoint reference
- **[Method Reference](https://benjaminwatts.github.io/balancing/api/method-reference)** - Organized by category
- **[Client API](https://benjaminwatts.github.io/balancing/api/client)** - Auto-generated docs
- **[Full Documentation](https://benjaminwatts.github.io/balancing/)** - Complete documentation site

#### Quick Examples

##### Market Index Data

```python
# Get market index data
market_index = client.get_market_index(
    settlement_date="2024-01-01",
    settlement_period=10  # Optional: specific half-hour period (1-50)
)
```

#### Generation Data

```python
# Get generation by fuel type
generation = client.get_generation_by_fuel_type(
    from_date="2024-01-01",
    to_date="2024-01-02",
    settlement_period_from=1,  # Optional
    settlement_period_to=48     # Optional
)

# Get actual generation output by BMU
actual_generation = client.get_actual_generation_output(
    settlement_date="2024-01-01",
    settlement_period=10
)

# Get wind generation forecast
wind_forecast = client.get_wind_generation_forecast(
    from_date="2024-01-01",
    to_date="2024-01-07"
)
```

#### Demand Data

```python
# Get system demand
demand = client.get_system_demand(
    from_date="2024-01-01",
    to_date="2024-01-02"
)

# Get forecast demand
forecast = client.get_forecast_demand(
    from_date="2024-01-01",
    to_date="2024-01-07"
)
```

#### System Frequency

```python
# Get system frequency data
frequency = client.get_system_frequency(
    from_date="2024-01-01",
    to_date="2024-01-02"
)
```

#### Pricing Data

```python
# Get system prices (buy and sell)
prices = client.get_system_prices(
    settlement_date="2024-01-01",
    settlement_period=10
)

# Get imbalance prices
imbalance = client.get_imbalance_prices(
    from_date="2024-01-01",
    to_date="2024-01-02"
)
```

#### Balancing Services

```python
# Get balancing services volume
balancing = client.get_balancing_services_volume(
    settlement_date="2024-01-01"
)
```

## Settlement Periods

The UK electricity market operates in half-hour settlement periods:
- Each day has 48 settlement periods (or 50 on clock change days)
- Period 1: 00:00-00:30
- Period 2: 00:30-01:00
- ...
- Period 48: 23:30-00:00

## Rate Limiting

The Elexon BMRS API implements rate limiting to ensure fair usage and system stability. While specific rate limits are not publicly documented, users are expected to use the API responsibly. Excessive requests may result in access restrictions or revocation of API keys.

### How Rate Limiting Works

When you exceed the rate limit, the API returns a `429 Too Many Requests` status code. The client automatically detects this and raises a `RateLimitError` with optional retry timing information.

### Handling Rate Limits

```python
from elexon_bmrs import BMRSClient
from elexon_bmrs.exceptions import RateLimitError
import time

client = BMRSClient(api_key="your-api-key")

try:
    data = client.get_system_demand(
        from_date="2024-01-01",
        to_date="2024-01-02"
    )
except RateLimitError as e:
    # The exception includes retry_after in seconds (if provided by API)
    if e.retry_after:
        print(f"Rate limited. Waiting {e.retry_after} seconds...")
        time.sleep(e.retry_after)
    else:
        print("Rate limited. Implementing backoff...")
        time.sleep(60)  # Default wait time
```

### Retry Logic with Exponential Backoff

For production applications, implement exponential backoff to handle rate limits gracefully:

```python
import time
from elexon_bmrs import BMRSClient
from elexon_bmrs.exceptions import RateLimitError

def fetch_with_retry(client, max_retries=3):
    """Fetch data with exponential backoff on rate limits."""
    for attempt in range(max_retries):
        try:
            return client.get_system_demand(
                from_date="2024-01-01",
                to_date="2024-01-02"
            )
        except RateLimitError as e:
            if attempt == max_retries - 1:
                raise  # Re-raise on final attempt
            
            # Use API's retry_after or exponential backoff
            wait_time = e.retry_after if e.retry_after else (2 ** attempt)
            print(f"Rate limited. Waiting {wait_time} seconds...")
            time.sleep(wait_time)

client = BMRSClient(api_key="your-api-key")
data = fetch_with_retry(client)
```

### Best Practices to Avoid Rate Limits

1. **Implement Exponential Backoff**: Wait progressively longer between retries
2. **Cache Responses**: Store frequently accessed data locally to reduce API calls
3. **Batch Requests**: Use date ranges instead of individual requests when possible
4. **Use Streaming Endpoints**: For real-time data, use `/stream` endpoints instead of polling
5. **Monitor Your Usage**: Track API calls and implement throttling in your application
6. **Avoid Parallel Requests**: Sequential requests are more rate-limit friendly
7. **Use Appropriate Intervals**: Don't poll more frequently than your data actually updates

### Rate Limit Headers

When a rate limit is encountered, the API may return:
- `Retry-After`: Number of seconds to wait before retrying
- The client automatically extracts this value and includes it in `RateLimitError.retry_after`

### Example: Production-Ready Rate Limit Handling

```python
import time
import logging
from elexon_bmrs import BMRSClient
from elexon_bmrs.exceptions import RateLimitError, APIError

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

class RateLimitedClient:
    def __init__(self, api_key, max_retries=3, base_delay=1):
        self.client = BMRSClient(api_key=api_key)
        self.max_retries = max_retries
        self.base_delay = base_delay
    
    def request_with_backoff(self, func, *args, **kwargs):
        """Execute any client method with automatic retry on rate limits."""
        for attempt in range(self.max_retries):
            try:
                return func(*args, **kwargs)
            except RateLimitError as e:
                if attempt == self.max_retries - 1:
                    logger.error("Max retries exceeded for rate limit")
                    raise
                
                # Use API's suggestion or exponential backoff
                if e.retry_after:
                    wait_time = e.retry_after
                else:
                    wait_time = self.base_delay * (2 ** attempt)
                
                logger.warning(f"Rate limited. Retry {attempt + 1}/{self.max_retries} "
                             f"after {wait_time}s")
                time.sleep(wait_time)
            except APIError as e:
                logger.error(f"API error: {e}")
                raise

# Usage
rate_limited_client = RateLimitedClient(api_key="your-api-key")
data = rate_limited_client.request_with_backoff(
    rate_limited_client.client.get_system_demand,
    from_date="2024-01-01",
    to_date="2024-01-02"
)
```

For more details on responsible API usage, see the [Elexon API Terms](https://www.elexon.co.uk/bsc/data/balancing-mechanism-reporting-agent/copyright-licence-use-bmrs-api/).

## Type Coverage

### 99% Fully Typed (284/287 endpoints) 🎉

The SDK provides **comprehensive type safety** with Pydantic models:

| Type Category | Count | Percentage | Status |
|---------------|-------|------------|--------|
| **Fully Typed (Pydantic)** | 284 | 98.9% | ✅ |
| Untyped (XML Endpoints) | 2 | 0.7% | ⚠️ Returns XML |
| Untyped (Deprecated/404) | 1 | 0.3% | ⚠️ Doesn't exist |

**Typed Response Types:**
- **Single Model**: 181 endpoints → Return `Model_Response`
- **List of Models**: 91 endpoints → Return `List[Model]`
- **List of Strings**: 4 endpoints → Return `List[str]`
- **Manual Models**: 8 endpoints → Created from actual API responses

### Untyped Endpoints (Only 3!)

Only 3 endpoints return `Dict[str, Any]` - and these CANNOT be typed:

```python
# ⚠️ The 3 untyped endpoints show clear warnings
result = client.get_interop_message_list_retrieval(...)
# IDE shows: "WARNING: This endpoint returns untyped Dict[str, Any]"
# Reason: Returns XML, not JSON
```

**The 3 Untyped Endpoints:**
1. `get_interop_message_list_retrieval()` - Returns XML (not JSON)
2. `get_interop_message_detail_retrieval()` - Returns XML (not JSON)
3. `get_lolpdrm_forecast_evolution()` - Returns 404 (deprecated endpoint)

**See [UNTYPED_ENDPOINTS.md](UNTYPED_ENDPOINTS.md)** for complete details.

### Type Safety Benefits

```python
# ✅ Typed endpoint - Great developer experience
result = client.get_balancing_dynamic(...)
for item in result.data:  
    print(item.dataset, item.value)  # ← Full autocomplete!

# ⚠️ Untyped endpoint - Manual handling required  
result = client.get_health()
if 'status' in result:  # ← No autocomplete
    print(result['status'])
```

## Error Handling

The library provides specific exceptions for different error scenarios:

```python
from elexon_bmrs import BMRSClient
from elexon_bmrs.exceptions import (
    APIError,
    AuthenticationError,
    RateLimitError,
    ValidationError
)

try:
    client = BMRSClient(api_key="invalid-key")
    data = client.get_system_demand(
        from_date="2024-01-01",
        to_date="2024-01-02"
    )
except AuthenticationError:
    print("Invalid API key")
except RateLimitError as e:
    print(f"Rate limit exceeded. Retry after {e.retry_after} seconds")
except ValidationError as e:
    print(f"Invalid input: {e}")
except APIError as e:
    print(f"API error: {e}")
```

## Development

### Setting Up Development Environment

```bash
# Clone the repository
git clone https://github.com/benjaminwatts/elexon-bmrs.git
cd elexon-bmrs

# Create a virtual environment
python -m venv venv
source venv/bin/activate  # On Windows: venv\Scripts\activate

# Install development dependencies
pip install -r requirements-dev.txt
```

### Code Generation from OpenAPI Specification

This project includes tools to automatically generate and validate client code from the official BMRS API OpenAPI specification:

```bash
# Download the latest API specification
python tools/download_schema.py

# Generate client methods from the spec
python tools/generate_client.py

# Generate Pydantic models from the spec (280 models!)
python tools/generate_models.py

# Or generate everything at once
make generate-all

# Validate existing client against the spec
python tools/validate_client.py
```

**Benefits:**
- ✓ Automatically stay up-to-date with API changes
- ✓ Ensure complete coverage of all endpoints
- ✓ Type-safe responses with Pydantic models
- ✓ Maintain consistency with official API documentation
- ✓ Detect breaking changes early
- ✓ IDE autocomplete for all response fields

See [tools/README.md](tools/README.md) for detailed documentation on code generation.

### Using Generated Pydantic Models (v0.3.0+)

The SDK includes **280 auto-generated Pydantic models** from the OpenAPI spec with **39 field mixins** for code reuse:

```python
# Import generated models
from elexon_bmrs.generated_models import (
    AbucDatasetRow_DatasetResponse,
    DynamicData_ResponseWithMetadata,
    RollingSystemDemand,
    SystemFrequency_DatasetResponse,
    # ... and 276 more!
)

# ✨ Models are automatically returned - no manual parsing needed!
abuc_response = client.get_datasets_abuc(
    publishDateTimeFrom="2024-01-01T00:00:00Z",
    publishDateTimeTo="2024-01-02T00:00:00Z"
)

# Response is already a Pydantic model with full validation
for row in abuc_response.data:
    # IDE autocomplete shows: dataset, publishTime, psrType, quantity, etc.
    print(f"PSR: {row.psrType}, Quantity: {row.quantity}, Year: {row.year}")
    # ↑ All fields validated and typed!

# Type checking works with mypy/pyright
def analyze_abuc(data: AbucDatasetRow_DatasetResponse) -> float:
    """Type-safe function with full IDE support."""
    return sum(row.quantity for row in data.data)
```

**Benefits:**
- ✅ **95% of endpoints** return typed models automatically
- ✅ **Zero manual parsing** required
- ✅ **Automatic validation** with clear error messages
- ✅ **Full IDE autocomplete** on all response fields
- ✅ **Type checking** with mypy/pyright

See [examples/typed_usage.py](examples/typed_usage.py) for comprehensive examples.

### Running Tests

```bash
# Run all tests
pytest

# Run with coverage
pytest --cov=elexon_bmrs --cov-report=html

# Run specific test file
pytest tests/test_client.py
```

### Code Quality

```bash
# Format code with black
black elexon_bmrs tests

# Sort imports
isort elexon_bmrs tests

# Run linter
flake8 elexon_bmrs tests

# Type checking
mypy elexon_bmrs
```

## Examples

See the `examples/` directory for more detailed examples:

- `basic_usage.py` - Basic client usage examples
- `advanced_usage.py` - Advanced features and error handling
- `data_analysis.py` - Example data analysis workflows

## Contributing

Contributions are welcome! Please feel free to submit a Pull Request. For major changes, please open an issue first to discuss what you would like to change.

### Development Workflow

1. Fork the repository
2. Create your feature branch (`git checkout -b feature/amazing-feature`)
3. Install dev dependencies (`pip install -e ".[dev]"`)
4. Make your changes
5. Run tests and checks (`make pre-release`)
6. Commit your changes (`git commit -m 'Add some amazing feature'`)
7. Push to the branch (`git push origin feature/amazing-feature`)
8. Open a Pull Request

### For Maintainers: PyPI Distribution

**🚀 Automated (Recommended)**: Use GitHub Actions for automated publishing
- Quick [5-minute setup](GITHUB_ACTIONS_SETUP.md)
- Publishes automatically on GitHub Release
- Secure with trusted publishing (no API tokens!)
- Full CI/CD with testing

**🛠️ Manual**: Traditional command-line publishing
- Detailed [manual guide](PYPI_DISTRIBUTION.md)
- Use when you need direct control

```bash
# Automated (GitHub Actions) - Recommended
git tag v0.1.0 && git push origin v0.1.0
# Create GitHub Release → Automatic publish! 🎉

# Manual (Traditional)
make pre-release    # Run all checks
make upload-test    # Test on TestPyPI
make upload         # Publish to PyPI
```

## Resources

- [Elexon BMRS Website](https://www.bmreports.com/)
- [Elexon Portal](https://www.elexonportal.co.uk/) - Register for API key
- [API Documentation](https://bmrs.elexon.co.uk/api-documentation/guidance) - Official API docs
- [API Base URL](https://data.elexon.co.uk/bmrs/api/v1) - Production endpoint

## License

This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

## Disclaimer

This is an unofficial client library and is not affiliated with or endorsed by Elexon Limited. Use of the BMRS API is subject to Elexon's terms and conditions.

## Changelog

See [CHANGELOG.md](CHANGELOG.md) for a detailed list of changes between versions.

### Latest Changes

See [CHANGELOG.md](CHANGELOG.md) for detailed version history.

**Version 0.3.0** - Full Type Safety
- 🎯 95% type coverage (275/287 endpoints fully typed)
- 📦 280 Pydantic models with 39 field mixins
- ⚠️ Clear warnings for 12 untyped endpoints
- 🔧 Automatic response parsing
- 📚 Comprehensive documentation

**Version 0.2.0** - Enhanced Endpoints
- Added BOALF, PN, BOD, B1610 support
- Fixed API endpoint URLs
- Improved error handling

**Version 0.1.0** - Initial Release
- Core BMRS API client functionality
- 287 endpoint coverage
- Basic type support

## Support

For issues, questions, or contributions, please visit the [GitHub repository](https://github.com/benjaminwatts/elexon-bmrs).

