usercode with all personal data

3 files changed, 49 insertions, 0 deletions

diff --git a/client/migrations/0016_usercode_branca_usercode_email_usercode_first_name_and_more.py b/client/migrations/0016_usercode_branca_usercode_email_usercode_first_name_and_more.py
new file mode 100644
index 0000000..9e31cf3
--- /dev/null
+++ b/client/migrations/0016_usercode_branca_usercode_email_usercode_first_name_and_more.py
@@ -0,0 +1,35 @@
+# Generated by Django 4.1.5 on 2023-07-24 10:14
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('auth', '0012_alter_user_first_name_max_length'),
+        ('client', '0015_hidegroup'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='usercode',
+            name='branca',
+            field=models.ForeignKey(default=1, on_delete=django.db.models.deletion.PROTECT, to='auth.group'),
+        ),
+        migrations.AddField(
+            model_name='usercode',
+            name='email',
+            field=models.CharField(default='', max_length=250),
+        ),
+        migrations.AddField(
+            model_name='usercode',
+            name='first_name',
+            field=models.CharField(default='', max_length=250),
+        ),
+        migrations.AddField(
+            model_name='usercode',
+            name='last_name',
+            field=models.CharField(default='', max_length=250),
+        ),
+    ]
diff --git a/client/models.py b/client/models.py
index 20c8fa1..c9fe42f 100644
--- a/client/models.py
+++ b/client/models.py
@@ -108,6 +108,10 @@ class UserCode(models.Model):
     user = models.ForeignKey(User, default=None, on_delete=models.CASCADE)

     medic = models.ForeignKey(MedicalData, default=None, on_delete=models.PROTECT)

     code = models.IntegerField(default=0)

+    branca = models.ForeignKey(Group, default=0, on_delete=models.PROTECT)

+    first_name = models.CharField(default="", max_length=250)

+    last_name = models.CharField(default="", max_length=250)

+    email = models.CharField(default="", max_length=250)

     parent_name = models.CharField(default="", max_length=250)

     via = models.CharField(default="", max_length=250)

     cap = models.CharField(default="", max_length=250)

diff --git a/client/views.py b/client/views.py
index d4e9709..79783f1 100644
--- a/client/views.py
+++ b/client/views.py
@@ -15,6 +15,16 @@ from datetime import datetime
 import pytz
 from random import randint
 
+def migration_usercode(void):
+    usercodes = UserCode.objects.all()
+    for uc in usercodes:
+        user = uc.user
+        uc.first_name = user.first_name
+        uc.last_name = user.last_name
+        uc.email = user.email
+        uc.branca = user.groups.all()[0]
+        uc.save()
+
 @login_required()
 def index(request):
     context = {}