Source code for rpaper.apps.reservations.admin

from django.contrib import admin
from rpaper.core.utils import get_client_ip
from .models import (
    Thing,
    Record,
)


@admin.register(Thing)
class ThingAdmin(admin.ModelAdmin):
    readonly_fields = ('pk', 'owner')

    def save_model(self, request, obj, form, change):
        if getattr(obj, 'owner', None) is None:
            obj.owner = request.user
        if getattr(obj, 'ipaddress', None) is None:
            obj.ipaddress = get_client_ip(request)
        obj.save()


@admin.register(Record)
class RecordAdmin(admin.ModelAdmin):
    readonly_fields = ('pk', 'owner')

    def save_model(self, request, obj, form, change):
        if getattr(obj, 'owner', None) is None:
            obj.owner = request.user
        if getattr(obj, 'ipaddress', None) is None:
            obj.ipaddress = get_client_ip(request)
        obj.save()