cubicweb/pyramid/debugtoolbar_panels.py
changeset 12760 d5ae5abd0876
child 12761 10b8352b0208
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/cubicweb/pyramid/debugtoolbar_panels.py	Wed Jul 31 04:16:20 2019 +0200
@@ -0,0 +1,56 @@
+# copyright 2019 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+#
+# This file is part of CubicWeb.
+#
+# CubicWeb is free software: you can redistribute it and/or modify it under the
+# terms of the GNU Lesser General Public License as published by the Free
+# Software Foundation, either version 2.1 of the License, or (at your option)
+# any later version.
+#
+# CubicWeb is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Lesser General Public License along
+# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
+
+from pyramid_debugtoolbar.panels import DebugPanel
+from cubicweb.debug import subscribe_to_debug_channel, unsubscribe_to_debug_channel
+from cubicweb.misc.source_highlight import highlight_html, generate_css
+
+
+class RQLDebugPanel(DebugPanel):
+    """
+    CubicWeb RQL debug panel
+    """
+    name = 'RQL'
+    title = 'RQL queries'
+    nav_title = 'RQL'
+    nav_subtitle_style = 'progress-bar-info'
+
+    has_content = True
+    template = 'cubicweb.pyramid:debug_toolbar_templates/rql.dbtmako'
+
+    def __init__(self, request):
+        self.data = {
+            'rql_queries': [],
+            'highlight': highlight_html,
+            'generate_css': generate_css,
+        }
+        subscribe_to_debug_channel("rql", self.collect_rql_queries)
+
+    @property
+    def nav_subtitle(self):
+        return '%d' % len(self.data['rql_queries'])
+
+    def collect_rql_queries(self, rql_query):
+        self.data["rql_queries"].append(rql_query)
+
+    def process_response(self, response):
+        unsubscribe_to_debug_channel("rql", self.collect_rql_queries)
+
+
+def includeme(config):
+    config.add_debugtoolbar_panel(RQLDebugPanel)