def get(self, request):
try:
usuario = Usuario.objects.get(user=self.request.user)
empresa_instance = MinhaEmpresa.objects.get(m_usuario=usuario.id)
form = MinhaEmpresaForm(instance=empresa_instance)
except MinhaEmpresa.DoesNotExist:
form = MinhaEmpresaForm()
except Usuario.DoesNotExist:
usuario = Usuario.objects.get_or_create(user=self.request.user)[0]
form = MinhaEmpresaForm()
return render(request, self.template_name, {'form': form})
def get(self, request, *args, **kwargs):
super(EditarPerfilView, self).get(request, *args, **kwargs)
form_class = self.get_form_class()
form = self.get_form(form_class)
try:
empresa_instance = MinhaEmpresa.objects.get(
m_usuario=self.object.id)
minha_empresa_form = MinhaEmpresaForm(
instance=empresa_instance, prefix='m_empresa_form')
except MinhaEmpresa.DoesNotExist:
minha_empresa_form = MinhaEmpresaForm(prefix='m_empresa_form')
return self.render_to_response(self.get_context_data(form=form, minha_empresa_form=minha_empresa_form, object=self.object))
def post(self, request):
self.object = self.get_object()
try:
instance = Usuario.objects.get(user=request.user)
form = self.form_class(request.POST,
request.FILES,
instance=instance)
except Usuario.DoesNotExist:
form = self.form_class(request.POST, request.FILES, instance=None)
try:
empresa_instance = MinhaEmpresa.objects.get(
m_usuario=self.object.id)
minha_empresa_form = MinhaEmpresaForm(request.POST,
prefix='m_empresa_form',
instance=empresa_instance)
except MinhaEmpresa.DoesNotExist:
minha_empresa_form = MinhaEmpresaForm(request.POST,
prefix='m_empresa_form',
instance=None)
user = User.objects.get(pk=request.user.id)
if form.is_valid() and minha_empresa_form.is_valid():
try:
perfil = form.save(commit=False)
user.first_name = request.POST.get("first_name")
user.last_name = request.POST.get("last_name")
user.username = request.POST.get("username")
user.email = request.POST.get("email")
user.full_clean()
user.save()
perfil.user = user
if 'user_foto' in request.FILES:
perfil.user_foto = request.FILES['user_foto']
perfil.save()
minha_empresa = minha_empresa_form.save(commit=False)
minha_empresa.m_usuario = perfil
minha_empresa.save()
return self.form_valid(form, minha_empresa_form)
except (DatabaseError, ValidationError) as e:
form.add_error(field=None, error=e)
return render(request, self.template_name, {
'form': form,
'minha_empresa_form': minha_empresa_form
})
def post(self, request):
try:
usuario = Usuario.objects.get(user=self.request.user)
empresa_instance = MinhaEmpresa.objects.get(m_usuario=usuario.id)
form = MinhaEmpresaForm(request.POST, instance=empresa_instance)
except MinhaEmpresa.DoesNotExist:
form = MinhaEmpresaForm(request.POST, instance=None)
if form.is_valid():
usuario = Usuario.objects.get(user=request.user)
minha_empresa = form.save(commit=False)
minha_empresa.m_usuario = usuario
minha_empresa.save()
return self.form_valid(form)
return render(request, self.template_name, {'form': form})