|
|
|
|
@ -9,6 +9,7 @@ from organiser.models import Organiser |
|
|
|
|
from django.db.models.loading import get_model |
|
|
|
|
import xlrd, xlwt |
|
|
|
|
from excel_settings import import_settings |
|
|
|
|
from functions.form_check import translit_with_separator |
|
|
|
|
|
|
|
|
|
languages = [code for code in settings.LANGUAGES] |
|
|
|
|
|
|
|
|
|
@ -211,6 +212,10 @@ class ImportPlaceConferenceForm(ImportForm): |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from import_xls.excel_settings import event_sett |
|
|
|
|
typical_errors = {'(1048, "Column \'city_id\' cannot be null")':u'Неправильная страна или город', |
|
|
|
|
'(1048, "Column \'country_id\' cannot be null")':u'Неправильная страна или город', |
|
|
|
|
'(1048, "Column \'data_end\' cannot be null")':u'НЕправильный формат или не заполнена дата окончания', |
|
|
|
|
'(1048, "Column \'data_end\' cannot be null")':u'НЕправильный формат или не заполнена дата начала'} |
|
|
|
|
|
|
|
|
|
class ImportEventForm(ImportForm): |
|
|
|
|
""" |
|
|
|
|
@ -221,40 +226,132 @@ class ImportEventForm(ImportForm): |
|
|
|
|
('conference.Conference', 'Конференция'), |
|
|
|
|
('seminar.Seminar', 'Семинар'), |
|
|
|
|
('webinar.Webinar', 'Вебинар')]) |
|
|
|
|
def save_file(self): |
|
|
|
|
""" |
|
|
|
|
save events from excel file |
|
|
|
|
in language from form |
|
|
|
|
""" |
|
|
|
|
|
|
|
|
|
def save_file_debug(self): |
|
|
|
|
data = self.cleaned_data |
|
|
|
|
lang = data['language'] |
|
|
|
|
f = data['excel_file'] |
|
|
|
|
book = xlrd.open_workbook(file_contents=f.read()) |
|
|
|
|
sheet = book.sheet_by_index(0) |
|
|
|
|
row_list = [sheet.row_values(row_number) for row_number in range(sheet.nrows)] |
|
|
|
|
# all field names in excel file (must be in second row) |
|
|
|
|
field_names = [name for name in row_list[0]] |
|
|
|
|
# model |
|
|
|
|
model = get_model(data['event'].split('.')[0], data['event'].split('.')[1]) |
|
|
|
|
|
|
|
|
|
list_dicts = [] |
|
|
|
|
labels = [label for label in row_list[0]] |
|
|
|
|
errors = [] |
|
|
|
|
|
|
|
|
|
for row_number, row in enumerate(row_list[1:]): |
|
|
|
|
d = {} |
|
|
|
|
for row_number, row in enumerate(row_list): |
|
|
|
|
if row_number == 0: |
|
|
|
|
continue |
|
|
|
|
|
|
|
|
|
if row[0] != '': |
|
|
|
|
# in first column id |
|
|
|
|
try: |
|
|
|
|
obj = self.model.objects.language(lang).get(id=int(row[0])) |
|
|
|
|
except ValueError: |
|
|
|
|
obj = self.model() |
|
|
|
|
obj.translate(lang) |
|
|
|
|
|
|
|
|
|
except self.model.DoesNotExist: |
|
|
|
|
obj = self.model(id= int(row[0])) |
|
|
|
|
obj.translate(lang) |
|
|
|
|
else: |
|
|
|
|
# if id blank - its a new event |
|
|
|
|
obj = model() |
|
|
|
|
obj.translate(lang) |
|
|
|
|
|
|
|
|
|
methods = [] |
|
|
|
|
for col_number, cell in enumerate(row): |
|
|
|
|
# through row |
|
|
|
|
field_name = field_names[col_number] |
|
|
|
|
setting = event_sett.get(field_name) |
|
|
|
|
if setting: |
|
|
|
|
d[setting['field']] = {'value':cell} |
|
|
|
|
# go through row cells |
|
|
|
|
# field name current cell |
|
|
|
|
label = labels[col_number] |
|
|
|
|
|
|
|
|
|
setting = event_sett.get(label) |
|
|
|
|
|
|
|
|
|
if setting is None: |
|
|
|
|
continue |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if setting.get('method'): |
|
|
|
|
# this cell contains data that must be written after creating object |
|
|
|
|
if cell != "": |
|
|
|
|
methods.append({'func': setting['func'], 'value': cell, 'purpose': setting.get('purpose')}) |
|
|
|
|
continue |
|
|
|
|
|
|
|
|
|
field_name = setting['field'] |
|
|
|
|
func = setting.get('func') |
|
|
|
|
if func is None: |
|
|
|
|
continue |
|
|
|
|
|
|
|
|
|
extra_value = setting.get('extra_values') |
|
|
|
|
if extra_value is not None: |
|
|
|
|
# if setting has extra value then |
|
|
|
|
# it is some field like city, theme, tag |
|
|
|
|
# that has relation and can be created |
|
|
|
|
|
|
|
|
|
# in function we add language(need for relation fields) |
|
|
|
|
# and extra value from object (like for city need country) |
|
|
|
|
try: |
|
|
|
|
extra = getattr(obj, extra_value) |
|
|
|
|
except Exception: |
|
|
|
|
continue |
|
|
|
|
value = func(cell, 'ru', extra) |
|
|
|
|
elif setting.get('bitfield'): |
|
|
|
|
value = func(obj, cell, setting['label']) |
|
|
|
|
|
|
|
|
|
else: |
|
|
|
|
value = func(cell) |
|
|
|
|
try: |
|
|
|
|
setattr(obj, field_name, value) |
|
|
|
|
except ValueError: |
|
|
|
|
continue |
|
|
|
|
if not obj.url: |
|
|
|
|
obj.url = translit_with_separator(obj.name) |
|
|
|
|
try: |
|
|
|
|
obj.save() |
|
|
|
|
|
|
|
|
|
except IntegrityError, e: |
|
|
|
|
error = str(e) |
|
|
|
|
if typical_errors.get(error): |
|
|
|
|
error = typical_errors[error] |
|
|
|
|
if error.startswith('(1062, "Duplicate entry') and error.endswith('for key \'url\'")'): |
|
|
|
|
error = u'Событие с таким названием или урлом уже существует' |
|
|
|
|
|
|
|
|
|
errors.append([obj.name, error]) |
|
|
|
|
continue |
|
|
|
|
|
|
|
|
|
for method in methods: |
|
|
|
|
func = method['func'] |
|
|
|
|
if method.get('purpose'): |
|
|
|
|
try: |
|
|
|
|
func(obj, method['value'], method['purpose']) |
|
|
|
|
except: |
|
|
|
|
continue |
|
|
|
|
else: |
|
|
|
|
func(obj, method['value']) |
|
|
|
|
|
|
|
|
|
d['lang'] = lang |
|
|
|
|
list_dicts.append(d) |
|
|
|
|
#------------- |
|
|
|
|
|
|
|
|
|
return list_dicts |
|
|
|
|
return errors |
|
|
|
|
|
|
|
|
|
def save_file(self): |
|
|
|
|
""" |
|
|
|
|
save events from excel file |
|
|
|
|
in language from form |
|
|
|
|
""" |
|
|
|
|
data = self.cleaned_data |
|
|
|
|
lang = data['language'] |
|
|
|
|
f = data['excel_file'] |
|
|
|
|
book = xlrd.open_workbook(file_contents=f.read()) |
|
|
|
|
sheet = book.sheet_by_index(0) |
|
|
|
|
row_list = [sheet.row_values(row_number) for row_number in range(sheet.nrows)] |
|
|
|
|
# all field names in excel file (must be in second row) |
|
|
|
|
field_names = [name for name in row_list[0]] |
|
|
|
|
# model |
|
|
|
|
model = get_model(data['event'].split('.')[0], data['event'].split('.')[1]) |
|
|
|
|
|
|
|
|
|
list_dicts = [] |
|
|
|
|
|
|
|
|
|
for row_number, row in enumerate(row_list): |
|
|
|
|
# go through all rows in file |
|
|
|
|
@ -320,7 +417,6 @@ class ImportEventForm(ImportForm): |
|
|
|
|
""" |
|
|
|
|
|
|
|
|
|
#object.save() |
|
|
|
|
sadsa |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|