Skip to main content

Shared Python utilities for AWS Lambda, CLI tools, and agents with Slack, Elasticsearch, and monitoring integrations

Project description

NUI Python Shared Utilities

PyPI version Python 3.9+ License: MIT

Production-ready shared Python utilities for AWS Lambda functions, CLI tools, and agents. Provides standardized integrations for Slack, Elasticsearch, database, and monitoring. Built and battle-tested on the NUI platform with sensible defaults configurable for any environment.

Table of Contents

Who This Package Is For

NUI Team: Drop-in utilities with NUI platform defaults pre-configured. Handles common patterns like Slack notifications, Elasticsearch logging, database connections, and CloudWatch metrics out of the box.

External Teams: Solid AWS Lambda patterns for serverless operations. Default configurations reflect NUI conventions (Elasticsearch index patterns, AWS Secrets Manager naming, Slack workspace structure) but are fully overridable via environment variables or programmatic configuration. Consider this package as production-tested reference implementations that you can adapt to your infrastructure.

Key Features

  • AWS Powertools Integration - Standardized logging, metrics, and error handling for Lambda functions
  • AWS Secrets Manager Integration - Secure credential management with caching
  • Slack Messaging - Rich formatting, threading, file uploads, and channel management
  • Elasticsearch Operations - Query builders, index management, and health monitoring
  • Database Connections - Connection pooling, automatic retries, and transaction management
  • CloudWatch Metrics - Batched publishing with custom dimensions
  • JWT Authentication - RS256 token validation for API Gateway Lambdas (lightweight, no PyJWT needed)
  • Error Handling - Intelligent retry patterns with exponential backoff
  • Timezone Utilities - Timezone handling and formatting
  • Configurable Defaults - Environment-aware configuration system

Documentation

New to this package? Start with our comprehensive guides:

Complete documentation: See docs/ for all guides and references.

Component Guides:

Quick Start

Installation

pip install nui-python-shared-utils

# With specific extras for optional dependencies
pip install nui-python-shared-utils[all]          # All integrations
pip install nui-python-shared-utils[powertools]   # AWS Powertools only
pip install nui-python-shared-utils[slack]        # Slack only
pip install nui-python-shared-utils[elasticsearch] # Elasticsearch only
pip install nui-python-shared-utils[database]     # Database only
pip install nui-python-shared-utils[jwt]          # JWT authentication only

Basic Configuration

import nui_lambda_shared_utils as nui

# Configure for your environment (optional - uses sensible defaults)
nui.configure(
    es_host="your-elasticsearch-host:9200",
    es_credentials_secret="your-es-secret-name",
    slack_credentials_secret="your-slack-secret-name",
    db_credentials_secret="your-database-secret-name"
)

# Or use environment variables:
# ES_HOST, ES_CREDENTIALS_SECRET, SLACK_CREDENTIALS_SECRET, etc.

What's Next?

After installing the package:

  1. Configuration → Set up environment variables or programmatic config (guide)
  2. AWS Setup → Configure Secrets Manager and IAM permissions (guide)
  3. Integration → Choose your integration and follow the detailed guide:
  4. Testing → Learn testing strategies (guide)

Complete documentation: docs/

Usage Examples

Below are minimal examples to get you started. For complete examples and detailed usage, see docs/getting-started/quickstart.md.

Secrets Management

from nui_lambda_shared_utils import get_secret, get_slack_credentials

# Generic secret retrieval
api_keys = get_secret("my-service/api-keys")

# Specialized getters with normalized field names
slack_creds = get_slack_credentials()  # Uses configured secret name

→ See full secrets management guide

AWS Powertools Integration

from nui_lambda_shared_utils import get_powertools_logger, powertools_handler

# Create logger with Elasticsearch-compatible formatting
logger = get_powertools_logger("my-service", level="INFO")

