add 1:n relation User to UserExtension

Author: jrief

testapp/forms/user.py

@@ -11,7 +11,7 @@
 class UserForm(ModelForm):
     class Meta:
         model = User
-        exclude = ['username', 'password', 'last_login', 'date_joined']
+        exclude = ['password', 'last_login', 'date_joined']
         widgets = {
             'password': PasswordInput,
             'groups': SelectizeMultiple,

testapp/migrations/0001_initial.py

@@ -164,6 +164,14 @@ class Migration(migrations.Migration):
                 ('phone_number', models.CharField(blank=True, max_length=25, null=True, verbose_name='Phone Number')),
             ],
         ),
+        migrations.CreateModel(
+            name='UserExtension',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('phone_number', models.CharField(blank=True, max_length=25, null=True, verbose_name='Phone Number')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='user_extensions', to='testapp.user')),
+            ],
+        ),
         migrations.CreateModel(
             name='Gallery',
             fields=[

testapp/models/user.py

@@ -53,3 +53,17 @@ class ExtendUser(models.Model):
         blank=True,
         null=True,
     )
+
+
+class UserExtension(models.Model):
+    user = models.ForeignKey(
+        User,
+        on_delete=models.CASCADE,
+        related_name='user_extensions',
+    )
+    phone_number = models.CharField(
+        verbose_name="Phone Number",
+        max_length=25,
+        blank=True,
+        null=True,
+    )

testapp/views.py

@@ -281,13 +281,8 @@ class DemoFormCollectionView(DemoFormCollectionViewMixin, FormCollectionView):
     pass
 
 
-class UserCollectionView(DemoFormCollectionViewMixin, EditCollectionView):
-    model = User
-    template_name = 'testapp/form-collection.html'
-
-    def get_object(self, queryset=None):
-        user, _ = self.model.objects.get_or_create(username='demo')
-        return user
+class DemoModelCollectionView(DemoFormCollectionViewMixin, SessionFormCollectionViewMixin, EditCollectionView):
+    pass
 
 
 class CompanyCollectionView(DemoFormCollectionViewMixin, SessionFormCollectionViewMixin, EditCollectionView):
@@ -726,9 +721,10 @@ class CompleteForm(FormMixin, forms.Form):
     ), name='poll'),
     path('company', CompanyCollectionView.as_view(), name='company'),
     path('companies', CompaniesCollectionView.as_view(), name='company'),
-    path('user', UserCollectionView.as_view(
-        collection_class=UserCollection
-    ), name='user'),
+    path('user-collection', DemoModelCollectionView.as_view(
+        collection_class=UserCollection,
+        model=User,
+    ), name='user-collection'),
     # path('userlist', UserCollectionView.as_view(
     #     collection_class=UserListCollection
     # ), name='userlist'),