aegis-stack 0.2.0rc2__py3-none-any.whl

aegis/templates/cookiecutter-aegis-project/hooks/post_gen_project.py

@@ -0,0 +1,214 @@
+#!/usr/bin/env python
+import os
+import shutil
+import sys
+from pathlib import Path
+
+from jinja2 import Environment
+
+# Add aegis-stack to path to import shared post-generation tasks
+# This hook runs from within the generated project, so we need to go up to aegis-stack root
+AEGIS_STACK_ROOT = Path(__file__).parent.parent.parent.parent.parent
+sys.path.insert(0, str(AEGIS_STACK_ROOT))
+
+from aegis.core.post_gen_tasks import run_post_generation_tasks  # noqa: E402
+
+PROJECT_DIRECTORY = os.path.realpath(os.path.curdir)
+
+
+def remove_file(filepath):
+    """Removes a file from the generated project."""
+    full_path = os.path.join(PROJECT_DIRECTORY, filepath)
+    if os.path.exists(full_path):
+        os.remove(full_path)
+
+
+def remove_dir(dirpath):
+    """Removes a directory from the generated project."""
+    full_path = os.path.join(PROJECT_DIRECTORY, dirpath)
+    if os.path.exists(full_path):
+        shutil.rmtree(full_path)
+
+
+def process_j2_templates():
+    """
+    Process all .j2 template files in the generated project.
+    Renders them with cookiecutter context and removes the .j2 originals.
+    Returns list of output files that were created.
+    """
+    # Cookiecutter context variables - these template strings are processed
+    # by cookiecutter before this hook runs, so they contain actual values
+    context = {
+        "cookiecutter": {
+            "project_name": "{{ cookiecutter.project_name }}",
+            "project_slug": "{{ cookiecutter.project_slug }}",
+            "project_description": "{{ cookiecutter.project_description }}",
+            "author_name": "{{ cookiecutter.author_name }}",
+            "author_email": "{{ cookiecutter.author_email }}",
+            "version": "{{ cookiecutter.version }}",
+            "python_version": "{{ cookiecutter.python_version }}",
+            "include_redis": "{{ cookiecutter.include_redis }}",
+            "include_scheduler": "{{ cookiecutter.include_scheduler }}",
+            "include_worker": "{{ cookiecutter.include_worker }}",
+            "include_database": "{{ cookiecutter.include_database }}",
+            "include_cache": "{{ cookiecutter.include_cache }}",
+            "include_auth": "{{ cookiecutter.include_auth }}",
+            "include_ai": "{{ cookiecutter.include_ai }}",
+        }
+    }
+
+    # Find all .j2 files in the project
+    project_path = Path(PROJECT_DIRECTORY)
+    j2_files = list(project_path.rglob("*.j2"))
+    processed_files = []
+
+    for j2_file in j2_files:
+        # Read the template content
+        with open(j2_file, encoding="utf-8") as f:
+            template_content = f.read()
+
+        # Create Jinja2 environment and render the template
+        env = Environment()
+        template = env.from_string(template_content)
+        rendered_content = template.render(context)
+
+        # Write the rendered content to the final file (without .j2 extension)
+        output_file = j2_file.with_suffix("")
+        with open(output_file, "w", encoding="utf-8") as f:
+            f.write(rendered_content)
+
+        # Remove the original .j2 file
+        j2_file.unlink()
+
+        # Track processed files for later reporting
+        processed_files.append(output_file)
+
+        # Ensure file ends with newline
+        if not rendered_content.endswith("\n"):
+            with open(output_file, "a", encoding="utf-8") as f:
+                f.write("\n")
+
+    return processed_files
+
+
+def main():
+    """
+    Runs the post-generation cleanup to remove files for unselected
+    components and process template files.
+    """
+    # Process .j2 template files first
+    processed_files = process_j2_templates()
+
+    # Remove components not selected
+    if "{{ cookiecutter.include_scheduler }}" != "yes":
+        # Remove scheduler-specific files
+        remove_file("app/entrypoints/scheduler.py")
+        remove_dir("app/components/scheduler")
+        remove_file("tests/components/test_scheduler.py")
+        remove_file("docs/components/scheduler.md")
+        # Remove scheduler API endpoints (empty when scheduler not included)
+        remove_file("app/components/backend/api/scheduler.py")
+        remove_file("tests/api/test_scheduler_endpoints.py")
+        # Remove scheduler card (should not display when scheduler not included)
+        remove_file("app/components/frontend/dashboard/cards/scheduler_card.py")
+        # Remove scheduler-specific test file
+        remove_file("tests/services/test_scheduled_task_manager.py")
+
+    # Remove scheduler service if scheduler backend is memory
+    # The service is only useful when we can persist to a database
+    if "{{ cookiecutter.scheduler_backend }}" == "memory":
+        remove_dir("app/services/scheduler")
+        remove_file("app/cli/tasks.py")
+        # Remove scheduler API endpoints (empty when using memory backend)
+        remove_file("app/components/backend/api/scheduler.py")
+        remove_file("tests/api/test_scheduler_endpoints.py")
+        # Remove task manager test (only exists with persistence)
+        remove_file("tests/services/test_scheduled_task_manager.py")
+
+    if "{{ cookiecutter.include_worker }}" != "yes":
+        # Remove worker-specific files
+        remove_dir("app/components/worker")
+        remove_file("app/cli/load_test.py")
+        remove_file("app/services/load_test.py")
+        remove_file("app/services/load_test_models.py")
+        remove_file("tests/services/test_load_test_models.py")
+        remove_file("tests/services/test_load_test_service.py")
+        remove_file("tests/services/test_worker_health_registration.py")
+        # Remove worker API endpoints (empty when worker not included)
+        remove_file("app/components/backend/api/worker.py")
+        remove_file("tests/api/test_worker_endpoints.py")
+        # Remove worker card (should not display when worker not included)
+        remove_file("app/components/frontend/dashboard/cards/worker_card.py")
+
+    # Remove shared component integration tests only when BOTH scheduler AND worker disabled
+    if (
+        "{{ cookiecutter.include_scheduler }}" != "yes"
+        and "{{ cookiecutter.include_worker }}" != "yes"
+    ):
+        remove_file("tests/services/test_component_integration.py")
+        remove_file("tests/services/test_health_logic.py")
+
+    if "{{ cookiecutter.include_database }}" != "yes":
+        remove_file("app/core/db.py")
+
+    if "{{ cookiecutter.include_cache }}" != "yes":
+        # remove_file("app/services/cache_service.py")
+        pass  # Placeholder for cache component
+
+    # Remove services not selected
+    if "{{ cookiecutter.include_auth }}" != "yes":
+        # Remove auth service files
+        remove_dir("app/components/backend/api/auth")
+        remove_file("app/models/user.py")
+        remove_dir("app/services/auth")
+        remove_file("app/core/security.py")
+        # Remove auth CLI
+        remove_file("app/cli/auth.py")
+        # Remove auth-related tests if they exist
+        remove_file("tests/api/test_auth_endpoints.py")
+        remove_file("tests/services/test_auth_service.py")
+        remove_file("tests/services/test_auth_integration.py")
+        remove_file("tests/models/test_user.py")
+
+    if "{{ cookiecutter.include_ai }}" != "yes":
+        # Remove AI service files
+        remove_dir("app/components/backend/api/ai")
+        remove_dir("app/services/ai")
+        # Remove AI CLI and rendering
+        remove_file("app/cli/ai.py")
+        remove_file("app/cli/ai_rendering.py")
+        remove_file("app/cli/marko_terminal_renderer.py")
+        # Remove AI-related tests if they exist
+        remove_file("tests/api/test_ai_endpoints.py")
+        remove_file("tests/services/test_conversation_persistence.py")
+        remove_file("tests/cli/test_ai_rendering.py")
+        remove_file("tests/cli/test_conversation_memory.py")
+        # Remove AI test directory (contains test_service.py, test_health.py, etc.)
+        remove_dir("tests/services/ai")
+
+    # Clean up empty docs/components directory if no components selected
+    if (
+        "{{ cookiecutter.include_scheduler }}" != "yes"
+        and "{{ cookiecutter.include_worker }}" != "yes"
+        and "{{ cookiecutter.include_database }}" != "yes"
+        and "{{ cookiecutter.include_cache }}" != "yes"
+    ):
+        remove_dir("docs/components")
+
+    # Remove Alembic directory if auth service not included
+    if "{{ cookiecutter.include_auth }}" != "yes":
+        remove_dir("alembic")
+
+    # Print only templates that survived cleanup
+    for file_path in processed_files:
+        if file_path.exists():
+            print(f"Processed template: {file_path.name}")
+
+    # Complete project setup: dependencies, env file, migrations, formatting
+    # Use shared post-generation tasks module (also used by Copier)
+    include_auth = "{{ cookiecutter.include_auth }}" == "yes"
+    run_post_generation_tasks(Path(PROJECT_DIRECTORY), include_auth=include_auth)
+
+
+if __name__ == "__main__":
+    main()

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.dockerignore

