usetraceforge 1.0.0__tar.gz

usetraceforge-1.0.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: usetraceforge
+Version: 1.0.0
+Summary: TraceForge Python SDK for unhandled panic logging
+Home-page: https://github.com/khushalp2004/TraceForge
+Author: Khushal Patil
+Requires-Dist: requests>=2.31.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: summary

usetraceforge-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

usetraceforge-1.0.0/setup.py

@@ -0,0 +1,13 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="usetraceforge",
+    version="1.0.0",
+    packages=find_packages(),
+    install_requires=[
+        "requests>=2.31.0"
+    ],
+    author="Khushal Patil",
+    description="TraceForge Python SDK for unhandled panic logging",
+    url="https://github.com/khushalp2004/TraceForge"
+)

usetraceforge-1.0.0/traceforge/__init__.py

@@ -0,0 +1,3 @@
+from .client import init, capture_exception, capture_message
+
+__all__ = ["init", "capture_exception", "capture_message"]

usetraceforge-1.0.0/traceforge/client.py

@@ -0,0 +1,101 @@
+import os
+import traceback
+import requests
+import threading
+import sys
+
+_config = {}
+_setup_handshake_sent = False
+
+def init(api_key: str, endpoint: str = "http://localhost:3001/ingest", auto_capture: bool = True):
+    if not api_key:
+        raise ValueError("TraceForge.init() failed: Missing API Key.")
+    
+    _config["api_key"] = api_key
+    _config["endpoint"] = endpoint
+
+    # Send setup handshake
+    threading.Thread(target=_send_setup_handshake, daemon=True).start()
+
+    if auto_capture:
+        _setup_auto_capture()
+
+def _get_setup_endpoint() -> str:
+    base = _config.get("endpoint", "").rstrip("/")
+    if base.endswith("/setup"):
+        return base
+    return f"{base}/setup"
+
+def _send_setup_handshake():
+    global _setup_handshake_sent
+    if _setup_handshake_sent or not _config.get("api_key"):
+        return
+    
+    try:
+        res = requests.post(
+            _get_setup_endpoint(),
+            json={"environment": "python"},
+            headers={
+                "Content-Type": "application/json",
+                "X-Traceforge-Key": _config["api_key"]
+            },
+            timeout=5
+        )
+        if res.status_code < 400:
+            _setup_handshake_sent = True
+    except Exception:
+        pass
+
+def _send_payload(payload):
+    api_key = _config.get("api_key")
+    endpoint = _config.get("endpoint")
+    if not api_key or not endpoint:
+        return
+
+    headers = {
+        "X-Traceforge-Key": api_key,
+        "Content-Type": "application/json"
+    }
+    
+    try:
+        requests.post(endpoint, json=payload, headers=headers, timeout=5)
+    except Exception as e:
+        pass
+
+def capture_exception(exc: Exception, tags: dict = None, payload: dict = None):
+    if not _config.get("api_key"):
+        return
+
+    message = str(exc) or exc.__class__.__name__
+    
+    try:
+        stack_trace = "".join(traceback.format_exception(type(exc), exc, exc.__traceback__))
+    except Exception:
+        stack_trace = traceback.format_exc()
+
+    event = {
+        "message": message,
+        "stackTrace": stack_trace,
+        "environment": "python",
+        "tags": tags or {},
+        "payload": payload or {}
+    }
+    
+    # We use threading to send this synchronously in the background
+    threading.Thread(target=_send_payload, args=(event,), daemon=True).start()
+
+def capture_message(message: str, level: str = "info", context: dict = None):
+    # Backward compatibility
+    pass
+
+def _setup_auto_capture():
+    # Hook into sys.excepthook to catch all unhandled exceptions
+    original_excepthook = sys.excepthook
+
+    def traceforge_excepthook(exc_type, exc_value, exc_traceback):
+        if exc_value:
+            capture_exception(exc_value, tags={"uncaught": "true"})
+        # Call the original excepthook to not break default behavior
+        original_excepthook(exc_type, exc_value, exc_traceback)
+
+    sys.excepthook = traceforge_excepthook

usetraceforge-1.0.0/traceforge/integrations/__init__.py

@@ -0,0 +1 @@
+# Integrations for TraceForge

usetraceforge-1.0.0/traceforge/integrations/django.py

@@ -0,0 +1,21 @@
+from traceforge.client import capture_exception
+
+class TraceForgeMiddleware:
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        return self.get_response(request)
+
+    def process_exception(self, request, exception):
+        # Capture the exception asynchronously
+        payload = {
+            "url": request.build_absolute_uri(),
+            "method": request.method,
+        }
+        
+        # We don't want to block the Django response
+        capture_exception(exception, tags={"framework": "django"}, payload=payload)
+        
+        # Return None to let Django's default exception handling continue
+        return None

usetraceforge-1.0.0/traceforge/integrations/fastapi.py

@@ -0,0 +1,32 @@
+from traceforge.client import capture_exception
+
+def init(app):
+    """
+    Registers a global exception handler for FastAPI/Starlette applications.
+    """
+    try:
+        from fastapi import Request
+        from starlette.responses import JSONResponse
+        from starlette.exceptions import HTTPException
+    except ImportError:
+        return
+
+    @app.exception_handler(Exception)
+    async def traceforge_exception_handler(request: Request, exc: Exception):
+        # Don't capture standard HTTPExceptions (like 404s) as crashes
+        if isinstance(exc, HTTPException) and exc.status_code < 500:
+            raise exc
+
+        payload = {
+            "url": str(request.url),
+            "method": request.method,
+        }
+        
+        capture_exception(exc, tags={"framework": "fastapi"}, payload=payload)
+        
+        # Re-raise the exception to let FastAPI handle it natively
+        # Or return a 500 JSON response if we prefer to mask it
+        return JSONResponse(
+            status_code=500,
+            content={"detail": "Internal Server Error"}
+        )

usetraceforge-1.0.0/usetraceforge.egg-info/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: usetraceforge
+Version: 1.0.0
+Summary: TraceForge Python SDK for unhandled panic logging
+Home-page: https://github.com/khushalp2004/TraceForge
+Author: Khushal Patil
+Requires-Dist: requests>=2.31.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: summary

usetraceforge-1.0.0/usetraceforge.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+setup.py
+traceforge/__init__.py
+traceforge/client.py
+traceforge/integrations/__init__.py
+traceforge/integrations/django.py
+traceforge/integrations/fastapi.py
+usetraceforge.egg-info/PKG-INFO
+usetraceforge.egg-info/SOURCES.txt
+usetraceforge.egg-info/dependency_links.txt
+usetraceforge.egg-info/requires.txt
+usetraceforge.egg-info/top_level.txt

usetraceforge-1.0.0/usetraceforge.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

usetraceforge-1.0.0/usetraceforge.egg-info/requires.txt

@@ -0,0 +1 @@
+requests>=2.31.0

usetraceforge-1.0.0/usetraceforge.egg-info/top_level.txt

@@ -0,0 +1 @@
+traceforge