swapnil-tracepilot 0.1.0__tar.gz

swapnil_tracepilot-0.1.0/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: swapnil-tracepilot
+Version: 0.1.0
+Summary: TracePilot OpenTelemetry SDK for Python
+Author: TracePilot Team
+Author-email: support@tracepilot.ai
+Requires-Python: >=3.9,<4.0
+Classifier: Programming Language :: Python :: 3
+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: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: opentelemetry-api (>=1.20.0,<2.0.0)
+Requires-Dist: opentelemetry-exporter-otlp (>=1.20.0,<2.0.0)
+Requires-Dist: opentelemetry-instrumentation
+Requires-Dist: opentelemetry-instrumentation-requests
+Requires-Dist: opentelemetry-sdk (>=1.20.0,<2.0.0)

swapnil_tracepilot-0.1.0/pyproject.toml

@@ -0,0 +1,31 @@
+[tool.poetry]
+name = "swapnil-tracepilot"
+version = "0.1.0"
+description = "TracePilot OpenTelemetry SDK for Python"
+authors = ["TracePilot Team <support@tracepilot.ai>"]
+packages = [
+    { include = "tracepilot" }
+]
+
+[tool.poetry.dependencies]
+python = "^3.9"
+opentelemetry-api = "^1.20.0"
+opentelemetry-sdk = "^1.20.0"
+opentelemetry-exporter-otlp = "^1.20.0"
+opentelemetry-instrumentation = "*"
+opentelemetry-instrumentation-requests = "*"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+tracepilot = "tracepilot.instrument:main"
+
+[tool.poetry.group.dev.dependencies]
+responses = "^0.23.3"
+pytest = "^8.0.0"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]

swapnil_tracepilot-0.1.0/tracepilot/__init__.py

@@ -0,0 +1,112 @@
+import os
+import sys
+import traceback
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.instrumentation.requests import RequestsInstrumentor
+
+_provider = None
+
+def init():
+    """
+    Initialize the TracePilot OpenTelemetry exporter.
+    Requires TRACEPILOT_TOKEN to be set.
+    """
+    project_id = os.getenv("TRACEPILOT_TOKEN")
+    
+    # Silent disable pattern
+    if not project_id:
+        return trace.get_tracer("tracepilot")
+        
+    ingestor_url = os.getenv("TRACEPILOT_COLLECTOR_URL", "https://ingestor.tracepilot.io")
+
+    # Set up the resource (identifies the service)
+    resource = Resource(attributes={
+        "service.name": os.getenv("SERVICE_NAME", "python-service"),
+        "tracepilot.project.id": project_id
+    })
+
+    global _provider
+    _provider = TracerProvider(resource=resource)
+    
+    # Configure OTLP HTTP Exporter
+    exporter = OTLPSpanExporter(
+        endpoint=f"{ingestor_url}/v1/traces",
+        headers={"x-tracepilot-project-id": project_id}
+    )
+    
+    # Use BatchSpanProcessor for performance
+    _provider.add_span_processor(BatchSpanProcessor(exporter))
+    trace.set_tracer_provider(_provider)
+
+    # Base instrumentation
+    RequestsInstrumentor().instrument()
+    
+    # Auto-instrument web frameworks based on what's installed
+    _try_instrument_fastapi()
+    _try_instrument_django()
+    _try_instrument_flask()
+    
+    # Setup global unhandled exception hook
+    _setup_excepthook()
+
+    return trace.get_tracer("tracepilot")
+
+def _try_instrument_fastapi():
+    try:
+        from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+        FastAPIInstrumentor().instrument()
+    except ImportError:
+        pass  # FastAPI not installed - skip silently
+
+def _try_instrument_django():
+    try:
+        from opentelemetry.instrumentation.django import DjangoInstrumentor
+        DjangoInstrumentor().instrument()
+    except ImportError:
+        pass
+
+def _try_instrument_flask():
+    try:
+        from opentelemetry.instrumentation.flask import FlaskInstrumentor
+        FlaskInstrumentor().instrument()
+    except ImportError:
+        pass
+
+def _setup_excepthook():
+    _original_excepthook = sys.excepthook
+    def _tracepilot_excepthook(exc_type, exc_value, exc_traceback):
+        tracer = trace.get_tracer("tracepilot.exception")
+        with tracer.start_as_current_span("Unhandled Exception") as span:
+            span.record_exception(exc_value)
+            span.set_status(trace.StatusCode.ERROR, str(exc_value))
+            # Include traceback as a span attribute
+            tb_str = "".join(traceback.format_exception(exc_type, exc_value, exc_traceback))
+            span.set_attribute("exception.traceback", tb_str)
+        
+        # Flush the provider to ensure the span is sent before crashing
+        _provider.force_flush()
+        
+        # Call original
+        _original_excepthook(exc_type, exc_value, exc_traceback)
+
+    sys.excepthook = _tracepilot_excepthook
+
+    # Auto-instrumentation hooks
+    try:
+        from opentelemetry.instrumentation.requests import RequestsInstrumentor
+        RequestsInstrumentor().instrument()
+    except ImportError:
+        pass # Requests instrumentor not installed
+
+    try:
+        from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+        # FastAPI requires the app instance to instrument, typically done via FastAPIInstrumentor.instrument_app(app)
+        # We can expose this helper if needed, or if instrument() works globally, do it here.
+    except ImportError:
+        pass
+
+    return trace.get_tracer("tracepilot")

swapnil_tracepilot-0.1.0/tracepilot/instrument.py

@@ -0,0 +1,24 @@
+import os
+import sys
+from tracepilot import init
+
+def main():
+    """
+    Zero-config entry point.
+    Usage: tracepilot python app.py
+    """
+    if len(sys.argv) < 2:
+        print("Usage: tracepilot python app.py")
+        print("Or:    tracepilot uvicorn main:app")
+        sys.exit(1)
+
+    # Initialize TracePilot instrumentation globally
+    init()
+
+    # Execute the target application
+    # sys.argv[1] is the command (e.g., 'python', 'uvicorn', 'gunicorn')
+    # os.execvp replaces the current process with the target process
+    os.execvp(sys.argv[1], sys.argv[1:])
+
+if __name__ == "__main__":
+    main()