boundary-analyzer 0.2.0__py3-none-any.whl

boundary_analyzer/__main__.py

@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+from boundary_analyzer.cli import main
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

boundary_analyzer/auto_setup/__init__.py

@@ -0,0 +1,4 @@
+# auto_setup/__init__.py
+# This file makes auto_setup a proper Python package.
+# You can import from it like:
+#   from auto_setup.setup_instrumentation import detect_framework

boundary_analyzer/auto_setup/django_wrapper.py

@@ -0,0 +1,48 @@
+"""
+otel_instrumentation.py – Django / Django REST Framework
+==========================================================
+This file sets up OpenTelemetry tracing for your Django application.
+
+HOW IT WORKS:
+  - Every HTTP request Django handles becomes a span.
+  - Every ORM database query also becomes a span.
+  - All spans are sent to Jaeger.
+
+YOU DO NOT NEED TO EDIT THIS FILE.
+Add these 2 lines BEFORE django.setup() in your manage.py or wsgi.py:
+
+    from otel_instrumentation import init_tracing
+    init_tracing()
+"""
+
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.instrumentation.django import DjangoInstrumentor
+from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
+
+
+def init_tracing():
+    """
+    Call this function ONCE, before django.setup() is called.
+    """
+
+    resource = Resource.create({"service.name": "{{SERVICE_NAME}}"})
+
+    exporter = OTLPSpanExporter(
+        endpoint="http://{{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}"
+    )
+
+    provider = TracerProvider(resource=resource)
+    provider.add_span_processor(BatchSpanProcessor(exporter))
+    trace.set_tracer_provider(provider)
+
+    # Automatically trace every Django view / URL
+    DjangoInstrumentor().instrument()
+
+    # Automatically trace every ORM query
+    SQLAlchemyInstrumentor().instrument()
+
+    print(f"[OTel] Tracing enabled → Jaeger at {{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}")

boundary_analyzer/auto_setup/djangorest_wrapper.py

@@ -0,0 +1,44 @@
+"""
+otel_instrumentation.py – Django REST Framework
+=================================================
+Same setup as Django — DRF sits on top of Django,
+so we instrument at the Django level.
+
+Add these 2 lines BEFORE django.setup():
+
+    from otel_instrumentation import init_tracing
+    init_tracing()
+"""
+
+# Django REST Framework uses the same Django instrumentation
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.instrumentation.django import DjangoInstrumentor
+from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
+
+
+def init_tracing():
+    """
+    Initialize tracing for Django REST Framework.
+    DRF is built on top of Django, so we instrument Django directly.
+    """
+    resource = Resource.create({"service.name": "{{SERVICE_NAME}}"})
+
+    exporter = OTLPSpanExporter(
+        endpoint="http://{{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}"
+    )
+
+    provider = TracerProvider(resource=resource)
+    provider.add_span_processor(BatchSpanProcessor(exporter))
+    trace.set_tracer_provider(provider)
+
+    # Instrument every DRF/Django view automatically
+    DjangoInstrumentor().instrument()
+
+    # Instrument every database query automatically
+    SQLAlchemyInstrumentor().instrument()
+
+    print(f"[OTel] Tracing enabled → Jaeger at {{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}")

boundary_analyzer/auto_setup/fastapi_wrapper.py

@@ -0,0 +1,59 @@
+"""
+otel_instrumentation.py – FastAPI
+===================================
+This file sets up OpenTelemetry tracing for your FastAPI application.
+
+HOW IT WORKS:
+  - Every HTTP request becomes a span (recorded event).
+  - Every database query (via SQLAlchemy) also becomes a span.
+  - All spans are sent to Jaeger so we can see and analyze them.
+
+YOU DO NOT NEED TO EDIT THIS FILE.
+Just call init_tracing() at the top of your main.py (before app = FastAPI()).
+"""
+
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
+
+
+def init_tracing(app=None):
+    """
+    Call this function ONCE at the start of your app.
+
+    If you pass your FastAPI app object, HTTP routes will be traced automatically.
+    Example:
+        app = FastAPI()
+        init_tracing(app)
+
+    If you do not pass the app, call FastAPIInstrumentor().instrument_app(app)
+    after creating it.
+    """
+
+    # Tell Jaeger the name of this service
+    resource = Resource.create({"service.name": "{{SERVICE_NAME}}"})
+
+    # Send traces to Jaeger via gRPC
+    exporter = OTLPSpanExporter(
+        endpoint="http://{{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}"
+    )
+
+    provider = TracerProvider(resource=resource)
+    provider.add_span_processor(BatchSpanProcessor(exporter))
+    trace.set_tracer_provider(provider)
+
+    # Instrument database queries automatically
+    SQLAlchemyInstrumentor().instrument()
+
+    # Instrument the FastAPI app if provided
+    if app is not None:
+        FastAPIInstrumentor.instrument_app(app)
+    else:
+        # Will instrument the first FastAPI app created after this call
+        FastAPIInstrumentor().instrument()
+
+    print(f"[OTel] Tracing enabled → Jaeger at {{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}")

boundary_analyzer/auto_setup/flask_wrapper.py

@@ -0,0 +1,53 @@
+"""
+otel_instrumentation.py – Flask
+================================
+This file sets up OpenTelemetry tracing for your Flask application.
+
+HOW IT WORKS:
+  - Every HTTP request your app receives becomes a "span" (a recorded event).
+  - Every database query also becomes a span.
+  - All spans are sent to Jaeger so we can see them.
+
+YOU DO NOT NEED TO EDIT THIS FILE.
+Just call init_tracing() at the top of your main app file.
+"""
+
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.instrumentation.flask import FlaskInstrumentor
+from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
+
+
+def init_tracing():
+    """
+    Call this function ONCE at the start of your app.
+    It configures OpenTelemetry to send traces to Jaeger.
+    """
+
+    # 'resource' tells Jaeger which service these traces belong to
+    resource = Resource.create({"service.name": "{{SERVICE_NAME}}"})
+
+    # 'exporter' sends the traces to Jaeger over gRPC
+    exporter = OTLPSpanExporter(
+        endpoint="http://{{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}"
+    )
+
+    # 'provider' is the main tracing engine
+    provider = TracerProvider(resource=resource)
+
+    # 'processor' batches spans and sends them to the exporter
+    provider.add_span_processor(BatchSpanProcessor(exporter))
+
+    # Register the provider globally
+    trace.set_tracer_provider(provider)
+
+    # Automatically trace every Flask HTTP request
+    FlaskInstrumentor().instrument()
+
+    # Automatically trace every SQLAlchemy database query
+    SQLAlchemyInstrumentor().instrument()
+
+    print(f"[OTel] Tracing enabled → Jaeger at {{JAEGER_HOST}}:{{JAEGER_GRPC_PORT}}")