def add(request, app, cls, pk=None, related_field_name=None, related_pk=None):
if not request.user.is_authenticated:
return httprr(request,
'/admin/login/?next={}'.format(request.get_full_path()))
try:
_model = apps.get_model(app, cls)
except LookupError as e:
return page_not_found(request, e, 'error404.html')
obj = pk and _model.objects.all(
request.user).filter(pk=pk).first() or _model()
obj.request = request
obj._user = request.user
title = pk and str(obj) or get_metadata(_model, 'verbose_name')
if related_field_name is None:
if obj.pk:
if not permissions.has_edit_permission(
request, _model) or not permissions.can_edit(request, obj):
return HttpResponseForbidden()
else:
if not permissions.has_add_permission(
request, _model) or not permissions.can_add(request, obj):
return HttpResponseForbidden()
form = factory.get_register_form(request, obj)
title = form.title
elif is_one_to_many(_model, related_field_name):
if not permissions.can_add(request, obj) and not permissions.can_edit(
request, obj):
return HttpResponseForbidden()
form = factory.get_one_to_many_form(request, obj, related_field_name)
elif is_many_to_many(_model, related_field_name):
if not permissions.can_add(request, obj) and not permissions.can_edit(
request, obj):
return HttpResponseForbidden()
form = factory.get_many_to_many_form(request, obj, related_field_name,
related_pk)
elif is_one_to_many_reverse(_model, related_field_name):
form = factory.get_many_to_many_reverse_form(request, obj,
related_field_name)
elif is_one_to_one(_model, related_field_name):
if not permissions.can_add(request, obj) and not permissions.can_edit(
request, obj):
return HttpResponseForbidden()
form = factory.get_one_to_one_form(request, obj, related_field_name,
related_pk)
else:
# many to one
for rel in list_related_objects(_model):
if hasattr(rel, 'get_accessor_name'):
if rel.get_accessor_name() in (
'{}_set'.format(related_field_name),
related_field_name):
related_queryset = rel.related_model.objects.all(
request.user)
related_obj = related_pk and related_queryset.get(
pk=related_pk) or rel.related_model()
related_obj.request = request
setattr(related_obj, rel.field.name, obj)
setattr(related_obj, '{}_id'.format(rel.field.name),
obj.pk)
if related_pk:
if not permissions.has_edit_permission(
request,
rel.related_model) or not permissions.can_edit(
request, related_obj):
return HttpResponseForbidden()
else:
if not permissions.has_add_permission(
request,
rel.related_model) or not permissions.can_add(
request, related_obj):
return HttpResponseForbidden()
form = factory.get_many_to_one_form(
request, obj, rel.get_accessor_name(), related_obj)
title = form.title
if form.is_valid():
is_editing = form.instance.pk is not None
try:
form.save()
obj = form.instance
if 'select' in request.GET:
return HttpResponse('{}|{}|{}'.format(obj.pk, obj,
request.GET['select']))
elif related_field_name:
message = 'Ação realizada com sucesso'
url = '..'
else:
message = get_metadata(form.instance.__class__, 'add_message')
if message and not is_editing:
if hasattr(obj, 'get_absolute_url'):
url = obj.get_absolute_url()
else:
url = '/view/{}/{}/{}/'.format(
get_metadata(obj.__class__, 'app_label'),
obj.__class__.__name__.lower(), obj.pk)
else:
url = '..'
if is_editing:
message = message or 'Atualização realizada com sucesso'
else:
message = message or 'Cadastro realizado com sucesso'
return httprr(request, url, message)
except ValidationError as e:
form.add_error(None, str(e.message))
return render(request, 'default.html', locals())
def __init__(self,
request,
obj,
current_tab=None,
parent=None,
fieldsets=None,
complete=True):
super(ModelPanel, self).__init__(request=request)
self.obj = obj
self.request = request
self.title = unicode(obj)
self.tabs = []
self.current_tab = current_tab
self.message = None
self.complete = complete
self.drop_down = None
fieldsets = fieldsets or get_metadata(type(obj), 'view_fieldsets', [])
if not fieldsets:
fields = []
for field in get_metadata(type(obj), 'fields')[1:]:
fields.append(field.name)
for field in get_metadata(type(obj), 'local_many_to_many'):
fields.append(field.name)
fieldsets = ((u'Dados Gerais', dict(fields=fields)), )
if self.complete:
self.drop_down = ModelDropDown(self.request, type(self.obj))
self.drop_down.add_actions(self.obj, fieldset_title='')
else:
self.drop_down = GroupDropDown(self.request)
self.fieldsets_with_tab_name = []
self.fieldsets_without_tab_name = []
model = type(self.obj)
obj.as_pdf = self.as_pdf
for fieldset in fieldsets:
title, info = fieldset
tab_name = None
drop_down = ModelDropDown(self.request, model)
fieldset_actions = info.get('actions', [])
if fieldset_actions:
drop_down.add_actions(self.obj, fieldset_title=title)
if '::' in title:
tab_name, title = title.split('::')
url = '/view/%s/%s/%s/%s/' % (get_metadata(
model, 'app_label'), model.__name__.lower(), self.obj.pk,
slugify(tab_name))
tab = (tab_name, url)
if not self.tabs and not self.current_tab:
self.current_tab = slugify(tab_name)
if tab not in self.tabs:
self.tabs.append(tab)
if not tab_name or slugify(
tab_name) == self.current_tab or self.as_pdf:
fieldset_dict = dict(title=title or u'Dados Gerais',
tab_name=tab_name,
fields=[],
paginators=[],
drop_down=drop_down,
image=None)
relations = fieldset[1].get('relations', [])
if tab_name or self.as_pdf:
self.fieldsets_with_tab_name.append(fieldset_dict)
else:
self.fieldsets_without_tab_name.append(fieldset_dict)
if 'can_view' in fieldset[1]:
can_view = fieldset[1]['can_view']
if not permissions.check_group_or_permission(
self.request, can_view):
continue
if 'condition' in fieldset[1]:
condition = fieldset[1]['condition']
self.obj.request = self.request
if not check_condition(condition, self.obj):
continue
if 'image' in fieldset[1]:
fieldset_dict['image'] = fieldset[1]['image']
if 'fields' in fieldset[1]:
for name_or_tuple in fieldset[1]['fields']:
if not type(name_or_tuple) == tuple:
name_or_tuple = (name_or_tuple, )
attr_names = []
for attr_name in name_or_tuple:
if attr_name != parent:
attr = getattr(model, attr_name)
field = None
if hasattr(attr, 'field_name'):
field = getattr(model, '_meta').get_field(
attr.field_name)
elif hasattr(attr, 'field'):
field = attr.field
if not field or not hasattr(
field, 'display') or field.display:
if is_one_to_one(
model, attr_name
) and attr.field.display == 'detail':
relations.append(attr_name)
elif is_one_to_many(
model, attr_name
) and field.display == 'detail':
relations.append(attr_name)
elif is_many_to_one(
model, attr_name
) and field.display == 'detail':
relations.append(attr_name)
elif is_many_to_many(
model, attr_name) and (
not hasattr(field, 'display')
or field.display == 'detail'):
relations.append(attr_name)
elif is_one_to_many_reverse(
model, attr_name):
relations.append(attr_name)
else:
verbose_name, lookup, sortable, to = get_fiendly_name(
model, attr_name, as_tuple=True)
if to and not should_filter_or_display(
self.request, model, to):
continue
attr_names.append(
dict(verbose_name=verbose_name,
name=attr_name))
if attr_names:
fieldset_dict['fields'].append(attr_names)
if self.complete:
for relation_name in relations:
if relation_name in [
field.name
for field in get_metadata(model, 'get_fields')
]:
relation_field = find_field_by_name(
model, relation_name)
relation = getattr(self.obj, relation_name)
if not relation and hasattr(
relation_field,
'rel') and relation_field.rel.to:
relation = relation_field.rel.to()
if hasattr(relation.__class__, 'pk'):
if relation.pk:
fieldset_title = relation_field.verbose_name
panel_fieldsets = get_metadata(
type(relation), 'view_fieldsets', [])
panel_fieldsets = (
(fieldset_title,
panel_fieldsets[0][1]), )
panel = ModelPanel(
request,
relation,
fieldsets=panel_fieldsets,
complete=False)
if is_one_to_one(model, relation_name):
app_label = get_metadata(
model, 'app_label')
model_name = model.__name__.lower()
related_model_name = type(
relation).__name__.lower()
add_url = '/add/%s/%s/%s/%s' % (
app_label, model_name, self.obj.pk,
relation_name)
delete_url = None
if relation.pk:
add_url = '%s/%s/' % (add_url,
relation.pk)
app_label = get_metadata(
type(relation), 'app_label')
delete_url = '/delete/%s/%s/%s/' % (
app_label, related_model_name,
relation.pk)
if permissions.has_add_permission(
self.request, model
) or permissions.has_edit_permission(
self.request, model):
if delete_url:
panel.drop_down.add_action(
'Excluir %s' %
relation_field.
verbose_name, delete_url,
'popup', 'fa-close', None)
panel.drop_down.add_action(
'Atualizar %s' %
relation_field.verbose_name,
add_url, 'popup', 'fa-edit')
fieldset_dict['paginators'].append(panel)
else:
fieldset_title = len(
relations
) > 1 and title or relation_field.verbose_name
if is_one_to_many(
model,
relation_name) or is_many_to_many(
model, relation_name):
to = model.__name__.lower()
else:
to = relation_name
related_paginator = Paginator(
self.request,
relation.all(),
title=fieldset_title,
to=to,
list_subsets=[])
add_url = '/add/%s/%s/%s/%s/' % (get_metadata(
model, 'app_label'), model.__name__.lower(
), self.obj.pk, relation_name)
if permissions.has_add_permission(
self.request, model
) or permissions.has_relate_permission(
self.request, model):
related_paginator.add_action(
'Adicionar %s' % unicode(
get_metadata(
relation.model,
'verbose_name')), add_url,
'popup', 'fa-plus')
fieldset_dict['paginators'].append(
related_paginator)
else:
is_object_set = False
for related_object in list_related_objects(model):
if relation_name == related_object.get_accessor_name(
):
is_object_set = True
break
relation = getattr(self.obj, relation_name)
if hasattr(relation, 'all'):
qs = relation.all()
elif hasattr(relation, '__call__'):
qs = relation()
else:
qs = relation
to = is_object_set and related_object.field.name or None
fieldset_title = len(
relations) > 1 and get_metadata(
qs.model, 'verbose_name_plural') or title
if hasattr(relation, '_metadata'):
fieldset_title = relation._metadata[
'%s:verbose_name' % relation_name]
exclude = [
is_object_set and related_object.field.name
or ''
]
related_paginator = Paginator(
self.request,
qs,
fieldset_title,
exclude=exclude,
list_subsets=[],
to=to,
readonly=not is_object_set)
if is_object_set and (
permissions.has_add_permission(
self.request, qs.model)
or permissions.has_relate_permission(
self.request, qs.model)):
instance = qs.model()
setattr(instance, related_object.field.name,
self.obj)
if permissions.can_add(self.request, instance):
add_inline = get_metadata(
qs.model, 'add_inline')
if add_inline:
form_name = get_metadata(
qs.model, 'add_form')
if form_name:
fromlist = get_metadata(
qs.model, 'app_label')
forms_module = __import__(
'%s.forms' % fromlist,
fromlist=fromlist)
Form = getattr(
forms_module, form_name)
else:
class Form(ModelForm):
class Meta:
model = qs.model
fields = get_metadata(
qs.model,
'form_fields',
'__all__')
exclude = get_metadata(
qs.model,
'exclude_fields', ())
submit_label = 'Adicionar'
title = u'Adicionar %s' % get_metadata(
qs.model,
'verbose_name')
form = Form(self.request,
instance=instance,
inline=True)
if related_object.field.name in form.fields:
del (form.fields[
related_object.field.name])
related_paginator.form = form
if form.is_valid():
try:
form.save()
self.message = u'Ação realizada com sucesso'
except ValidationError, e:
form.add_error(
None, unicode(e.message))
else:
add_url = '/add/%s/%s/%s/%s/' % (
get_metadata(model, 'app_label'),
model.__name__.lower(),
self.obj.pk,
relation_name.replace('_set', ''))
add_label = u'Adicionar %s' % get_metadata(
qs.model, 'verbose_name')
add_label = get_metadata(
qs.model, 'add_label', add_label)
related_paginator.add_action(
add_label, add_url, 'popup',
'fa-plus')
fieldset_dict['paginators'].append(
related_paginator)
if 'extra' in fieldset[1]:
fieldset_dict['extra'] = []
for info in fieldset[1]['extra']:
fieldset_dict['extra'].append(info)