pbp_com/apps/darkroom/inlines.py

from django_inlines.inlines import TemplateInline
from darkroom.models import Photo


class PhotoInline(TemplateInline):
    """
    An inline that takes a darkroom photo id, slug, or just a url returns a photo template with proper width and height.

    Examples::

        {{ photo 34 }}

    The inluded template supports width and height arguments::

        {{ photo 32 size=small }} 

    """

    help_text = (
        "Takes a darkroom photo id or slug and returns a template for the photo."
    )
    inline_args = [dict(name="size", help_text="Small/Medium/Large/Full")]

    def get_context(self):
        try:
            photo = Photo.objects.get(pk=self.value)
        except:
            photo = Photo.objects.get(slug=self.value)
        return {"photo": photo}


class GraphicInline(TemplateInline):
    """
    An inline that takes a darkroom graphic id, slug, or just a url returns a template with size and position classes set.

    Examples::

        {{ graphic 34 }}

    The inluded template supports width and height arguments::

        {{ graphic 32 size=small float=right }} 

    """

    help_text = (
        "Takes a darkroom graphic id or slug and returns a template for the graphic."
    )
    inline_args = [dict(name="size", help_text="Small/Medium/Large/Full")]

    def get_context(self):
        try:
            photo = Graphic.objects.get(pk=self.value)
        except:
            photo = Graphic.objects.get(slug=self.value)
        return {"graphic": graphic}