IdHub/idhub/admin/tables.py

103 lines
2.7 KiB
Python
Raw Normal View History

import django_tables2 as tables
2023-11-29 13:41:15 +00:00
from django.utils.html import format_html
from idhub.models import Rol, Event
from idhub_auth.models import User
2023-11-29 13:41:15 +00:00
class ButtonColumn(tables.Column):
attrs = {
"a": {
"type": "button",
"class": "text-primary",
}
}
2023-11-30 16:05:31 +00:00
# it makes no sense to order a column of buttons
orderable = False
2023-11-29 13:41:15 +00:00
# django_tables will only call the render function if it doesn't find
# any empty values in the data, so we stop it from matching the data
# to any value considered empty
empty_values = ()
def render(self):
return format_html('<i class="bi bi-eye"></i>')
class UserTable(tables.Table):
2023-11-29 13:41:15 +00:00
view_user = ButtonColumn(
linkify={
"viewname": "idhub:admin_people",
"args": [tables.A("pk")]
},
2023-11-30 16:05:31 +00:00
orderable=False
)
2023-11-29 13:41:15 +00:00
membership = tables.Column(empty_values=())
role = tables.Column(empty_values=())
2023-11-30 16:05:31 +00:00
def render_view_user(self):
return format_html('<i class="bi bi-eye"></i>')
2023-11-29 13:41:15 +00:00
def render_membership(self, record):
return record.get_memberships()
def order_membership(self, queryset, is_descending):
# TODO: Test that this doesn't return more rows than it should
queryset = queryset.order_by(
("-" if is_descending else "") + "memberships__type"
)
return (queryset, True)
def render_role(self, record):
return record.get_roles()
def order_role(self, queryset, is_descending):
queryset = queryset.order_by(
("-" if is_descending else "") + "roles"
)
return (queryset, True)
class Meta:
model = User
template_name = "idhub/custom_table.html"
2023-11-29 13:41:15 +00:00
fields = ("last_name", "first_name", "email", "membership", "role",
"view_user")
class RolesTable(tables.Table):
2023-11-30 16:05:31 +00:00
view_role = ButtonColumn(
linkify={
"viewname": "idhub:admin_rol_edit",
"args": [tables.A("pk")]
},
orderable=False
)
delete_role = ButtonColumn(
linkify={
"viewname": "idhub:admin_rol_del",
"args": [tables.A("pk")]
},
orderable=False
)
def render_view_role(self):
return format_html('<i class="bi bi-pencil-square"></i>')
def render_delete_role(self):
return format_html('<i class="bi bi-trash">')
class Meta:
model = Rol
template_name = "idhub/custom_table.html"
fields = ("name", "description")
class DashboardTable(tables.Table):
class Meta:
model = Event
template_name = "idhub/custom_table.html"
fields = ("type", "message", "created")