codeforamerica/intake

from collections import OrderedDict
from intake import models
from rest_framework import serializers
from .status_update_serializer import MinimalStatusUpdateSerializer, \
    StatusUpdateCSVDownloadSerializer
from intake.services.display_form_service import \
    get_display_form_for_csv_download


class LatestStatusBase(serializers.ModelSerializer):

    def to_representation(self, *args, **kwargs):
        data = super().to_representation(*args, **kwargs)
        sorted_status_updates = sorted(
            data.get('status_updates', []), key=lambda d: d['created'],
            reverse=True)
        latest_status = \
            sorted_status_updates[0] if sorted_status_updates else None
        data.update(latest_status=latest_status)
        return data


class ApplicationFollowupListSerializer(LatestStatusBase):
    organization = serializers.SlugRelatedField(
        read_only=True, slug_field='slug')
    status_updates = MinimalStatusUpdateSerializer(many=True)

    class Meta:
        model = models.Application
        fields = [
            'organization',
            'status_updates'
        ]


class ApplicationAutocompleteSerializer(serializers.ModelSerializer):
    name = serializers.SerializerMethodField(read_only=True)
    url = serializers.SerializerMethodField(read_only=True)

    class Meta:
        model = models.Application
        fields = ['name', 'url']

    def get_name(self, instance):
        return instance.form_submission.get_full_name()

    def get_url(self, instance):
        return instance.form_submission.get_absolute_url()


class ApplicationCSVDownloadSerializer(LatestStatusBase):
    status_updates = StatusUpdateCSVDownloadSerializer(many=True)

    def to_representation(self, app, *args, **kwargs):
        app_fields = super().to_representation(app, *args, **kwargs)
        display_form, letter = get_display_form_for_csv_download(app)
        data = OrderedDict(id=app.form_submission_id)
        data['Link'] = app.form_submission.get_external_url()
        data['Application Date'] = \
            app.form_submission.get_local_date_received('%m/%d/%Y')
        for field in display_form.get_usable_fields():
            data[field.get_display_label()] = field.get_display_value()
        data['Was transferred out'] = app_fields['was_transferred_out']
        data['Has been opened'] = app_fields['has_been_opened']

        if app_fields['latest_status']:
            data['Latest status'] = app_fields['latest_status']['status_type']
            data['Latest status date'] = \
                app_fields['latest_status']['created'].strftime('%m/%d/%Y')
            data['Latest status author'] = \
                app_fields['latest_status']['author_email']
        else:
            data['Latest status'] = 'New'
            data['Latest status date'] = ''
            data['Latest status author'] = ''

        data['Status history link'] = \
            app.form_submission.get_external_history_url()
        if letter:
            for field in letter.get_usable_fields():
                data[field.get_display_label()] = field.get_display_value()
        return data

    class Meta:
        model = models.Application
        fields = [
            'was_transferred_out',
            'has_been_opened',
            'status_updates'
        ]