penracourses/atlas/tables.py

import django_tables2 as tables
from django_tables2.utils import A

from .models import (
    Case,
    Condition,
    PathologicalProcess,
    Presentation,
    Series,
    Structure,
    Finding,
    Subspecialty,
)

from django.utils.html import format_html

from easy_thumbnails.files import get_thumbnailer

from easy_thumbnails.exceptions import InvalidImageFormatError


class AtlasImageColumn(tables.Column):
    def render(self, value):
        blocks = []
        for obj in value.all():
            blocks.append(obj.get_block())
        return format_html(
            """<span class='multi-image-block'>
            {}
            </span>""".format("".join(blocks))
        )

        obj = value.first()

        if obj is None:
            return format_html("<span>No image<span>")

        return obj.get_thumbnail()[0]

        # image_object = obj.image
        # try:
        #    thumbnailer = get_thumbnailer(image_object)
        #    return format_html('<img src="/media/{}" />', thumbnailer["exam-list"])
        # except InvalidImageFormatError:
        #    return format_html('<span title="{}">Invalid image url<span>', image_object)


class SeriesImageColumn(tables.Column):
    def render(self, value):
        obj = value.first()

        if obj is None:
            return "None"

        image_object = obj.image
        try:
            thumbnailer = get_thumbnailer(image_object)
            return format_html(
                '<img src="/media/{}" /><br/>[{}]',
                thumbnailer["exam-list"],
                value.count(),
            )
        except InvalidImageFormatError:
            return format_html(
                '<span title="{}">Invalid image url<span><br/>[{}]',
                image_object,
                value.count(),
            )


class CaseTable(tables.Table):
    edit = tables.LinkColumn(
        "atlas:case_update", text="Edit", args=[A("pk")], orderable=False
    )
    view = tables.LinkColumn(
        "atlas:case_detail", text="View", args=[A("pk")], orderable=False
    )
    clone = tables.LinkColumn(
        "atlas:case_clone", text="Clone", args=[A("pk")], orderable=False
    )
    delete = tables.LinkColumn(
        "atlas:case_delete", text="Delete", args=[A("pk")], orderable=False
    )
    #series = AtlasImageColumn("Images", orderable=False)

    #differential = tables.ManyToManyColumn(verbose_name="Differential")

    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    class Meta:
        model = Case
        template_name = "django_tables2/bootstrap4.html"
        fields = (
            "title",
            "description",
            #"history",
            #"condition",
            "created_date",
            "author",
        )
        sequence = ("view", )#, "series")

    def __init__(self, data=None, *args, **kwargs):
        super().__init__(
            data.prefetch_related(
                #"condition", "condition__synonym"
            ),
            *args,
            **kwargs,
        )

class PopupLinkColumn(tables.Column):
    def render(self, value):
        return format_html("<span>test: {}<span>", value)
        obj = value.first()

        if obj is None:
            return format_html("<span>No image<span>")

        return obj.get_thumbnail()[0]


def create_link(test):
    print(test)
    return "Hello"


class SeriesTable(tables.Table):
    edit = tables.LinkColumn(
        "atlas:series_update", text="Edit", args=[A("pk")], orderable=False
    )
    view = tables.LinkColumn(
        "atlas:series_detail", text="View", args=[A("pk")], orderable=False
    )
    popup = tables.Column(empty_values=(), orderable=False)

    images = SeriesImageColumn("Images", orderable=False)

    delete = tables.LinkColumn(
        "atlas:series_delete", text="Delete", args=[A("pk")], orderable=False
    )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    case = tables.ManyToManyColumn(verbose_name="Cases")
    findings = tables.ManyToManyColumn(verbose_name="Findings")
    structures = tables.ManyToManyColumn(verbose_name="Structures")

    class Meta:
        model = Series
        template_name = "django_tables2/bootstrap4.html"
        fields = (
            "modality",
            "description",
            "examination",
            "plane",
            "contrast",
            "author",
        )
        sequence = ("view", "popup", "images", "case")

    def render_popup(self, value, record):
        print(self)
        return format_html(
            """<a href="#" onclick="return window.create_popup_window('/atlas/series/{}', 'Series')" >Popup</a>""",
            record.pk,
        )


class ConditionTable(tables.Table):
    name = tables.Column(
        linkify=("atlas:condition_detail", {"pk": tables.A("pk")}),
        verbose_name="Condition",
    )

    edit = tables.LinkColumn(
        "atlas:condition_update", text="Edit", args=[A("pk")], orderable=False
    )
    delete = tables.LinkColumn(
        "atlas:condition_delete", text="Delete", args=[A("pk")], orderable=False
    )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    # synonym = tables.ManyToManyColumn(verbose_name="Synonyms")
    synonym = tables.Column(empty_values=())
    parent = tables.ManyToManyColumn(verbose_name="Parents")

    class Meta:
        model = Condition
        template_name = "django_tables2/bootstrap4.html"
        fields = ("primary", "subspecialty")
        sequence = ("name",)

    def render_synonym(self, value, record):
        return format_html(record.get_synonym_link())
        return f"{record.get_synonym_link()}"


