from django.db import connection
from django.template import Template, Context

class SQLLogMiddleware:
    """
    http://www.djangosnippets.org/snippets/161/
    
    To enable, add the following line to your settings.py:
    MIDDLEWARE_CLASSES += ('DjHDGutils.profiling.SQLLogMiddleware',)
    """

    def process_response ( self, request, response ):

        
        time = 0.0
        for q in connection.queries:
		time += float(q['time'])

        if response['Content-type'].find('text/html')==-1:
            return response

        t = Template(u'''
            <p><em>Total query count:</em> {{ count }}<br/>
            <em>Total execution time:</em> {{ time }}</p>
            <ul class="sqllog">
                {% for sql in sqllog %}
                    <li>{{ sql.time }}: {{ sql.sql }}</li>
                {% endfor %}
            </ul>
        ''')
        
        content = response.content.decode('utf-8')
        content += t.render(Context({'sqllog':connection.queries,'count':len(connection.queries),'time':time}))
        response.content = content.encode('utf-8')
        
        return response