django-devbar 0.1.0__tar.gz → 0.1.2__tar.gz

{django_devbar-0.1.0 → django_devbar-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-devbar
-Version: 0.1.0
+Version: 0.1.2
 Summary: Lightweight performance devbar for Django
 Keywords: django,debug,performance,monitoring,development
 Author: amureki
@@ -31,9 +31,9 @@ Description-Content-Type: text/markdown
 
 # django-devbar
 
-Lightweight performance devbar for Django. Shows DB query count, query duration, and response time.
+Lightweight performance devbar for Django. Shows DB query count, query duration, application time, and detects duplicate queries with visual severity indicators.
 
-![devbar example](docs/devbar-example.svg)
+![devbar example](https://raw.githubusercontent.com/amureki/django-devbar/main/docs/devbar-example.svg)
 
 ## Installation
 
@@ -60,8 +60,19 @@ DEVBAR_POSITION = "top-left"
 # Show HTML overlay (default: DEBUG)
 DEVBAR_SHOW_BAR = True
 
-# Add X-DevBar-* response headers (default: False)
+# Add DevBar-* response headers (default: False)
 DEVBAR_SHOW_HEADERS = True
+
+# Enable console logging for duplicate queries (default: True)
+DEVBAR_ENABLE_CONSOLE = True
+
+# Performance thresholds for warning/critical levels (defaults shown)
+DEVBAR_THRESHOLDS = {
+    "time_warning": 500,    # ms
+    "time_critical": 1500,  # ms
+    "count_warning": 20,    # queries
+    "count_critical": 50,   # queries
+}
 ```
 
 ## Response Headers
@@ -74,9 +85,9 @@ When `DEVBAR_SHOW_HEADERS = True`, performance metrics are added as HTTP respons
 
 Headers included:
 
-| Header | Description |
-|--------|-------------|
-| `X-DevBar-Query-Count` | Number of database queries executed |
-| `X-DevBar-Query-Duration` | Total time spent in database queries (ms) |
-| `X-DevBar-Response-Time` | Total request-response cycle time (ms) |
-| `X-DevBar-Has-Duplicates` | Present (value `1`) if duplicate queries detected |
+| Header | Example | Description |
+|--------|---------|-------------|
+| `DevBar-Query-Count` | `12` | Number of database queries executed |
+| `DevBar-DB-Time` | `87ms` | Total time spent in database queries |
+| `DevBar-App-Time` | `41ms` | Application time (total time minus DB time) |
+| `DevBar-Duplicates` | `3` | Number of duplicate queries detected (only present if duplicates found) |

{django_devbar-0.1.0 → django_devbar-0.1.2}/README.md

@@ -1,8 +1,8 @@
 # django-devbar
 
-Lightweight performance devbar for Django. Shows DB query count, query duration, and response time.
+Lightweight performance devbar for Django. Shows DB query count, query duration, application time, and detects duplicate queries with visual severity indicators.
 
-![devbar example](docs/devbar-example.svg)
+![devbar example](https://raw.githubusercontent.com/amureki/django-devbar/main/docs/devbar-example.svg)
 
 ## Installation
 
@@ -29,8 +29,19 @@ DEVBAR_POSITION = "top-left"
 # Show HTML overlay (default: DEBUG)
 DEVBAR_SHOW_BAR = True
 
-# Add X-DevBar-* response headers (default: False)
+# Add DevBar-* response headers (default: False)
 DEVBAR_SHOW_HEADERS = True
+
+# Enable console logging for duplicate queries (default: True)
+DEVBAR_ENABLE_CONSOLE = True
+
+# Performance thresholds for warning/critical levels (defaults shown)
+DEVBAR_THRESHOLDS = {
+    "time_warning": 500,    # ms
+    "time_critical": 1500,  # ms
+    "count_warning": 20,    # queries
+    "count_critical": 50,   # queries
+}
 ```
 
 ## Response Headers
@@ -43,9 +54,9 @@ When `DEVBAR_SHOW_HEADERS = True`, performance metrics are added as HTTP respons
 
 Headers included:
 
-| Header | Description |
-|--------|-------------|
-| `X-DevBar-Query-Count` | Number of database queries executed |
-| `X-DevBar-Query-Duration` | Total time spent in database queries (ms) |
-| `X-DevBar-Response-Time` | Total request-response cycle time (ms) |
-| `X-DevBar-Has-Duplicates` | Present (value `1`) if duplicate queries detected |
+| Header | Example | Description |
+|--------|---------|-------------|
+| `DevBar-Query-Count` | `12` | Number of database queries executed |
+| `DevBar-DB-Time` | `87ms` | Total time spent in database queries |
+| `DevBar-App-Time` | `41ms` | Application time (total time minus DB time) |
+| `DevBar-Duplicates` | `3` | Number of duplicate queries detected (only present if duplicates found) |

{django_devbar-0.1.0 → django_devbar-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "django-devbar"
-version = "0.1.0"
+version = "0.1.2"
 description = "Lightweight performance devbar for Django"
 readme = "README.md"
 license = "MIT"

{django_devbar-0.1.0 → django_devbar-0.1.2}/src/django_devbar/__init__.py

@@ -1,4 +1,4 @@
 from .middleware import DevBarMiddleware
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"
 __all__ = ["DevBarMiddleware"]

{django_devbar-0.1.0 → django_devbar-0.1.2}/src/django_devbar/conf.py

@@ -7,6 +7,13 @@ POSITIONS = {
     "top-left": "top:0;left:0",
 }
 
+DEFAULT_THRESHOLDS = {
+    "time_warning": 500,
+    "time_critical": 1500,
+    "count_warning": 20,
+    "count_critical": 50,
+}
+
 
 def get_position():
     key = getattr(settings, "DEVBAR_POSITION", "bottom-right")
@@ -19,3 +26,14 @@ def get_show_bar():
 
 def get_show_headers():
     return getattr(settings, "DEVBAR_SHOW_HEADERS", False)
+
+
+def get_enable_console():
+    return getattr(settings, "DEVBAR_ENABLE_CONSOLE", True)
+
+
+def get_thresholds():
+    user_thresholds = getattr(settings, "DEVBAR_THRESHOLDS", {})
+    final = DEFAULT_THRESHOLDS.copy()
+    final.update(user_thresholds)
+    return final

django_devbar-0.1.2/src/django_devbar/middleware.py

@@ -0,0 +1,153 @@
+import json
+import re
+from contextlib import ExitStack
+from time import perf_counter
+
+from django.db import connections
+
+from . import tracker
+from .conf import (
+    get_position,
+    get_show_bar,
+    get_show_headers,
+    get_thresholds,
+    get_enable_console,
+)
+
+STYLE_BLOCK = """<style>
+#django-devbar {
+    position: fixed; %s; z-index: 99999;
+    font-family: -apple-system, system-ui, sans-serif;
+    font-size: 11px; font-weight: 500;
+    padding: 4px 8px; border-radius: 4px;
+    backdrop-filter: blur(4px); -webkit-backdrop-filter: blur(4px);
+    box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+    transition: all 0.2s ease;
+    cursor: default;
+    line-height: 1.3;
+    background: rgba(20, 20, 20, 0.85);
+    color: #e5e5e5;
+}
+#django-devbar.level-warn { border-left: 3px solid #f59e0b; }
+#django-devbar.level-crit { border-left: 3px solid #dc2626; }
+#django-devbar span { opacity: 0.7; margin-right: 2px; }
+#django-devbar strong { opacity: 1; font-weight: 600; margin-right: 6px; }
+</style>"""
+
+BAR_TEMPLATE = """<div id="django-devbar" class="level-%s">
+<span>db</span> <strong>%.0fms</strong>
+<span>py</span> <strong>%.0fms</strong>
+<span>count</span> <strong>%d</strong>%s
+</div>"""
+
+SCRIPT_TEMPLATE = """<script>
+(function() {
+    const stats = %s;
+    if (stats.duplicates && stats.duplicates.length > 0) {
+        console.groupCollapsed(`⚠️ Django DevBar: Duplicate Queries Detected (${stats.duplicates.length})`);
+        console.table(stats.duplicates.map(d => ({SQL: d.sql, Parameters: d.params, Duration_ms: d.duration})));
+        console.groupEnd();
+    }
+})();
+</script>"""
+
+BODY_CLOSE_RE = re.compile(rb"</body\s*>", re.IGNORECASE)
+
+
+class DevBarMiddleware:
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        tracker.reset()
+        request_start = perf_counter()
+
+        with ExitStack() as stack:
+            for alias in connections:
+                stack.enter_context(
+                    connections[alias].execute_wrapper(tracker.tracking_wrapper)
+                )
+            response = self.get_response(request)
+
+        total_time = (perf_counter() - request_start) * 1000
+        stats = tracker.get_stats()
+
+        db_time = stats["duration"]
+        python_time = max(0, total_time - db_time)
+
+        stats["python_time"] = python_time
+        stats["total_time"] = total_time
+
+        thresholds = get_thresholds()
+        level = self._determine_level(stats, total_time, thresholds)
+
+        if get_show_headers():
+            self._add_headers(response, stats)
+
+        if get_show_bar() and self._can_inject(response):
+            self._inject_devbar(response, stats, level)
+
+        return response
+
+    def _determine_level(self, stats, total_time, thresholds):
+        if (
+            total_time > thresholds["time_critical"]
+            or stats["count"] > thresholds["count_critical"]
+        ):
+            return "crit"
+        if (
+            stats["has_duplicates"]
+            or total_time > thresholds["time_warning"]
+            or stats["count"] > thresholds["count_warning"]
+        ):
+            return "warn"
+        return "ok"
+
+    def _add_headers(self, response, stats):
+        response["DevBar-Query-Count"] = str(stats["count"])
+        response["DevBar-DB-Time"] = f"{stats['duration']:.0f}ms"
+        response["DevBar-App-Time"] = f"{stats['python_time']:.0f}ms"
+        if stats["has_duplicates"]:
+            response["DevBar-Duplicates"] = str(len(stats["duplicate_queries"]))
+
+    def _can_inject(self, response):
+        if getattr(response, "streaming", False):
+            return False
+        content_type = response.get("Content-Type", "").lower()
+        if "html" not in content_type:
+            return False
+        if response.get("Content-Encoding"):
+            return False
+        return hasattr(response, "content")
+
+    def _inject_devbar(self, response, stats, level):
+        content = response.content
+        matches = list(BODY_CLOSE_RE.finditer(content))
+        if not matches:
+            return
+
+        dup_marker = (
+            ' <strong style="color:#f59e0b">(D)</strong>'
+            if stats["has_duplicates"]
+            else ""
+        )
+
+        css = STYLE_BLOCK % get_position()
+        html = BAR_TEMPLATE % (
+            level,
+            stats["duration"],
+            stats["python_time"],
+            stats["count"],
+            dup_marker,
+        )
+
+        script = ""
+        if get_enable_console() and stats.get("duplicate_queries"):
+            console_data = {"duplicates": stats["duplicate_queries"]}
+            script = SCRIPT_TEMPLATE % json.dumps(console_data)
+
+        payload = (css + html + script).encode(response.charset or "utf-8")
+
+        idx = matches[-1].start()
+        response.content = content[:idx] + payload + content[idx:]
+        response["Content-Length"] = str(len(response.content))

{django_devbar-0.1.0 → django_devbar-0.1.2}/src/django_devbar/tracker.py

@@ -3,24 +3,23 @@ from time import perf_counter
 
 _query_count: ContextVar[int] = ContextVar("query_count", default=0)
 _query_duration: ContextVar[float] = ContextVar("query_duration", default=0.0)
-_seen_queries: ContextVar[dict] = ContextVar(
-    "seen_queries"
-)  # sql -> set of params hashes
-_has_duplicates: ContextVar[bool] = ContextVar("has_duplicates", default=False)
+_seen_queries: ContextVar[dict] = ContextVar("seen_queries", default={})
+_duplicate_log: ContextVar[list] = ContextVar("duplicate_log", default=[])
 
 
 def reset():
     _query_count.set(0)
     _query_duration.set(0.0)
     _seen_queries.set({})
-    _has_duplicates.set(False)
+    _duplicate_log.set([])
 
 
 def get_stats():
     return {
         "count": _query_count.get(),
         "duration": _query_duration.get(),
-        "has_duplicates": _has_duplicates.get(),
+        "has_duplicates": bool(_duplicate_log.get()),
+        "duplicate_queries": _duplicate_log.get(),
     }
 
 
@@ -36,14 +35,22 @@ def _record(sql, params, duration):
     _query_duration.set(_query_duration.get() + duration)
 
     seen = _seen_queries.get()
+    params_hash = _hash_params(params)
+
     if sql in seen:
-        params_hash = _hash_params(params)
         if params_hash in seen[sql]:
-            _has_duplicates.set(True)
+            param_str = str(params)
+            if len(param_str) > 200:
+                param_str = param_str[:200] + "..."
+
+            duplicates = _duplicate_log.get()
+            duplicates.append(
+                {"sql": sql, "params": param_str, "duration": round(duration, 2)}
+            )
         else:
             seen[sql].add(params_hash)
     else:
-        seen[sql] = {_hash_params(params)}
+        seen[sql] = {params_hash}
 
 
 def tracking_wrapper(execute, sql, params, many, context):

django_devbar-0.1.0/src/django_devbar/middleware.py

@@ -1,89 +0,0 @@
-import re
-from contextlib import ExitStack
-from time import perf_counter
-
-from django.db import connections
-
-from . import tracker
-from .conf import get_position, get_show_bar, get_show_headers
-
-DEVBAR_HTML = (
-    '<div id="django-devbar" style="position:fixed;{position};'
-    "background:rgba(0,0,0,0.7);color:rgba(255,255,255,0.85);"
-    "padding:4px 8px;margin:8px;"
-    "font-family:-apple-system,BlinkMacSystemFont,'Segoe UI',Roboto,sans-serif;"
-    "font-size:10px;line-height:1.3;font-weight:500;letter-spacing:0.02em;"
-    "z-index:99999;border-radius:3px;"
-    'backdrop-filter:blur(8px);-webkit-backdrop-filter:blur(8px)">'
-    '<span style="opacity:0.5">queries</span> {count}{duplicates} <span style="opacity:0.5">·</span> '
-    '<span style="opacity:0.5">db</span> {duration:.0f}ms <span style="opacity:0.5">·</span> '
-    '<span style="opacity:0.5">total</span> {response_time:.0f}ms</div>'
-)
-
-DUPLICATE_MARKER = (
-    ' <span style="color:#f59e0b" title="Duplicate queries detected">(d)</span>'
-)
-
-BODY_CLOSE_RE = re.compile(rb"</body\s*>", re.IGNORECASE)
-
-
-class DevBarMiddleware:
-    def __init__(self, get_response):
-        self.get_response = get_response
-
-    def __call__(self, request):
-        request_start = perf_counter()
-        tracker.reset()
-
-        with ExitStack() as stack:
-            for alias in connections:
-                stack.enter_context(
-                    connections[alias].execute_wrapper(tracker.tracking_wrapper)
-                )
-            response = self.get_response(request)
-
-        response_time = (perf_counter() - request_start) * 1000
-        stats = tracker.get_stats()
-
-        if get_show_headers():
-            self._add_headers(response, stats, response_time)
-
-        if get_show_bar() and self._can_inject(response):
-            self._inject_devbar(response, stats, response_time)
-
-        return response
-
-    def _add_headers(self, response, stats, response_time):
-        response["X-DevBar-Query-Count"] = str(stats["count"])
-        response["X-DevBar-Query-Duration"] = f"{stats['duration']:.1f}"
-        response["X-DevBar-Response-Time"] = f"{response_time:.1f}"
-        if stats["has_duplicates"]:
-            response["X-DevBar-Has-Duplicates"] = "1"
-
-    def _can_inject(self, response):
-        if getattr(response, "streaming", False):
-            return False
-        content_type = response.get("Content-Type", "").lower()
-        if "text/html" not in content_type:
-            return False
-        if response.get("Content-Encoding"):
-            return False
-        return hasattr(response, "content")
-
-    def _inject_devbar(self, response, stats, response_time):
-        content = response.content
-        matches = list(BODY_CLOSE_RE.finditer(content))
-        if not matches:
-            return
-
-        devbar_html = DEVBAR_HTML.format(
-            position=get_position(),
-            count=stats["count"],
-            duplicates=DUPLICATE_MARKER if stats["has_duplicates"] else "",
-            duration=stats["duration"],
-            response_time=response_time,
-        ).encode(response.charset or "utf-8")
-
-        idx = matches[-1].start()
-        response.content = content[:idx] + devbar_html + content[idx:]
-        response["Content-Length"] = str(len(response.content))