path: root/chimere/admin.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2008-2012  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# See the file COPYING for details.

"""
Settings for administration pages
"""
import datetime

from django import forms
from django.conf import settings
from django.contrib import admin
from django.http import HttpResponse
from django.shortcuts import render_to_response
from django.utils.translation import ugettext_lazy as _

try:
    from chimere import tasks
except ImportError:
    pass

from chimere.forms import MarkerAdminForm, RouteAdminForm, AreaAdminForm,\
                          NewsAdminForm, CategoryAdminForm
from chimere.models import Category, Icon, SubCategory, Marker, \
     PropertyModel, News, Route, Area, ColorTheme, Color, RouteFile,\
     MultimediaType, MultimediaFile, PictureFile, Importer, Layer, AreaLayers
from chimere.utils import unicode_normalize, ShapefileManager, KMLManager
from chimere.widgets import TextareaWidget

def get_areas_for_user(user):
    """
    Getting subcats for a specific user
    """
    perms = user.get_all_permissions()
    areas = set()
    prefix = 'chimere.change_area_'
    for perm in perms:
        if perm.startswith(prefix):
            area = Area.objects.get(urn=perm[len(prefix):])
            areas.add(area)
    return areas

def validate(modeladmin, request, queryset):
    for item in queryset:
        item.status = 'A'
        item.save()
validate.short_description = _(u"Validate")

def export_to_kml(modeladmin, request, queryset):
    u"""
    Export data to KML
    """
    filename, result = KMLManager.export(queryset)
    response = HttpResponse(result,
                        mimetype='application/vnd.google-earth.kml+xml')
    response['Content-Disposition'] = 'attachment; filename=%s' % filename
    return response
export_to_kml.short_description = _(u"Export to KML")

def export_to_shapefile(modeladmin, request, queryset):
    u"""
    Export data to Shapefile
    """
    filename, zip_stream = ShapefileManager.export(queryset)
    # Stick it all in a django HttpResponse
    response = HttpResponse()
    response['Content-Disposition'] = 'attachment; filename=%s.zip' % filename
    response['Content-length'] = str(len(zip_stream))
    response['Content-Type'] = 'application/zip'
    response.write(zip_stream)
    return response
export_to_shapefile.short_description = _(u"Export to Shapefile")

class PictureInline(admin.TabularInline):
    model = PictureFile
    extra = 1

class MultimediaInline(admin.TabularInline):
    model = MultimediaFile
    extra = 1

class MarkerAdmin(admin.ModelAdmin):
    """
    Specialized the Point field.
    """
    search_fields = ("name",)
    list_display = ('name', 'status')
    list_filter = ('status', 'categories')
    actions = [validate, export_to_kml, export_to_shapefile]
    exclude = ['submiter_session_key', 'import_key', 'import_version',
               'available_date']
    readonly_fields = ['submiter_email', 'submiter_comment', 'import_source',
                       'ref_item']
    form = MarkerAdminForm
    inlines = [MultimediaInline, PictureInline]

    def queryset(self, request):
        qs = self.model._default_manager.get_query_set()
        if not request.user.is_superuser:
            areas = get_areas_for_user(request.user)
            if areas:
                in_areas = " or ".join([area.getIncludeSql() for area in areas])
                qs = qs.extra(where=[in_areas])
        ordering = self.ordering or ()
        if ordering:
            qs = qs.order_by(*ordering)
        return qs

class RouteAdmin(admin.ModelAdmin):
    """
    Specialized the Route field.
    """
    search_fields = ("name",)
    list_display = ('name', 'status')
    list_filter = ('status', 'categories')
    exclude = ['height', 'width']
    form = RouteAdminForm
    readonly_fields = ('associated_file',)
    actions = [validate]

    def queryset(self, request):
        qs = self.model._default_manager.get_query_set()
        if not request.user.is_superuser:
            areas = get_areas_for_user(request.user)
            if areas:
                in_areas = " or ".join([area.getIncludeSql(
                           geometry='"chimere_route".route') for area in areas])
                qs = qs.extra(where=[in_areas])
        ordering = self.ordering or ()
        if ordering:
            qs = qs.order_by(*ordering)
        return qs

