From 88117b96a4b420d88549b8df2649c3eb9c61c2a5 Mon Sep 17 00:00:00 2001
From: Nora Shapiro <nora.shapiro@sentry.io>
Date: Wed, 2 Oct 2024 19:05:30 -0700
Subject: [PATCH] add upload_token_required_for_public_repos, backfill and make
 non-nullable (#373)

* add upload_token_required_for_public_repos, backfill and make non-nullable

* risky migration part 1

* change update query around

* remove second migration

* split into 3 steps

* one step
---
 ..._upload_token_required_for_public_repos.py | 29 +++++++++++++++++++
 shared/django_apps/codecov_auth/models.py     |  1 +
 2 files changed, 30 insertions(+)
 create mode 100644 shared/django_apps/codecov_auth/migrations/0060_owner_upload_token_required_for_public_repos.py

diff --git a/shared/django_apps/codecov_auth/migrations/0060_owner_upload_token_required_for_public_repos.py b/shared/django_apps/codecov_auth/migrations/0060_owner_upload_token_required_for_public_repos.py
new file mode 100644
index 000000000..a1e889d9b
--- /dev/null
+++ b/shared/django_apps/codecov_auth/migrations/0060_owner_upload_token_required_for_public_repos.py
@@ -0,0 +1,29 @@
+# Generated by Django 4.2.16 on 2024-10-02 00:21
+
+from django.db import migrations, models
+
+from shared.django_apps.migration_utils import RiskyAddField
+
+
+class Migration(migrations.Migration):
+    """
+    BEGIN;
+    --
+    -- Add field upload_token_required_for_public_repos to owner
+    --
+    ALTER TABLE "owners" ADD COLUMN "upload_token_required_for_public_repos" boolean DEFAULT true NOT NULL;
+    ALTER TABLE "owners" ALTER COLUMN "upload_token_required_for_public_repos" DROP DEFAULT;
+    COMMIT;
+    """
+
+    dependencies = [
+        ("codecov_auth", "0059_alter_accountsusers_options"),
+    ]
+
+    operations = [
+        RiskyAddField(
+            model_name="owner",
+            name="upload_token_required_for_public_repos",
+            field=models.BooleanField(default=True),
+        ),
+    ]
diff --git a/shared/django_apps/codecov_auth/models.py b/shared/django_apps/codecov_auth/models.py
index 1d5ba0a2c..02a4896dd 100644
--- a/shared/django_apps/codecov_auth/models.py
+++ b/shared/django_apps/codecov_auth/models.py
@@ -358,6 +358,7 @@ class Meta:
     onboarding_completed = models.BooleanField(default=False)
     is_superuser = models.BooleanField(null=True, default=False)
     max_upload_limit = models.IntegerField(null=True, default=150, blank=True)
+    upload_token_required_for_public_repos = models.BooleanField(default=True)
 
     sentry_user_id = models.TextField(null=True, blank=True, unique=True)
     sentry_user_data = models.JSONField(null=True)