# Decorate Lambda handler with logging, metrics, and error handling
@powertools_handler(
    service_name="my-service",
    metrics_namespace="MyApp/Service",
    slack_alert_channel="#production-alerts"
)
@logger.inject_lambda_context
def lambda_handler(event, context):
    logger.info("Processing event", extra={"event_id": event.get("id")})
    return {"statusCode": 200, "body": "Success"}

Features:

  • ✅ Elasticsearch-compatible timestamps (2025-01-18T04:39:27.788Z)
  • ✅ Automatic Lambda context injection (function name, request ID, cold start)
  • ✅ CloudWatch metrics publishing
  • ✅ Slack error alerts with graceful degradation
  • ✅ Local development with colored logs

→ See comprehensive Powertools integration guide

Slack Integration

from nui_lambda_shared_utils import SlackClient, SlackBlockBuilder

slack = SlackClient()

# Simple message
slack.send_message(channel='#alerts', text='Service deployment complete')

# Rich formatted message with blocks
builder = SlackBlockBuilder()
blocks = builder.add_header("Alert", emoji="warning").add_section("Status", "Active").build()
slack.send_message(channel='#incidents', blocks=blocks)

→ See comprehensive Slack integration guide

Elasticsearch Operations

from nui_lambda_shared_utils import ElasticsearchClient, ESQueryBuilder

es = ElasticsearchClient()
query_builder = ESQueryBuilder()
query = query_builder.date_range("@timestamp", "now-1h", "now").term("level", "ERROR").build()
results = es.search(index="logs-*", body={"query": query})

→ See full Elasticsearch guide

Database Connections

from nui_lambda_shared_utils import DatabaseClient

db = DatabaseClient()

# Execute queries with automatic retry and connection pooling
async with db.get_connection() as conn:
    results = await conn.execute("SELECT * FROM orders WHERE status = %s", ["confirmed"])

→ See full database guide

CloudWatch Metrics

from nui_lambda_shared_utils import MetricsPublisher, track_lambda_performance

metrics = MetricsPublisher(namespace="MyApplication")

@track_lambda_performance(metrics)
def lambda_handler(event, context):
    metrics.put_metric("ProcessedItems", 150, "Count")
    return {"statusCode": 200}

→ See full metrics guide

JWT Authentication

from nui_lambda_shared_utils import require_auth, AuthenticationError

def lambda_handler(event, context):
    try:
        claims = require_auth(event)  # Validates Bearer token from Authorization header
    except AuthenticationError as e:
        return {"statusCode": 401, "body": "Unauthorized"}

    user_id = claims["sub"]
    return {"statusCode": 200, "body": f"Hello {user_id}"}

→ See JWT authentication guide

Error Handling

from nui_lambda_shared_utils import with_retry, handle_lambda_error

@handle_lambda_error
@with_retry(max_attempts=3)
def lambda_handler(event, context):
    # Your Lambda logic with automatic error handling and retries
    return {"statusCode": 200}

→ See full error handling guide

Configuration

The package supports multiple configuration methods. For detailed configuration options, see docs/getting-started/configuration.md.

Environment Variables

ES_HOST=localhost:9200                    # Elasticsearch host
ES_CREDENTIALS_SECRET=elasticsearch-creds # AWS secret name for ES
DB_CREDENTIALS_SECRET=database-creds      # AWS secret name for database
SLACK_CREDENTIALS_SECRET=slack-bot-token  # AWS secret name for Slack
AWS_REGION=us-east-1                      # AWS region

Programmatic Configuration

import nui_lambda_shared_utils as nui

nui.configure(
    es_host="localhost:9200",
    slack_credentials_secret="dev/slack-token"
)

→ See complete configuration guide

AWS Infrastructure Requirements

This package requires AWS Secrets Manager for credential storage and IAM permissions for CloudWatch metrics.

For detailed AWS setup instructions, see docs/getting-started/configuration.md#aws-infrastructure.

Quick Reference