@@ -0,0 +1,71 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Virtual environments - don't mount host venv into container
+.venv/
+.env/
+venv/
+ENV/
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+.nox/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db
+
+# Logs
+*.log
+
+# Temporary files
+*.tmp
+*.temp
+
+# Git
+.git/
+.gitignore
+
+# Documentation build
+site/
+docs/_build/
+
+# Cache
+.cache/
+.mypy_cache/
+.ruff_cache/

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.env.example.j2

@@ -0,0 +1,130 @@
+# Aegis Stack Environment Variables Example
+# Copy this file to .env and customize for your environment
+
+# =============================================================================
+# APPLICATION SETTINGS
+# =============================================================================
+
+# Environment: dev, qa, stg, prod
+APP_ENV=dev
+
+# Port to run the webserver on
+PORT=8000
+
+# Logging level: DEBUG, INFO, WARNING, ERROR
+LOG_LEVEL=INFO
+
+# Auto-reload server on code changes (development only)
+AUTO_RELOAD=true
+
+# =============================================================================
+# DOCKER SETTINGS
+# =============================================================================
+
+# Docker image tag (used by docker-compose)
+AEGIS_STACK_TAG=aegis-stack:latest
+
+# Version tag for builds
+AEGIS_STACK_VERSION=dev
+
+# Environment file to use (defaults to .env)
+# AEGIS_STACK_ENV_FILE=.env
+
+# =============================================================================
+# COMPONENT SETTINGS
+# =============================================================================
+
+{% if cookiecutter.include_database == "yes" %}
+# Database settings
+# DATABASE_URL=sqlite:///./data/app.db  # SQLite database (default)
+# DATABASE_ENGINE_ECHO=false  # Set to true to log all SQL queries (debugging)
+{% endif %}
+
+{% if cookiecutter.include_redis == "yes" or cookiecutter.include_cache == "yes" %}
+# Redis settings
+# For Docker: redis://redis:6379 (default)
+# For local development: redis://localhost:6379
+REDIS_URL=redis://redis:6379
+
+# Override for local CLI commands (when not running in Docker)
+# Uncomment for local development:
+REDIS_URL_LOCAL=redis://localhost:6379
+# REDIS_HOST=redis
+# REDIS_PORT=6379
+# REDIS_PASSWORD=
+
+# Redis connection settings (for high-load scenarios)
+# REDIS_CONN_TIMEOUT=5  # Connection timeout in seconds
+# REDIS_CONN_RETRIES=5  # Number of connection retry attempts
+# REDIS_CONN_RETRY_DELAY=1  # Delay between retries in seconds
+{% endif %}
+
+{% if cookiecutter.include_scheduler == "yes" %}
+# Scheduler settings
+# SCHEDULER_TIMEZONE=UTC
+# SCHEDULER_MAX_WORKERS=10
+{% endif %}
+
+# API Keys and Secrets (add as needed)
+# SECRET_KEY=your-super-secret-key-here
+# API_KEY=your-api-key-here
+
+{% if cookiecutter.include_ai == "yes" %}
+# =============================================================================
+# AI SERVICE CONFIGURATION
+# =============================================================================
+
+# AI Service Settings
+AI_ENABLED=true
+AI_PROVIDER=public  # Primary provider: openai, anthropic, google, groq, mistral, cohere, public
+AI_MODEL=auto  # Model to use (PUBLIC provider uses available free models)
+AI_TEMPERATURE=0.7  # Response creativity (0.0-2.0)
+AI_MAX_TOKENS=1000  # Maximum response length
+AI_TIMEOUT_SECONDS=30.0  # Request timeout
+
+# Conversation Management
+AI_MAX_CONVERSATION_LENGTH=50  # Max messages per conversation
+AI_CONVERSATION_TIMEOUT_HOURS=24  # Auto-cleanup old conversations
+
+# Provider API Keys (optional - many offer free tiers)
+# ⚡ PUBLIC provider works immediately without API keys (currently set as default)
+# Get your API keys from:
+# - OpenAI: https://platform.openai.com/api-keys
+# - Anthropic: https://console.anthropic.com/
+# - Google: https://aistudio.google.com/app/apikey
+# - Groq: https://console.groq.com/keys (FREE - Recommended for getting started)
+# - Mistral: https://console.mistral.ai/api-keys/
+# - Cohere: https://dashboard.cohere.com/api-keys
+
+# OPENAI_API_KEY=sk-...
+# ANTHROPIC_API_KEY=sk-ant-...
+# GOOGLE_API_KEY=AI...
+# GROQ_API_KEY=gsk_...  # FREE tier available!
+# MISTRAL_API_KEY=...
+# COHERE_API_KEY=...
+
+# Provider-Specific Notes:
+# - Public: Works immediately without API keys (uses free endpoints, actual model varies)
+# - Groq: Excellent free tier with fast inference (recommended for testing)
+# - Google: Generous free tier with Gemini Flash model
+# - OpenAI: Requires paid account but excellent quality
+# - Anthropic: Requires paid account, very good for reasoning
+# - Mistral: Good open source models, some free options
+# - Cohere: Enterprise focused, limited free tier
+{% endif %}
+
+# =============================================================================
+# DEVELOPMENT OVERRIDES
+# =============================================================================
+
+# Worker auto-reload settings
+# WORKER_WATCH=false  # Disable worker auto-reload (useful for debugging)
+
+{%- if cookiecutter.include_worker == "yes" %}
+# Worker health check settings
+# WORKER_HEALTH_CHECK_INTERVAL=15  # Health check interval in seconds (default: 15)
+{%- endif %}
+
+# Override for local development
+# Use this to point to local services instead of Docker services
+# DATABASE_HOST=host.docker.internal  # For connecting to host DB from container

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.gitignore