class LayerInline(admin.TabularInline):
    model = AreaLayers
    extra = 1

class AreaAdmin(admin.ModelAdmin):
    """
    Specialized the area field.
    """
    form = AreaAdminForm
    exclude = ['upper_left_corner', 'lower_right_corner']
    inlines = [LayerInline]

class SubCategoryAdmin(admin.ModelAdmin):
    """
    Specialized the subcategory admin
    """
    list_display = ('name', 'category', 'available')
    list_filter = ('category',)

def importing(modeladmin, request, queryset):
    for importer in queryset:
        importer.state = unicode(tasks.IMPORT_MESSAGES['import_pending'][0])
        importer.save()
        tasks.importing(importer.pk)
importing.short_description = _(u"Import")

def cancel_import(modeladmin, request, queryset):
    for importer in queryset:
        importer.state = tasks.IMPORT_MESSAGES['import_cancel'][0]
        importer.save()
cancel_import.short_description = _(u"Cancel import")

def cancel_export(modeladmin, request, queryset):
    for importer in queryset:
        importer.state = tasks.IMPORT_MESSAGES['export_cancel'][0]
        importer.save()
cancel_export.short_description = _(u"Cancel export")

def export_to_osm(modeladmin, request, queryset):
    importers = modeladmin.model.objects.filter(importer_type='OSM')
    for importer in queryset:
        importer.state = unicode(tasks.IMPORT_MESSAGES['export_pending'][0])
        importer.save()
        tasks.exporting(importer.pk)
export_to_osm.short_description = _(u"Export to osm")

class ImporterAdmin(admin.ModelAdmin):
    list_display = ('importer_type', 'source', 'state', 'filtr')
    list_filter = ('importer_type', 'source')
    readonly_fields = ('state',)
    actions = [importing, cancel_import, cancel_export]
admin.site.register(Importer, ImporterAdmin)


class NewsAdmin(admin.ModelAdmin):
    """
    Use the TinyMCE widget for the news content
    """
    form = NewsAdminForm

class CategoryAdmin(admin.ModelAdmin):
    """
    Use the TinyMCE widget for categories
    """
    form = CategoryAdminForm

class ColorInline(admin.TabularInline):
    model = Color

class ColorThemeAdmin(admin.ModelAdmin):
    inlines = [ColorInline,]

class IconAdmin(admin.ModelAdmin):
    exclude = ['height', 'width']

class RouteFileAdmin(admin.ModelAdmin):
    list_display = ['name', 'file_type']

class MultimediaTypeAdmin(admin.ModelAdmin):
    search_fields = ("name",)
    list_display = ('name', 'media_type', 'mime_type', 'iframe', 'available')
    list_filter = ('media_type', 'available')

class MultimediaFileAdmin(admin.ModelAdmin):
    search_fields = ("name",)
    list_display = ('name', 'multimedia_type', 'order')
    list_filter = ('multimedia_type',)

class PictureFileAdmin(admin.ModelAdmin):
    search_fields = ("name",)
    list_display = ('name', 'picture', 'miniature', 'order')
    list_filter = ('miniature',)

# register of differents database fields
admin.site.register(News, NewsAdmin)
admin.site.register(Category, CategoryAdmin)
admin.site.register(Icon, IconAdmin)
admin.site.register(SubCategory, SubCategoryAdmin)
admin.site.register(Marker, MarkerAdmin)
admin.site.register(RouteFile, RouteFileAdmin)
admin.site.register(Route, RouteAdmin)
admin.site.register(PropertyModel)
admin.site.register(Area, AreaAdmin)
admin.site.register(ColorTheme, ColorThemeAdmin)
admin.site.register(MultimediaType, MultimediaTypeAdmin)
admin.site.register(MultimediaFile, MultimediaFileAdmin)
admin.site.register(PictureFile, PictureFileAdmin)
admin.site.register(Layer)