# -*- coding: utf-8 -*-
from django.shortcuts import  render_to_response
from django.http import HttpResponseRedirect, HttpResponse
from django.core.context_processors import csrf
from django.conf import settings
from django.contrib.contenttypes.models import ContentType
from django.contrib.auth.decorators import login_required
#models and forms
from models import Company
from forms import CompanyForm
from theme.models import Tag
from city.models import City
from file.models import FileModel, TmpFile
from file.forms import FileModelForm
#custom views
from functions.custom_views import objects_list, add_object_with_file


def company_all(request):
    """
    Return list of all companies with pagination
    """
    return objects_list(request, Company, 'company_all.html')


@login_required
def company_add(request):
    """
    Return form of company and post it on the server.
    If form is posted redirect on the page of all companies.
    """
    return add_object_with_file(request, CompanyForm, 'company_add.html', '/company/all/',
                                {'city': City, 'tag': Tag})


@login_required
def company_change(request, company_id):
    """
    Return form and fill it with existing Company object data.

    If form is posted redirect on the page of all companies.
    """
    try:
    #check if company_id exists else redirect to the list of companies
        company = Company.objects.get(id=company_id)
        file_form = FileModelForm(initial={'model': 'company.Company'})
    except:
        return HttpResponseRedirect('/company/all/')

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

        if form.is_valid():
            form.save(company_id)
            return HttpResponseRedirect('/company/all/')
    else:
        #fill form with data from database
        data = {'url':company.url, 'staff_number':company.staff_number, 'address': company.address,
                'phone':company.phone, 'fax':company.fax, 'web_page':company.web_page,
                'email':company.email, 'social':company.social, 'foundation': company.foundation}

        if company.country:
            data['country'] = company.country.id

        if company.city:
            data['city'] = company.city.id

        data['theme'] = [item.id for item in company.theme.all()]
        data['tag'] = [item.id for item in company.tag.all()]
        #data from translated fields
        for code, name in settings.LANGUAGES:
            obj = Company._meta.translations_model.objects.get(language_code = code,master__id=company_id) #access to translated fields
            data['name_%s' % code] = obj.name
            data['description_%s' % code] = obj.description
            data['specialization_%s' % code] = obj.specialization
            data['address_inf_%s' % code] = obj.address_inf
            data['title_%s' % code] = obj.title
            data['keywords_%s' % code] = obj.keywords
            data['descriptions_%s' % code] = obj.descriptions
        #fill form
        form = CompanyForm(data)
        #set choices filled by ajax
        form.fields['city'].choices = [(item.id, item.name) for item in City.objects.filter(country=data['country'])]
        form.fields['tag'].choices = [(item.id, item.name) for item in Tag.objects.filter(theme__in=data['theme'])]

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

    args['form'] = form
    args['languages'] = settings.LANGUAGES
    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(company), object_id=getattr(company, 'id'))
    args['obj_id'] = company_id

    return render_to_response('company_add.html', args)