expomap/import_xls/excel_settings.py

# -*- coding: utf-8 -*-
import xlwt
import datetime

def get_bool(value):
    if value:
        return 1
    return 0

def get_int(value):
    if not value:
        return ''
    return value

def get_audience(value):
    if not value:
        return 'Не выбрано'
    elif value == 'experts':
        return 'Специалисты'
    elif value == 'experts and consumers':
        return 'Специалисты и потребители'
    elif value == 'general public':
        return 'Широкая публика'
    else:
        return ''

field_settings = [
    {'name': 'id', 'verbose_name': 'id', 'type': get_int, 'width':1500},
    {'name': 'url', 'verbose_name': 'url', 'type': str},
    {'name': 'name', 'verbose_name': 'Имя', 'type': str, 'width':8000,},
    {'name': 'data_begin', 'verbose_name': 'Дата начала', 'type': str},
    {'name': 'data_end', 'verbose_name': 'Дата окончания', 'type': str},
    {'name': 'main_title', 'verbose_name': 'Краткое описание', 'type': str},
    {'name': 'description', 'verbose_name': 'Описание', 'type': str},
    {'name': 'audience', 'verbose_name': 'Аудитория', 'type': get_audience},
    {'name': 'country', 'verbose_name': 'Страна', 'type': str},
    {'name': 'city', 'verbose_name': 'Город', 'type': str},
    {'name': 'periodic', 'verbose_name': 'Периодичность', 'type': str},
    {'name': 'web_page', 'verbose_name': 'Веб страница', 'type': str},
    {'name': 'time', 'verbose_name': 'Время проведения', 'type': str},
    {'name': 'products', 'verbose_name': 'Экспонируемые продукты', 'type': str},
    {'name': 'foundation_year', 'verbose_name': 'Год основания', 'type': get_int},
    {'name': 'tax', 'verbose_name': 'Налог включен', 'type': get_bool, 'width':1000},
    {'name': 'currency', 'verbose_name': 'Валюта', 'type': str},
    {'name': 'max_price', 'verbose_name': 'Максимальная цена', 'type': get_int},
    {'name': 'min_price', 'verbose_name': 'Минимальная цена', 'type': get_int},
    {'name': 'registration_payment', 'verbose_name': 'Регистрационный взнос', 'type': get_int},
    {'name': 'min_closed_area', 'verbose_name': 'Минимальная цена закрытой НЕ оборудованной площади', 'type': get_int},
    {'name': 'max_closed_area', 'verbose_name': 'Максимальная цена закрытой НЕ оборудованной площади', 'type': get_int},
    {'name': 'min_closed_equipped_area', 'verbose_name': 'Минимальная цена закрытой оборудованной площади ', 'type': get_int},
    {'name': 'max_closed_equipped_area', 'verbose_name': 'Максимальная цена закрытой оборудованной площади', 'type': get_int},
    {'name': 'min_open_area', 'verbose_name': 'Минимальная цена закрытой площади', 'type': get_int},
    {'name': 'max_open_area', 'verbose_name': 'Максимальная цена открытой площади', 'type': get_int},
    {'name': 'min_area', 'verbose_name': 'Минимальная площадь', 'type': get_int},
    {'name': 'max_area', 'verbose_name': 'Максимальная площадь', 'type': get_int},
    {'name': 'is_published', 'verbose_name': 'Опубликована', 'type': get_bool, 'width':1000},
    {'name': 'canceled_by_administrator', 'verbose_name': 'Отменена администратором', 'type': get_bool, 'width':1000}
    ]



def to_int(val):
    """
    Reverse function to get_int
    return None if value isnt integer
    """
    try:
        return int(val)
    except ValueError:
        return None

from country.models import Country
from hvad.utils import get_translation_aware_manager
def to_country(value):
    try:
        query = get_translation_aware_manager(Country)
        country = query.filter(name=value)[0]
        return country
    except IndexError:
        return None


from city.models import City
def to_city(value):
    try:
    # get city by name
        objects = get_translation_aware_manager(City)
        # except IndexError if no found
        city = objects.filter(name=value)[0]
        return city
    except IndexError:
        return None

AUDIENCE1 = ((None,'Не выбрано'),
            ('experts', 'Специалисты'),
            ('experts and consumers', 'Специалисты и потребители'),
            ('general public', 'Широкая публика'))

def to_audience(value):
    try:
        if value == 'Специалисты':
            return 'experts'
        elif value == 'Специалисты и потребители':
            return 'experts and consumers'
        elif value == 'Широкая публика':
            return 'general public'
        else:
            return ''
    except:
        return ''


def get_cell(field_name, value):
    return {#'id': int(value),
            'url': str(value),
            'name': str(value),
            'data_begin': str(value),
            'data_end': str(value),
            'main_title': str(value),
            'description': str(value),
            'audience': to_audience(value),
            'country': to_country(value),
            'city': to_city(value),
            #'periodic': str(value),
            'web_page': str(value),
            'time': str(value),
            'products': str(value),
            'foundation_year': to_int(value),
            'tax': bool(value),
            'currency': str(value),
            'max_price': to_int(value),
            'min_price': to_int(value),
            'registration_payment': to_int(value),
            'min_closed_area': to_int(value),
            'max_closed_area': to_int(value),
            'min_closed_equipped_area': to_int(value),
            'max_closed_equipped_area': to_int(value),
            'min_open_area': to_int(value),
            'max_open_area': to_int(value),
            'min_area': to_int(value),
            'max_area': to_int(value),
            'is_published': bool(value),
            'canceled_by_administrator': bool(value)
            }.get(field_name)