Fix migrations.

mozmeao · Jun 12, 2017 · 4ccb7a1 · 4ccb7a1
1 parent 2505514
commit 4ccb7a1
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 7 deletions.
diff --git a/snippets/base/migrations/0014_auto_20160419_1316.py b/snippets/base/migrations/0014_auto_20160419_1316.py
@@ -6,11 +6,21 @@
 from django_mysql.models.functions import AsType, ColumnAdd
 
 
-def set_isdefaultbrowser_clientoption(apps, schema_editor):
+def migrate_aboutaccounts_snippets(apps, schema_editor):
     """Migrate existing snippets that used about:accounts link filtering."""
+    Snippet = apps.get_model('base', 'Snippet')
+    Snippet.objects.filter(data__icontains='href=\\"about:accounts').update(
+        client_options=ColumnAdd('client_options', {'has_fxaccount': AsType('yes', 'CHAR')})
+    )
+    Snippet.objects.exclude(data__icontains='href=\\"about:accounts').update(
+        client_options=ColumnAdd('client_options', {'has_fxaccount': AsType('any', 'CHAR')})
+    )
+
+
+def set_has_testpilot(apps, schema_editor):
     Snippet = apps.get_model('base', 'Snippet')
     Snippet.objects.update(
-        client_options=ColumnAdd('client_options', {'is_default_browser': AsType('any', 'CHAR')})
+        client_options=ColumnAdd('client_options', {'has_testpilot': AsType('any', 'CHAR')})
     )
 
 
@@ -26,5 +36,6 @@ class Migration(migrations.Migration):
     ]
 
     operations = [
-        migrations.RunPython(set_isdefaultbrowser_clientoption, noop),
+        migrations.RunPython(migrate_aboutaccounts_snippets, noop),
+        migrations.RunPython(set_has_testpilot, noop),
     ]
diff --git a/snippets/base/migrations/0015_auto_20160420_1317.py b/snippets/base/migrations/0015_auto_20160420_1317.py
@@ -6,8 +6,7 @@
 from django_mysql.models.functions import AsType, ColumnAdd
 
 
-def migrate_aboutaccounts_snippets(apps, schema_editor):
-    """Migrate existing snippets that used about:accounts link filtering."""
+def set_browser_versions(apps, schema_editor):
     Snippet = apps.get_model('base', 'Snippet')
     Snippet.objects.update(
         client_options=ColumnAdd(
@@ -31,5 +30,5 @@ class Migration(migrations.Migration):
     ]
 
     operations = [
-        migrations.RunPython(migrate_aboutaccounts_snippets, noop),
+        migrations.RunPython(set_browser_versions, noop),
     ]
diff --git a/snippets/base/migrations/0016_auto_20160419_1316.py b/snippets/base/migrations/0016_auto_20160419_1316.py
@@ -7,7 +7,6 @@
 
 
 def set_isdefaultbrowser_clientoption(apps, schema_editor):
-    """Migrate existing snippets that used about:accounts link filtering."""
     Snippet = apps.get_model('base', 'Snippet')
     Snippet.objects.update(
         client_options=ColumnAdd('client_options', {'is_default_browser': AsType('any', 'CHAR')})