How to save the fk of a model in django?

Question

How to save the fk of a model in django?

Navigation

#1 by (1 votes)

0

I have a model called a device, this creates the folio and the name and additional I want to store the id of the user that is logged. This is my class:

class RegistroDispositivo(CreateView):
    model = Device
    template_name = 'registroDispositivo.html'
    form_class = RegistroDevice
    success_url = reverse_lazy('login')
    def post(self, request, *args,**kwargs):
        usuario = request.user
        if usuario:
            user_id = User.objects.get(email=usuario)
            form = self.form_class(initial={'user_id':user_id})
            return HttpResponseRedirect(self.get_success_url())
        else:
            form = self.form_class()
        return super(RegistroDispositivo, self).post(form)

My models.py includes an abstractBaseUser for additional functionalities to the Django user

class UserManager(BaseUserManager):
    use_in_migrations = True

    def _create_user(self, email, password, **extra_fields):
        if not email:
            raise ValueError('The given email must be set')
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_user(self, email, password=None, **extra_fields):
        extra_fields.setdefault('is_superuser', False)
        return self._create_user(email, password, **extra_fields)

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_superuser', True)

        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')

        return self._create_user(email, password, **extra_fields)

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(_('email address'), unique=True)
    first_name = models.CharField(_('first name'), max_length=30, blank=True)
    last_name = models.CharField(_('last name'), max_length=30, blank=True)
    date_joined = models.DateTimeField(_('date joined'), auto_now_add=True)
    is_active = models.BooleanField(_('active'), default=True)
    is_staff = models.BooleanField(default=False)
    nombreEmpresa = models.CharField(max_length=100, blank=True)

    USERNAME_FIELD = 'email'

    objects = UserManager()
    def get_full_name(self):
        return self.email

    def get_short_name(self):
        return self.email

class Device(models.Model):
    folio = models.CharField(max_length=15, unique=True)
    deviceName = models.CharField(max_length=50, unique=True)
    usuario = models.ForeignKey(User)

python django vistas

asked by Diego D. García 03.02.2018 в 00:46

source

1 answer

How to compare different arrays of the $ _session Symfony, do not show a field in a Type form

score 1 · Accepted Answer

There are several ways to do this. The simplest would be to overwrite form_valid in your class. It would be something like this:

class RegistroDispositivo(CreateView):
    model = Device
    template_name = 'registroDispositivo.html'
    form_class = RegistroDevice
    success_url = reverse_lazy('login')

    def form_valid(self,form):
        self.object = form.save(commit=False)
        self.object.usuario = self.request.user
        self.object.save()
        return super(RegistroDispositivo, self).form_valid(form)

Keep in mind that you have to modify the user field of your Device model to include blank = True, null = True, or the form will not be valid.