celutz/panorama/admin.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import os

from django.contrib import admin

from .models import Panorama, PanoramaPhoto, ReferencePoint, Reference
from .utils import path_exists


class ReferenceInline(admin.TabularInline):
    model = Reference
    fk_name = "panorama"
    extra = 1


@admin.register(PanoramaPhoto)
class PanoramaPhotoAdmin(admin.ModelAdmin):
    model = PanoramaPhoto


class PanoramaPhotoInline(admin.StackedInline):
    model = PanoramaPhoto
    fk_name = "panorama"
    extra = 0


@admin.register(Panorama)
class PanoramaAdmin(admin.ModelAdmin):
    model = Panorama
    inlines = (PanoramaPhotoInline, ReferenceInline)
    list_display = ('name', 'has_tiles', 'latitude', 'longitude', 'altitude', 'loop')
    fields = ('name', ('image', 'image_width', 'image_height'),
              'loop', ('latitude', 'longitude'), 'altitude')
    readonly_fields = ('image_width', 'image_height')
    actions = ('generate_panorama', 'regenerate_tiles')

    def generate_panorama(self, request, queryset):
        for pano in queryset:
            pano.generate_panorama()
        self.message_user(request, "Launched panorama generation, it will take a few minutes")

    def regenerate_tiles(self, request, queryset):
        for pano in queryset:
            pano.delete_tiles()
            pano.generate_tiles()
        self.message_user(request, "Launched tiles regeneration, it may take some time to complete")
    regenerate_tiles.short_description = "Regenerate tiles for the selected panoramas"


@admin.register(ReferencePoint)
class ReferencePointAdmin(admin.ModelAdmin):
    model = ReferencePoint
    list_display = ('name', 'latitude', 'longitude', 'altitude')
    fields = ('name', ('latitude', 'longitude'), 'altitude')