from ajax_select import make_ajax_form
from django.contrib import admin

from commonnet.admin_site import admin_site
from commcrawler import models


class CrawlAdmin(admin.ModelAdmin):
    model = models.Crawl
    list_display = ("name", "status", "target_nb", "created", "started",
                    "ended")
    list_filter = ("status",)
    readonly_fields = ("status", "created", "started", "ended")
    exclude = ("targets", )


admin_site.register(models.Crawl, CrawlAdmin)


class CrawlResultAdmin(admin.ModelAdmin):
    model = models.CrawlResult
    list_display = ("target", "crawl", "is_online")
    list_filter = ("crawl",)
    form = make_ajax_form(model, {'target': 'target'})


admin_site.register(models.CrawlResult, CrawlResultAdmin)


class CrawlLinkAdmin(admin.ModelAdmin):
    model = models.CrawlLink
    list_display = ("result", "link")


admin_site.register(models.CrawlLink, CrawlLinkAdmin)


class CrawlRelationAdmin(admin.ModelAdmin):
    model = models.CrawlRelation
    list_display = ("crawl", "source", "destination", "number")
    list_filter = ("crawl",)
    search_fields = ["source__name", "destination__name"]
    form = make_ajax_form(model, {'source': 'target', 'destination': 'target'})


admin_site.register(models.CrawlRelation, CrawlRelationAdmin)


class ExcludedDomainAdmin(admin.ModelAdmin):
    list_display = ('url',)
    search_fields = ('url',)


admin_site.register(models.ExludedDomains, ExcludedDomainAdmin)