class FindingTable(tables.Table):
    name = tables.Column(
        linkify=("atlas:finding_detail", {"pk": tables.A("pk")}), verbose_name="Finding"
    )
    edit = tables.LinkColumn(
        "atlas:finding_update", text="Edit", args=[A("pk")], orderable=False
    )
    delete = tables.LinkColumn(
        "atlas:finding_delete", text="Delete", args=[A("pk")], orderable=False
    )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    # synonym = tables.ManyToManyColumn(verbose_name="Synonyms")
    synonym = tables.Column(empty_values=())
    parent = tables.ManyToManyColumn(verbose_name="Parents")

    class Meta:
        model = Finding
        template_name = "django_tables2/bootstrap4.html"
        fields = ("name", "primary")
        sequence = ("name",)

    def render_synonym(self, value, record):
        return format_html(record.get_synonym_link())


class StructureTable(tables.Table):
    name = tables.Column(
        linkify=("atlas:structure_detail", {"pk": tables.A("pk")}),
        verbose_name="Structure",
    )
    edit = tables.LinkColumn(
        "atlas:structure_update", text="Edit", args=[A("pk")], orderable=False
    )
    delete = tables.LinkColumn(
        "atlas:structure_delete", text="Delete", args=[A("pk")], orderable=False
    )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    # synonym = tables.ManyToManyColumn(verbose_name="Synonyms")
    synonym = tables.Column(empty_values=())
    parent = tables.ManyToManyColumn(verbose_name="Parents")

    class Meta:
        model = Structure
        template_name = "django_tables2/bootstrap4.html"
        fields = ("name", "primary")
        sequence = ("name",)

    def render_synonym(self, value, record):
        return format_html(record.get_synonym_link())


class PresentationTable(tables.Table):
    name = tables.Column(
        linkify=("atlas:presentation_detail", {"pk": tables.A("pk")}),
        verbose_name="Presentation",
    )

    # edit = tables.LinkColumn(
    #    "atlas:presentation_update", text="Edit", args=[A("pk")], orderable=False
    # )
    # delete = tables.LinkColumn(
    #    "atlas:presentation_delete", text="Delete", args=[A("pk")], orderable=False
    # )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    subspecialty = tables.ManyToManyColumn(verbose_name="Subspecialty")

    class Meta:
        model = Presentation
        template_name = "django_tables2/bootstrap4.html"
        fields = ("name",)
        sequence = ("name",)


class PathologicalProcessTable(tables.Table):
    name = tables.Column(
        linkify=("atlas:pathological_process_detail", {"pk": tables.A("pk")}),
        verbose_name="PathologicalProcess",
    )

    # edit = tables.LinkColumn(
    #    "atlas:pathological_process_update", text="Edit", args=[A("pk")], orderable=False
    # )
    # delete = tables.LinkColumn(
    #    "atlas:pathological_process_delete", text="Delete", args=[A("pk")], orderable=False
    # )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    class Meta:
        model = PathologicalProcess
        template_name = "django_tables2/bootstrap4.html"
        fields = ("name",)
        sequence = ("name",)


class SubspecialtyTable(tables.Table):
    name = tables.Column(
        linkify=("atlas:subspecialty_detail", {"pk": tables.A("pk")}),
        verbose_name="Subspecialty",
    )

    # edit = tables.LinkColumn(
    #    "atlas:subspecialty_update", text="Edit", args=[A("pk")], orderable=False
    # )
    # delete = tables.LinkColumn(
    #    "atlas:subspecialty_delete", text="Delete", args=[A("pk")], orderable=False
    # )
    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    class Meta:
        model = Subspecialty
        template_name = "django_tables2/bootstrap4.html"
        fields = ()
        sequence = ("name",)

    def render_synonym(self, value, record):
        return format_html(record.get_synonym_link())
        return f"{record.get_synonym_link()}"

class CaseCollectionTable(tables.Table):
    edit = tables.LinkColumn(
        "atlas:collection_update", text="Edit", args=[A("pk")], orderable=False
    )
    view = tables.LinkColumn(
        "atlas:collection_detail", text="View", args=[A("pk")], orderable=False
    )
    #clone = tables.LinkColumn(
    #    "atlas:case_clone", text="Clone", args=[A("pk")], orderable=False
    #)
    delete = tables.LinkColumn(
        "atlas:collection_delete", text="Delete", args=[A("pk")], orderable=False
    )
    #series = AtlasImageColumn("Images", orderable=False)

    #differential = tables.ManyToManyColumn(verbose_name="Differential")

    selection = tables.CheckBoxColumn(accessor="pk", orderable=False)

    class Meta:
        model = Case
        template_name = "django_tables2/bootstrap4.html"
        fields = (
            "name",
            "collection_type",
            "publish_results",
            "active",
            "author",
        )
        sequence = ("view", )#, "series")

    def __init__(self, data=None, *args, **kwargs):
        super().__init__(
            data.prefetch_related(
                #"condition", "condition__synonym"
            ),
            *args,
            **kwargs,
        )