def post(self, request):
"""View's dispatch method automatically calls this"""
try:
workbook = WorkbookJSONReader(request.file)
except AttributeError:
return HttpResponseBadRequest("Error processing your Excel (.xlsx) file")
try:
data_types = workbook.get_worksheet(title='types')
except KeyError:
return HttpResponseBadRequest("Workbook does not have a sheet called 'types'")
for dt in data_types:
print "DataType"
print dt
for di in workbook.get_worksheet(title=dt['tag']):
print "DataItem"
print di
for dt in data_types:
data_type = FixtureDataType(
domain=self.domain,
name=dt['name'],
tag=dt['tag'],
fields=dt['field'],
)
data_type.save()
data_items = workbook.get_worksheet(data_type.tag)
for di in data_items:
print di
data_item = FixtureDataItem(
domain=self.domain,
data_type_id=data_type.get_id,
fields=di['field']
)
data_item.save()
for group_name in di.get('group', []):
group = Group.by_name(self.domain, group_name)
if group:
data_item.add_group(group)
else:
messages.error(request, "Unknown group: %s" % group_name)
for raw_username in di.get('user', []):
username = normalize_username(raw_username, self.domain)
user = CommCareUser.get_by_username(username)
if user:
data_item.add_user(user)
else:
messages.error(request, "Unknown user: %s" % raw_username)
return HttpResponseRedirect(reverse('fixture_view', args=[self.domain]))
def _run_upload(self, request, workbook):
group_memoizer = GroupMemoizer(self.domain)
data_types = workbook.get_worksheet(title='types')
def _get_or_raise(container, attr, message):
try:
return container[attr]
except KeyError:
raise Exception(message.format(attr=attr))
with CouchTransaction() as transaction:
for dt in data_types:
err_msg = "Workbook 'types' has no column '{attr}'"
data_type = FixtureDataType(
domain=self.domain,
name=_get_or_raise(dt, 'name', err_msg),
tag=_get_or_raise(dt, 'tag', err_msg),
fields=_get_or_raise(dt, 'field', err_msg),
)
transaction.save(data_type)
data_items = workbook.get_worksheet(data_type.tag)
for sort_key, di in enumerate(data_items):
data_item = FixtureDataItem(
domain=self.domain,
data_type_id=data_type.get_id,
fields=di['field'],
sort_key=sort_key
)
transaction.save(data_item)
for group_name in di.get('group', []):
group = group_memoizer.by_name(group_name)
if group:
data_item.add_group(group, transaction=transaction)
else:
messages.error(request, "Unknown group: %s" % group_name)
for raw_username in di.get('user', []):
username = normalize_username(raw_username, self.domain)
user = CommCareUser.get_by_username(username)
if user:
data_item.add_user(user)
else:
messages.error(request, "Unknown user: %s" % raw_username)
for data_type in transaction.preview_save(cls=FixtureDataType):
for duplicate in FixtureDataType.by_domain_tag(domain=self.domain, tag=data_type.tag):
duplicate.recursive_delete(transaction)
def _run_upload(self, request, workbook):
group_memoizer = GroupMemoizer(self.domain)
data_types = workbook.get_worksheet(title='types')
with CouchTransaction() as transaction:
for dt in data_types:
data_type = FixtureDataType(
domain=self.domain,
name=dt['name'],
tag=dt['tag'],
fields=dt['field'],
)
transaction.save(data_type)
data_items = workbook.get_worksheet(data_type.tag)
for di in data_items:
data_item = FixtureDataItem(
domain=self.domain,
data_type_id=data_type.get_id,
fields=di['field']
)
transaction.save(data_item)
for group_name in di.get('group', []):
group = group_memoizer.by_name(group_name)
if group:
data_item.add_group(group, transaction=transaction)
else:
messages.error(request, "Unknown group: %s" % group_name)
for raw_username in di.get('user', []):
username = normalize_username(raw_username, self.domain)
user = CommCareUser.get_by_username(username)
if user:
data_item.add_user(user)
else:
messages.error(request, "Unknown user: %s" % raw_username)
for data_type in transaction.preview_save(cls=FixtureDataType):
for duplicate in FixtureDataType.by_domain_tag(domain=self.domain, tag=data_type.tag):
duplicate.recursive_delete(transaction)
