From 7c8f15ca20a74a62c13251421e4a26d7917b44c8 Mon Sep 17 00:00:00 2001
From: Ross <xkjq@ymail.com>
Date: Mon, 18 Dec 2023 09:38:43 +0000
Subject: [PATCH] add site to trainees

---
 generic/migrations/0010_userprofile_site.py | 19 +++++++++++++++++++
 generic/models.py                           |  2 +-
 2 files changed, 20 insertions(+), 1 deletion(-)
 create mode 100644 generic/migrations/0010_userprofile_site.py

diff --git a/generic/migrations/0010_userprofile_site.py b/generic/migrations/0010_userprofile_site.py
new file mode 100644
index 00000000..fa9e6ba0
--- /dev/null
+++ b/generic/migrations/0010_userprofile_site.py
@@ -0,0 +1,19 @@
+# Generated by Django 4.1.4 on 2023-12-18 09:38
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('generic', '0009_modality_short_code_alter_modality_modality'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='userprofile',
+            name='site',
+            field=models.ForeignKey(blank=True, help_text='Primary site / rotation location', null=True, on_delete=django.db.models.deletion.SET_NULL, to='generic.site'),
+        ),
+    ]
diff --git a/generic/models.py b/generic/models.py
index 7aeb9ac5..cd8f4336 100644
--- a/generic/models.py
+++ b/generic/models.py
@@ -1306,7 +1306,7 @@ class UserProfile(models.Model):
         on_delete=models.CASCADE,
     )
     peninsula_trainee = models.BooleanField(default=False)
-    #affiliation = models.ForeignKey("Site", on_delete=models.SET_NULL, blank=True, null=True)
+    site = models.ForeignKey("Site", on_delete=models.SET_NULL, blank=True, null=True, help_text="Primary site / rotation location")
 
     def getusername(self):
         return self.user.username