@@ -0,0 +1,131 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*.so
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Database files and data directory
+data/
+*.db
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS generated files
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/Dockerfile

@@ -0,0 +1,53 @@
+FROM python:3.11-slim
+
+# Install system dependencies and clean up
+RUN apt-get update -y && \
+    apt-get install -y --no-install-recommends \
+      build-essential \
+      ca-certificates \
+      curl \
+    && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+# Install uv with specific version for reproducibility
+RUN pip install --no-cache-dir uv==0.4.18
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONPATH=/code \
+    UV_CACHE_DIR=/tmp/uv-cache
+
+WORKDIR /code
+
+# Copy dependency files and README (needed by pyproject.toml)
+COPY pyproject.toml uv.lock README.md /code/
+
+# Install dependencies
+RUN uv sync --all-extras && \
+    rm -rf /tmp/uv-cache
+
+# Copy application code
+COPY . /code
+
+# Make entrypoint executable
+RUN chmod +x /code/scripts/entrypoint.sh
+
+# Set default port as an environment variable
+ARG PORT=8000
+ENV PORT=$PORT
+
+# Expose the port
+EXPOSE $PORT
+
+# Add health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+  CMD curl -f http://localhost:$PORT/health || exit 1
+
+
+# Labels for better container management
+LABEL maintainer="contact@aegis-stack.dev" \
+      version="0.1.0" \
+      description="Aegis Stack - Production-ready async Python foundation"
+
+ENTRYPOINT ["uv", "run", "/code/scripts/entrypoint.sh"]