#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2010  É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.

"""
Admin description
"""

from django.contrib import admin

from ishtar import settings
import models

class DepartementAdmin(admin.ModelAdmin):
    list_display = ('number', 'label',)
    model = models.Departement

admin.site.register(models.Departement, DepartementAdmin)

class PersonAdmin(admin.ModelAdmin):
    list_display = ('name', 'surname', 'email', 'person_type')
    list_filter = ("person_type", "in_charge_storage", "is_author",)
    search_fields = ('name', 'surname', 'email',)
    model = models.Person

admin.site.register(models.Person, PersonAdmin)

class FileAdmin(admin.ModelAdmin):
    list_display = ('year', 'number', 'is_active', 'file_type')
    list_filter = ("file_type", "year", 'is_active',)
    search_fields = ('towns__name',)
    model = models.File

admin.site.register(models.File, FileAdmin)

class OperationAdmin(admin.ModelAdmin):
    list_display = ['name', 'operation_code', 'start_date', 'end_date',
                    'operation_type']
    list_filter = ("operation_type",)
    search_fields = ['towns__name', 'operation_code']
    if settings.COUNTRY == 'fr':
        list_display += ['code_patriarche']
        search_fields += ['code_patriarche']
    model = models.Operation

admin.site.register(models.Operation, OperationAdmin)

class ParcelAdmin(admin.ModelAdmin):
    list_display = ('section', 'parcel_number', 'operation',)
    search_fields = ('operation__name',)
    model = models.Parcel

admin.site.register(models.Parcel, ParcelAdmin)

class PeriodAdmin(admin.ModelAdmin):
    list_display = ('label', 'start_date', 'end_date', 'parent')
    model = models.Period

admin.site.register(models.Period, PeriodAdmin)

class DatingAdmin(admin.ModelAdmin):
    list_display = ('period', 'start_date', 'end_date', 'dating_type',
                    'quality')
    list_filter = ("period", 'dating_type', 'quality')
    model = models.Dating

admin.site.register(models.Dating, DatingAdmin)

class RegistrationUnitAdmin(admin.ModelAdmin):
    list_display = ('label', 'lenght', 'width',
                    'thickness', 'depth')
    list_filter = ('has_furniture',)
    search_fields = ('parcel__operation__name', "datings__period__label")
    model = models.RegistrationUnit

admin.site.register(models.RegistrationUnit, RegistrationUnitAdmin)

class SourceAdmin(admin.ModelAdmin):
    list_display = ('title', 'source_type',)
    list_filter = ('source_type',)
    search_fields = ('title',)
    model = models.Source

admin.site.register(models.Source, SourceAdmin)

class ItemAdmin(admin.ModelAdmin):
    list_display = ('label', 'material_type', 'dating', 'volume', 'weight',
                    'item_number', 'is_isolated')
    list_filter = ('material_type',)
    search_fields = ('label', 'registration_unit__parcel__operation__name',
                     "dating__period__label")
    model = models.Item

admin.site.register(models.Item, ItemAdmin)

class WarehouseAdmin(admin.ModelAdmin):
    list_display = ('name', 'warehouse_type', 'town')
    list_filter = ('warehouse_type',)
    search_fields = ('name', 'town')
    model = models.Warehouse

admin.site.register(models.Warehouse, WarehouseAdmin)

class AdministrativeActAdmin(admin.ModelAdmin):
    list_display = ('operation', 'act_type', 'signature_date')
    list_filter = ('act_type',)
    search_fields = ('operation__name',)
    model = models.AdministrativeAct

admin.site.register(models.AdministrativeAct, AdministrativeActAdmin)

class ContainerTypeAdmin(admin.ModelAdmin):
    list_display = ('label', 'reference', 'lenght', 'width', 'height',
                    'volume')
    model = models.ContainerType

admin.site.register(models.ContainerType, ContainerTypeAdmin)

class ContainerAdmin(admin.ModelAdmin):
    list_display = ('reference', 'location', 'container_type',)
    list_filter = ("container_type",)
    model = models.Container

admin.site.register(models.Container, ContainerAdmin)

class TownAdmin(admin.ModelAdmin):
    list_display = ['name',]
    if settings.COUNTRY == 'fr':
        list_display += ['canton']
    model = models.Town

admin.site.register(models.Town, TownAdmin)

class AuthorAdmin(admin.ModelAdmin):
    list_display = ['person', 'source', 'author_type']
    list_filter = ("author_type",)
    model = models.Author

admin.site.register(models.Author, AuthorAdmin)

class PropertyAdmin(admin.ModelAdmin):
    list_display = ['item', 'person', 'start_date', 'end_date']
    search_fields = ('item__label', 'person__name')
    model = models.Property

admin.site.register(models.Property, PropertyAdmin)

basic_models = [models.PersonType, models.IshtarUser, models.FileType,
                models.OperationType, models.DatingType, models.DatingQuality,
                models.SourceType, models.MaterialType, models.ParcelOwner,
                models.WarehouseType, models.ActType, models.Treatment,
                models.InCharge, models.AuthorType]
if settings.COUNTRY == 'fr':
    basic_models += [models.Arrondissement, models.Canton]

for model in basic_models:
    admin.site.register(model)