lemoncurry/entries/kinds.py

from django.urls import reverse


class Entry:
    def __init__(self, id, plural, icon, on_home=True, slug=False):
        self.id = id
        self.plural = plural
        self.icon = icon
        self.on_home = on_home
        self.slug = slug

    @property
    def index(self):
        return self.index_page()

    def index_page(self, page=0):
        kwargs = {"kind": self}
        if page > 1:
            kwargs["page"] = page
        return reverse("entries:index", kwargs=kwargs)

    @property
    def entry(self):
        return self.plural + "_entry"

    @property
    def atom(self):
        return reverse("entries:atom_by_kind", kwargs={"kind": self})

    @property
    def rss(self):
        return reverse("entries:rss_by_kind", kwargs={"kind": self})


Note = Entry(
    id="note",
    icon="fas fa-paper-plane",
    plural="notes",
)


Article = Entry(
    id="article",
    icon="fas fa-file-alt",
    plural="articles",
    slug=True,
)

Photo = Entry(
    id="photo",
    icon="fas fa-camera",
    plural="photos",
)

Reply = Entry(
    id="reply",
    icon="fas fa-comment",
    plural="replies",
    on_home=False,
)

Like = Entry(
    id="like",
    icon="fas fa-heart",
    plural="likes",
    on_home=False,
)

Repost = Entry(
    id="repost",
    icon="fas fa-retweet",
    plural="reposts",
)

all = (Note, Article, Photo)
on_home = {k.id for k in all if k.on_home}
from_id = {k.id: k for k in all}
from_plural = {k.plural: k for k in all}


class EntryKindConverter:
    regex = "|".join(k.plural for k in all)

    def to_python(self, plural):
        return from_plural[plural]

    def to_url(self, k):
        return k.plural