from django.contrib.sitemaps import Sitemap
from django.urls import reverse
from .models import AdoptionNotice, RescueOrganization


class StaticViewSitemap(Sitemap):
    priority = 0.8
    changefreq = "weekly"

    def items(self):
        return ["index", "search", "map", "about", "rescue-organizations"]

    def location(self, item):
        return reverse(item)


class AdoptionNoticeSitemap(Sitemap):
    priority = 0.5
    changefreq = "daily"

    def items(self):
        return AdoptionNotice.get_active_ANs()

    def lastmod(self, obj):
        return obj.updated_at


class AnimalSitemap(Sitemap):
    priority = 0.2
    changefreq = "daily"

    def items(self):
        return AdoptionNotice.objects.all()

    def lastmod(self, obj):
        return obj.updated_at


class RescueOrganizationSitemap(Sitemap):
    priority = 0.3
    changefreq = "weekly"

    def items(self):
        return RescueOrganization.objects.all()

    def lastmod(self, obj):
        return obj.updated_at