Add migration to accounts

Rodrigo Siqueira de Melo
1 parent f59cc898
Showing 1 changed file with 50 additions and 0 deletions Show diff stats
colab/accounts/migrations/0004_auto_20150311_1818.py
@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+
+def trim_fields(apps, schema_editor):
+    trim_extra_account(apps, schema_editor, "facebook")
+    trim_extra_account(apps, schema_editor, "github")
+    trim_extra_account(apps, schema_editor, "twitter")
+
+def trim_extra_account(apps, schema_editor, types):
+    Users = apps.get_model('accounts', 'User')
+    for user in Users.objects.all():
+        field = getattr(user, types)
+        if not field:
+            continue
+        tmpString = field.split("/")[-1]
+        if len(field) >= 15:
+            setattr(user, types, tmpString[:14])
+            user.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('accounts', '0003_auto_20141218_1755'),
+    ]
+
+    operations = [
+        migrations.RunPython(trim_fields),
+        migrations.AlterField(
+            model_name='user',
+            name='facebook',
+            field=models.CharField(max_length=15, null=True, blank=True),
+            preserve_default=True,
+        ),
+
+        migrations.AlterField(
+            model_name='user',
+            name='github',
+            field=models.CharField(max_length=39, null=True, verbose_name='github', blank=True),
+            preserve_default=True,
+        ),
+        migrations.AlterField(
+            model_name='user',
+            name='twitter',
+            field=models.CharField(max_length=15, null=True, blank=True),
+            preserve_default=True,
+        ),
+    ]
@@ -0,0 +1,50 @@		@@ -0,0 +1,50 @@
	1	+# -- coding: utf-8 --
	2	+from __future__ import unicode_literals
	3	+
	4	+from django.db import models, migrations
	5	+
	6	+def trim_fields(apps, schema_editor):
	7	+ trim_extra_account(apps, schema_editor, "facebook")
	8	+ trim_extra_account(apps, schema_editor, "github")
	9	+ trim_extra_account(apps, schema_editor, "twitter")
	10	+
	11	+def trim_extra_account(apps, schema_editor, types):
	12	+ Users = apps.get_model('accounts', 'User')
	13	+ for user in Users.objects.all():
	14	+ field = getattr(user, types)
	15	+ if not field:
	16	+ continue
	17	+ tmpString = field.split("/")[-1]
	18	+ if len(field) >= 15:
	19	+ setattr(user, types, tmpString[:14])
	20	+ user.save()
	21	+
	22	+
	23	+class Migration(migrations.Migration):
	24	+
	25	+ dependencies = [
	26	+ ('accounts', '0003_auto_20141218_1755'),
	27	+ ]
	28	+
	29	+ operations = [
	30	+ migrations.RunPython(trim_fields),
	31	+ migrations.AlterField(
	32	+ model_name='user',
	33	+ name='facebook',
	34	+ field=models.CharField(max_length=15, null=True, blank=True),
	35	+ preserve_default=True,
	36	+ ),
	37	+
	38	+ migrations.AlterField(
	39	+ model_name='user',
	40	+ name='github',
	41	+ field=models.CharField(max_length=39, null=True, verbose_name='github', blank=True),
	42	+ preserve_default=True,
	43	+ ),
	44	+ migrations.AlterField(
	45	+ model_name='user',
	46	+ name='twitter',
	47	+ field=models.CharField(max_length=15, null=True, blank=True),
	48	+ preserve_default=True,
	49	+ ),
	50	+ ]