def get_context(self, request, encounter_id, service_cls, object_id):
try:
cls = get_model_for_class_name(service_cls)
except ValueError:
raise Http404
encounter = get_object_or_404(Encounter, pk=encounter_id)
ctx = {
'encounter': encounter,
'cls': service_cls,
'service': cls,
'is_edit': object_id is not None
}
kwargs = {}
if ctx['is_edit']:
obj = get_object_or_404(cls, pk=object_id)
ctx.update({
'obj':
obj,
'action_link':
reverse('services_handle_form_change',
kwargs={
'encounter_id': encounter_id,
'service_cls': service_cls,
'object_id': object_id
})
})
kwargs['instance'] = obj
else:
ctx.update({
'action_link':
reverse('services_handle_form_add',
kwargs={
'encounter_id': encounter_id,
'service_cls': service_cls
})
})
if request.method == 'POST':
form = cls.form()(encounter, request.POST, **kwargs)
else:
form = cls.form()(encounter, **kwargs)
ctx.update({'form': form})
return ctx
def get_context(self, request, encounter_id, service_cls, object_id):
try:
cls = get_model_for_class_name(service_cls)
except ValueError:
raise Http404
encounter = get_object_or_404(Encounter, pk=encounter_id)
ctx = {
'encounter': encounter,
'cls': service_cls,
'service': cls,
'is_edit': object_id is not None
}
kwargs = {}
if ctx['is_edit']:
obj = get_object_or_404(cls, pk=object_id)
ctx.update({
'obj': obj,
'action_link': reverse('services_handle_form_change', kwargs={
'encounter_id': encounter_id, 'service_cls': service_cls, 'object_id': object_id
})
})
kwargs['instance'] = obj
else:
ctx.update({
'action_link': reverse('services_handle_form_add', kwargs={
'encounter_id': encounter_id, 'service_cls': service_cls
})
})
if request.method == 'POST':
form = cls.form()(encounter, request.POST, **kwargs)
else:
form = cls.form()(encounter, **kwargs)
ctx.update({'form': form})
return ctx
def test_get_model_for_class_name_raises_valueerror_for_non_registered_class(self):
get_model_for_class_name('Dummy3')
def test_get_model_for_class_name_returns_correct_class(self):
tools.assert_equals(get_model_for_class_name('Dummy1'), self.d1)
tools.assert_equals(get_model_for_class_name('Dummy2'), self.d2)
