from django.contrib import admin
from paperless_mail.models import MailAccount, MailRule

from django.utils.translation import gettext_lazy as _


class MailAccountAdmin(admin.ModelAdmin):

    list_display = ("name", "imap_server", "username")


class MailRuleAdmin(admin.ModelAdmin):

    radio_fields = {
        "attachment_type": admin.VERTICAL,
        "action": admin.VERTICAL,
        "assign_title_from": admin.VERTICAL,
        "assign_correspondent_from": admin.VERTICAL
    }

    fieldsets = (
        (None, {
            'fields': ('name', 'order', 'account', 'folder')
        }),
        (_("Filter"), {
            'description':
                _("Paperless will only process mails that match ALL of the "
                  "filters given below."),
            'fields':
                ('filter_from',
                 'filter_subject',
                 'filter_body',
                 'filter_attachment_filename',
                 'maximum_age',
                 'attachment_type')
        }),
        (_("Actions"), {
            'description':
                _("The action applied to the mail. This action is only "
                  "performed when documents were consumed from the mail. "
                  "Mails without attachments will remain entirely untouched."),
            'fields': (
                'action',
                'action_parameter')
        }),
        (_("Metadata"), {
            'description':
                _("Assign metadata to documents consumed from this rule "
                  "automatically. If you do not assign tags, types or "
                  "correspondents here, paperless will still process all "
                  "matching rules that you have defined."),
            "fields": (
                'assign_title_from',
                'assign_tag',
                'assign_document_type',
                'assign_correspondent_from',
                'assign_correspondent')
        })
    )

    list_filter = ("account",)

    list_display = ("order", "name", "account", "folder", "action")

    list_editable = ("order", )

    list_display_links = ("name", )

    sortable_by = []

    ordering = ["order"]


admin.site.register(MailAccount, MailAccountAdmin)
admin.site.register(MailRule, MailRuleAdmin)