# Copyright The IETF Trust 2007, All Rights Reserved

try:
    import tidy
    tidyavail = True
except ImportError:
    tidyavail = False
from django.db import connection
from django.shortcuts import render_to_response
from django.template import RequestContext
from ietf.utils import log
import re
import smtplib
import sys
import traceback

options = dict(
               output_xhtml=True,
#               add_xml_decl=True,
#               doctype='transitional',
               indent=True,
               tidy_mark=False,
#               hide_comments=True,
               wrap=100)


class PrettifyMiddleware(object):
    """Prettify middleware
    From http://www.djangosnippets.org/snippets/172/
    Uses python-utidylib, http://utidylib.berlios.de/,
    which uses HTML Tidy, http://tidy.sourceforge.net/
    """

    def process_response(self, request, response):
        if tidyavail and response.headers['Content-Type'].split(';', 1)[0] in ['text/html']:
            content = response.content
            content = str(tidy.parseString(content, **options))
            response.content = content
        return response

class SQLLogMiddleware(object):
    def process_response(self, request, response):
	for q in connection.queries:
	    if re.match('(update|insert)', q['sql'], re.IGNORECASE):
		log(q['sql'])
        return response

class SMTPExceptionMiddleware(object):
    def process_exception(self, request, exception):
	if isinstance(exception, smtplib.SMTPException):
	    type = sys.exc_info()[0]
	    value = sys.exc_info()[1]
	    # See if it's a non-smtplib exception that we faked
	    if type == smtplib.SMTPException and len(value.args) == 1 and isinstance(value.args[0], dict) and value.args[0].has_key('really'):
		orig = value.args[0]
		type = orig['really']
		tb = traceback.format_tb(orig['tb'])
		value = orig['value']
	    else:
		tb = traceback.format_tb(sys.exc_info()[2])
	    return render_to_response('email_failed.html', {'exception': type, 'args': value, 'traceback': "".join(tb)},
		context_instance=RequestContext(request))
	return None