You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
3.7 KiB
113 lines
3.7 KiB
# -*- coding: utf-8 -*-
|
|
from django.core.context_processors import csrf
|
|
from django.shortcuts import render_to_response
|
|
from django.http import HttpResponseRedirect, HttpResponse
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.db.models.loading import get_model
|
|
#
|
|
import xlwt
|
|
import xlrd
|
|
#
|
|
from custom_forms import ImportEventForm, ExportEventForm
|
|
from excel_settings import field_settings, import_settings
|
|
|
|
relation_fields = ['city']
|
|
@login_required
|
|
def import_event(request):
|
|
args = {}
|
|
form = ImportEventForm()
|
|
args.update(csrf(request))
|
|
args['form'] = form
|
|
|
|
if request.POST:
|
|
|
|
form = ImportEventForm(request.POST, request.FILES)
|
|
if form.is_valid():
|
|
|
|
form.save_events()
|
|
|
|
args['message'] = 'Success'
|
|
return render_to_response('import_event.html', args)
|
|
|
|
return render_to_response('import_event.html', args)
|
|
|
|
|
|
|
|
def xls_to_response(xls, fname):
|
|
response = HttpResponse(mimetype="application/ms-excel")
|
|
response['Content-Disposition'] = 'attachment; filename=%s' % fname
|
|
xls.save(response)
|
|
return response
|
|
|
|
|
|
def export_event(request):
|
|
if request.POST:
|
|
form = ExportEventForm(request.POST)
|
|
if form.is_valid():
|
|
workbook = form.export_events()
|
|
if workbook is None:
|
|
pass
|
|
else:
|
|
return xls_to_response(workbook, 'My Worksheet.xls')
|
|
"""
|
|
data = form.cleaned_data
|
|
# get model
|
|
model = get_model(data['event'].split('.')[0], data['event'].split('.')[1])
|
|
# get objects
|
|
objects = model.objects.language(data['language']).filter(data_begin__range=[data['date_from'], data['date_to']],
|
|
country__in=data['country'],
|
|
theme__in=data['theme']
|
|
).distinct()
|
|
|
|
if not objects:
|
|
# not found any objects
|
|
message = 'По даному запросу объектов не найдено'
|
|
args = {}
|
|
args.update(csrf(request))
|
|
args['message'] = message
|
|
args['form'] = form
|
|
return render_to_response('export_event.html', args)
|
|
|
|
workbook = xlwt.Workbook(encoding = 'utf8')
|
|
# new tab
|
|
worksheet = workbook.add_sheet('My Worksheet')
|
|
|
|
# Create the Font for first rows
|
|
font = xlwt.Font()
|
|
font.name = 'Times New Roman'
|
|
font.bold = True
|
|
style = xlwt.XFStyle()
|
|
# Create the Style
|
|
style.font = font
|
|
|
|
for row, object in enumerate(objects):
|
|
# column number
|
|
col = 0
|
|
for field in field_settings:
|
|
try:
|
|
value = getattr(object, field['name'])
|
|
except AttributeError:
|
|
# doesnt have such field. exit from iteration
|
|
continue
|
|
|
|
if row == 0:
|
|
# first iteration. set names. set columns width
|
|
worksheet.write(0, col, field.get('verbose_name', 'default'), style)
|
|
worksheet.write(1, col, field.get('name'), style)
|
|
worksheet.col(col).width = field.get('width', 3333)
|
|
|
|
worksheet.write(row+2, col, field.get('type')(value))
|
|
col += 1
|
|
"""
|
|
|
|
#return xls_to_response(workbook, 'My Worksheet.xls')
|
|
else:
|
|
form = ExportEventForm()
|
|
|
|
|
|
args = {}
|
|
args.update(csrf(request))
|
|
|
|
args['form'] = form
|
|
|
|
return render_to_response('export_event.html', args) |