Cadastro do usuario com endereço separado

models.py

class User(models.Model):    
    email = models.EmailField()
    endereco = models.ForeignKey(Endereco)

class State(models.Model):
    name = models.CharField()
        
class City(models.Model):
    name = models.CharField()

class District(models.Model):
    name = models.CharField()
    
class UserAddress(models.Model):
    user = models.ForeignKey(User, related_name='addresses', on_delete=models.CASCADE)
    number = models.CharField()
    zip_code = models.CharField()
    street = models.CharField()
    complement = models.CharField()
    district = models.ForeignKey(District, on_delete=models.DO_NOTHING)
    city = models.ForeignKey(City, on_delete=models.DO_NOTHING)
    state = = models.ForeignKey(State, on_delete=models.DO_NOTHING)

urls.py

...

urlpatterns = [
    path('address/<int:user_id>/$', views.CustomerAddressCreate.as_view(), name='add_address'),
] 

views.py

...


class CustomerAddressCreateView(LoginRequiredMixin, CreateView):
    template_name = "caminho_do_template/template.html"
    model = UserAddress
    fields = [
      'state', 'city', 'district', 'address', 'number',
      'complement', 'zip_code'
    ]

    def form_valid(self, form):
        user = User.objects.get(pk=self.kwargs['user_id'])
        instance = form.save(commit=False)
        instance.user = user
        instance.save()
        return super().form_valid(form)

    def get_success_url(self):
        return reverse_lazy('customer_address:list', kwargs=self.kwargs)