from django.contrib import admin
from .models import Survey, UserSurvey, SurveyQuestion, SurveyQuestionAnswerValue, UserSurveyQuestionAnswer

from django.utils.safestring import mark_safe
from django.core.urlresolvers import reverse

from modeltranslation.admin import TranslationAdmin, TranslationTabularInline, TabbedTranslationAdmin

class TabbedTranslationMixin(object):

    class Media:
        """ These enable the tabbed translation interface. Also in
            TabbedTranslationAdmin """
        js = (
            'modeltranslation/js/force_jquery.js',
            'http://ajax.googleapis.com/ajax/libs/jqueryui/1.8.24/jquery-ui.min.js',
            'modeltranslation/js/tabbed_translation_fields.js',
        )
        css = {
            'screen': ('modeltranslation/css/tabbed_translation_fields.css',),
        }


class SurveyQuestionAnswerValueInlineAdmin(admin.TabularInline):
    model = SurveyQuestionAnswerValue
    extra = 1

class TranslatedAnswerValueInlineAdmin(SurveyQuestionAnswerValueInlineAdmin,
                                       TranslationTabularInline):
    pass

class EditLinkToInlineObject(object):
    """ Include a link field to edit the object, pops out to a new tab /
    window. """

    def edit_answers(self, instance):
        url = reverse('admin:%s_%s_change' % (
            instance._meta.app_label,  instance._meta.module_name),  args=[instance.pk] )
        if instance.pk:
            return mark_safe(u'<a href="{u}" target="blank">edit</a>'.format(u=url))
        else:
            return ''

class SurveyQuestionInlineAdmin(EditLinkToInlineObject, admin.TabularInline):
    model = SurveyQuestion
    readonly_fields = ('edit_answers', )
    extra = 1

class TranslatedSurveyQuestionInlineAdmin(SurveyQuestionInlineAdmin, TranslationTabularInline):
    pass

class SurveyQuestionAdmin(admin.ModelAdmin):
    model = SurveyQuestion
    inlines = [TranslatedAnswerValueInlineAdmin]

class TranslatedSurveyQuestionAdmin(SurveyQuestionAdmin, TabbedTranslationAdmin):
    pass

class SurveyResponseInline(admin.TabularInline):
    model = UserSurvey
    readonly_fields = ('user_anonymized', 'completed', )
    exclude = ('user', )
    extra = 0

class SurveyAdmin(admin.ModelAdmin):
    """ The main survey object with inlines for answers.
    """
    inlines = [TranslatedSurveyQuestionInlineAdmin, SurveyResponseInline]

    list_display = ('title', 'user_responses_submitted', )

    def user_responses_submitted(self, inst):
        return inst.responses.count()

    user_responses_submitted.admin_order_field = 'response_count'

    def export_survey_result_csv(self, request, queryset):
        from django.core import serializers
        from django.http import HttpResponse

        from cStringIO import StringIO
        import csv

        # only serialize one at a time

        if queryset.count() > 1 or queryset.count() == 0:
            self.message_user(request, "Can only export one survey at a time.")
            return False

        handle = StringIO()
        csvwriter = csv.writer(handle)

        survey = queryset[0]
        csv_rows = survey.serialize_survey()

        csvwriter.writerows(csv_rows)
        contents = handle.getvalue()
        handle.close()

        self.message_user(request, "Survey exported.")
        response = HttpResponse(content_type="text/csv")
        response['Content-Disposition'] = 'attachment; filename="%s"' % "survey-results.csv"
        response.content = contents
        return response

    export_survey_result_csv.short_description = "Download a CSV of one survey's results (NB: only choose one)."

    actions = [export_survey_result_csv]

class TranslatedSurveyAdmin(SurveyAdmin, TabbedTranslationAdmin):
    pass

class UserSurveyQuestionAnswer(admin.TabularInline):
    model = UserSurveyQuestionAnswer

class UserSurveyAdmin(admin.ModelAdmin):
    model = UserSurvey

    inlines = [UserSurveyQuestionAnswer]
    list_display = ('user_anonymized', 'completed', 'survey')
    exclude = ('user', )
    ordering = ('-completed', )

# admin.site.register(Survey, SurveyAdmin)
admin.site.register(Survey, TranslatedSurveyAdmin)
admin.site.register(SurveyQuestion, TranslatedSurveyQuestionAdmin)
admin.site.register(UserSurvey, UserSurveyAdmin)