Secrets Manager - Store credentials as JSON:

  • Elasticsearch: {"host": "...", "username": "...", "password": "..."}
  • Database: {"host": "...", "port": 3306, "username": "...", "password": "...", "database": "..."}
  • Slack: {"bot_token": "xoxb-...", "webhook_url": "..."}
  • JWT Public Key: {"TOKEN_PUBLIC_KEY": "-----BEGIN PUBLIC KEY-----\n..."}

IAM Permissions - Lambda execution role needs:

  • secretsmanager:GetSecretValue for credential access
  • cloudwatch:PutMetricData for metrics publishing

→ See complete AWS infrastructure guide

Testing

For comprehensive testing guide, see docs/development/testing.md.

# Install with dev dependencies
pip install nui-python-shared-utils[dev]

# Run all tests
pytest

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

# Run specific test categories
pytest -m unit           # Unit tests only
pytest -m integration    # Integration tests (requires AWS)

Contributing

We welcome contributions! This package currently supports MySQL/PostgreSQL, Elasticsearch, Slack, and core AWS services (Secrets Manager, CloudWatch). We're open to expanding support for additional databases (MongoDB, DynamoDB, etc.) and AWS services (SQS, SNS, EventBridge, etc.).

How to Contribute

  1. Fork the repository
  2. Create a feature branch (git checkout -b feature/amazing-feature)
  3. Commit your changes (git commit -m 'Add amazing feature')
  4. Push to the branch (git push origin feature/amazing-feature)
  5. Open a Pull Request

Contribution Ideas

  • Database integrations: MongoDB, DynamoDB, Redis, Cassandra
  • AWS services: SQS, SNS, EventBridge, Step Functions, S3, SES
  • Messaging platforms: Microsoft Teams, Discord, PagerDuty
  • Monitoring: Datadog, New Relic, Prometheus exporters
  • Search engines: OpenSearch, Algolia, Typesense
  • CLI enhancements: Additional automation commands for common workflows

See our development guide for testing patterns and architecture guidelines.

Built-in CLI Tools

The package includes slack-channel-setup - a CLI tool for automating Slack workspace channel creation from YAML configuration files. This generic tool works with any Slack workspace and can be used independently of Lambda functions.

# Install and use
pip install nui-python-shared-utils[slack]
slack-channel-setup --config channels.yaml

Documentation & Support

📚 Documentation

🔗 Links

💬 Support

License

This project is licensed under the MIT License - see the LICENSE file for details.

About NUI Markets

NUI Markets is a technology company focused on building innovative trading and marketplace platforms. This package represents our commitment to open-source tooling and production-grade infrastructure patterns for AWS Lambda development.

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

nui_python_shared_utils-1.3.0.tar.gz (154.0 kB view details)

Uploaded Source

Built Distribution

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

nui_python_shared_utils-1.3.0-py3-none-any.whl (69.0 kB view details)

Uploaded Python 3

File details

Details for the file nui_python_shared_utils-1.3.0.tar.gz.

File metadata

  • Download URL: nui_python_shared_utils-1.3.0.tar.gz
  • Upload date:
  • Size: 154.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.3

File hashes

Hashes for nui_python_shared_utils-1.3.0.tar.gz
Algorithm Hash digest
SHA256 d053b929444b65dd0a6324284ce8e0a33f555873c98c1dc16f90c1e0ecded302
MD5 d2655f54236f6a699a5e8a7cceb40094
BLAKE2b-256 b1fba1def8a1e909aebfe8c30f480e4cfb1c30e0445a3c870c4a7d9feb7ccad9

See more details on using hashes here.

File details

Details for the file nui_python_shared_utils-1.3.0-py3-none-any.whl.

File metadata

File hashes

Hashes for nui_python_shared_utils-1.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 5c55f00f0a2b75b16ae7e938c3d5b76773981ce66cc9b9e43697ac6d42ac64d4
MD5 daaa4733768b5f204504f6843e2f545e
BLAKE2b-256 9d993a89f2f07187771267298656577e434eae4f29734ee3fd26199de1fadb84

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