feat: switching to a defined enum

feedback from code review FIXES: APER-1322
eduNEXT · Jan 29, 2024 · 1c049a3 · 1c049a3
1 parent 5fedbc6
commit 1c049a3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lms/djangoapps/certificates/migrations/0037_fix_legacy_broken_invalid_certs.py b/lms/djangoapps/certificates/migrations/0037_fix_legacy_broken_invalid_certs.py
@@ -2,6 +2,8 @@
 
 from django.db import migrations
 
+from lms.djangoapps.certificates.data import CertificateStatuses
+
 
 class Migration(migrations.Migration):
     """
@@ -18,7 +20,9 @@ def make_invalid_certificates_unavailable(apps, schema_editor):
 
         GeneratedCertificate.objects.filter(
             certificateinvalidation__active=True
-        ).exclude(status="unavailable").update(status="unavailable")
+        ).exclude(status=CertificateStatuses.unavailable).update(
+            status=CertificateStatuses.unavailable
+        )
 
     operations = [
         migrations.RunPython(