expomap/place_exposition/admin.py

# -*- coding: utf-8 -*-
from django.shortcuts import  render_to_response
from django.http import HttpResponseRedirect
from django.core.context_processors import csrf
from django.conf import settings
from django.forms.formsets import formset_factory
from django.forms.models import modelformset_factory
from django.contrib.contenttypes.models import ContentType
from django.contrib.auth.decorators import login_required
from django.forms.formsets import BaseFormSet, formset_factory
from django.forms.models import modelformset_factory
#models and forms
from forms import ExpositionForm, PlaceExpositionFormDelete, HallForm
from models import PlaceExposition, Hall
from city.models import City
from file.models import FileModel, TmpFile
from file.forms import FileModelForm
#python
import random
#custom fields
from functions.custom_views import objects_list, delete_object


def exposition_all(request):
    """
    Return list of all place_expositions with pagination
    """
    return objects_list(request, PlaceExposition, 'place_exposition_all.html')


@login_required
def exposition_add(request):
    """
    Returns form of place_exposition and formset of pavilions and post it on the server.

    If forms is posted redirect on the page of all place_expositions.
    """
    #formset of HallForm
    HallFormSet = formset_factory(HallForm)
    if request.POST.get('key'):
        key = request.POST['key']
    else:
        key = random.getrandbits(128)

    file_form = FileModelForm(initial={'key': key})

    if request.POST:
        form = ExpositionForm(request.POST)
        #set choices filled by ajax
        form.fields['city'].choices = [(item.id, item.name) for item in City.objects.filter(country=request.POST['country'])]

        formset = HallFormSet(request.POST)

        if form.is_valid() and formset.is_valid():
            place_exposition = form.save()

            for item in formset.forms:
            #saves forms if its valid and not empty
                if item.is_valid() and item.has_changed():
                    hall = item.save(commit=False)
                    hall.place_exposition = place_exposition
                    hall.save()

            return HttpResponseRedirect('/admin/place_exposition/all')
    else:
        form = ExpositionForm(initial={'key': key})
        formset = HallFormSet()

    args = {}
    args.update(csrf(request))

    args['languages'] = settings.LANGUAGES
    args['form'] = form
    args['formset'] = formset
    args['file_form'] = file_form
    args['files'] = TmpFile.objects.filter(key=key)

    return render_to_response('place_exposition_add.html', args)


def exposition_delete(request, url):
    return delete_object(request, PlaceExposition, PlaceExpositionFormDelete, url, '/admin/place_exposition/all')


@login_required
def exposition_change(request, url):
    """
    Return form of place_expositions and formset of pavilions
    and fill it with existing PlaceExposition and Pavilion object data.

    If form of conference is posted redirect on the page of all conferences.

    """
    try:
    #check if exposition_id exists else redirect to the list of place of conference
        place = PlaceExposition.objects.get(url=url)
        exposition_id = getattr(place, 'id')
        file_form = FileModelForm(initial={'model': 'place_exposition.PlaceExposition'})
    except:
        return HttpResponseRedirect('/admin/place_exposition/all')

    if request.POST:
        #formset of HallForm
        HallFormSet = formset_factory(HallForm)
        form = ExpositionForm(request.POST)
        #set choices filled by ajax
        form.fields['city'].choices = [(item.id, item.name) for item in City.objects.filter(country=request.POST['country'])]
        formset = HallFormSet(request.POST)

        if form.is_valid() and formset.is_valid():
            place_exposition = form.save(exposition_id)
            #delete old halls
            Hall.objects.filter(place_exposition=getattr(place, 'id')).delete()
            for item in formset.forms:
            #saves new halls if its valid and not empty
                if item.is_valid() and item.has_changed():
                    hall = item.save(commit=False)
                    hall.place_exposition = place_exposition
                    hall.save()

            return HttpResponseRedirect('/admin/place_exposition/all')
    else:
        #initial HallFormSet
        HallFormSet = modelformset_factory(Hall, form=HallForm, exclude=('place_exposition',))
        #fill form with data from database
        data= {'type': place.type, 'address': place.address,
                    'phone': place.phone, 'fax': place.fax, 'web_page': place.web_page, 'email': place.email,
                    'foundation_year': place.foundation_year, 'total_area': place.total_area,
                    'closed_area': place.closed_area, 'open_area': place.open_area,
                    'total_pavilions': place.total_pavilions, 'total_halls': place.total_halls, 'wifi':place.wifi,
                    'bank': place.bank, 'children_room': place.children_room,
                    'disabled_service': place.disabled_service, 'conference_centre': place.conference_centre,
                    'business_centre': place.business_centre, 'online_registration': place.online_registration,
                    'cafe': place.cafe, 'terminals': place.terminals, 'parking': place.parking,
                    'press_centre': place.press_centre, 'mobile_application': place.mobile_application,
                    'place_exposition_id':place.id}

        if place.country:
            data['country'] = place.country.id
        if place.city:
            data['city'] = place.city.id
        #data from translated fields
        for code, name in settings.LANGUAGES:
            obj = PlaceExposition._meta.translations_model.objects.get(language_code = code,master__id=exposition_id) #access to translated fields
            data['name_%s'%code] = obj.name
            data['description_%s'%code] = obj.description
            data['adress_%s'%code] = obj.adress
            data['total_year_action_%s'%code] = obj.total_year_action
            data['title_%s'%code] = obj.title
            data['keywords_%s'%code] = obj.keywords
            data['descriptions_%s'%code] = obj.descriptions

        form = ExpositionForm(initial=data)
        #set choices filled by ajax
        form.fields['city'].choices = [(item.id, item.name) for item in City.objects.filter(country=data['country'])]
        #get existing halls
        halls = Hall.objects.filter(place_exposition=getattr(place, 'id'))
        #fill HallFormSet
        formset = HallFormSet(queryset=halls)


    args = {}
    args.update(csrf(request))

    args['languages'] = settings.LANGUAGES
    args['form'] = form
    args['formset'] = formset
    args['file_form'] = file_form

    #get list of files which connected with specific model object
    args['files'] = FileModel.objects.filter(content_type=ContentType.objects.get_for_model(place),
                                             object_id=getattr(place, 'id'))
    args['obj_id'] = exposition_id


    return render_to_response('place_exposition_add.html', args)