/main/migrations/0007_replace_special_chars_and_whitespace_in_usernames.py

https://github.com/SEL-Columbia/formhub · Python · 103 lines · 95 code · 6 blank · 2 comment · 2 complexity · a659068cf0d162a9ed92c8d69ccc3a47 MD5 · raw file 

    

  1. # encoding: utf-8
    2. 

  2. import datetime
    3. 

  3. from south.db import db
    4. 

  4. from south.v2 import DataMigration
    5. 

  5. from django.db import models
    6. 

  6. import re
    7. 

  7. 

  8. class Migration(DataMigration):
    9. 

  9. 

  10.     def forwards(self, orm):
    11. 

  11.         def update_username_for_user(user, match_on, sub_with):
    12. 

  12.             regex = re.compile(match_on)
    13. 

  13.             if regex.search(user.username):
    14. 

  14.                 user.username = regex.sub(sub_with, user.username)
    15. 

  15.                 user.save()
    16. 

  16. 

  17.         # remove whitespace and replace special chars with underscores
    18. 

  18.         for user in orm['auth.User'].objects.all():
    19. 

  19.             update_username_for_user(user, "\s", "")
    20. 

  20.             update_username_for_user(user, "\W", "_")
    21. 

  21. 

  22.     def backwards(self, orm):
    23. 

  23.         pass
    24. 

  24. 

  25.     models = {
    26. 

  26.         'auth.group': {
    27. 

  27.             'Meta': {'object_name': 'Group'},
    28. 

  28.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    29. 

  29.             'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'}),
    30. 

  30.             'permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'})
    31. 

  31.         },
    32. 

  32.         'auth.permission': {
    33. 

  33.             'Meta': {'ordering': "('content_type__app_label', 'content_type__model', 'codename')", 'unique_together': "(('content_type', 'codename'),)", 'object_name': 'Permission'},
    34. 

  34.             'codename': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
    35. 

  35.             'content_type': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['contenttypes.ContentType']"}),
    36. 

  36.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    37. 

  37.             'name': ('django.db.models.fields.CharField', [], {'max_length': '50'})
    38. 

  38.         },
    39. 

  39.         'auth.user': {
    40. 

  40.             'Meta': {'object_name': 'User'},
    41. 

  41.             'date_joined': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
    42. 

  42.             'email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'blank': 'True'}),
    43. 

  43.             'first_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
    44. 

  44.             'groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Group']", 'symmetrical': 'False', 'blank': 'True'}),
    45. 

  45.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    46. 

  46.             'is_active': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
    47. 

  47.             'is_staff': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
    48. 

  48.             'is_superuser': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
    49. 

  49.             'last_login': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
    50. 

  50.             'last_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
    51. 

  51.             'password': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
    52. 

  52.             'user_permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'}),
    53. 

  53.             'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '30'})
    54. 

  54.         },
    55. 

  55.         'contenttypes.contenttype': {
    56. 

  56.             'Meta': {'ordering': "('name',)", 'unique_together': "(('app_label', 'model'),)", 'object_name': 'ContentType', 'db_table': "'django_content_type'"},
    57. 

  57.             'app_label': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
    58. 

  58.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    59. 

  59.             'model': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
    60. 

  60.             'name': ('django.db.models.fields.CharField', [], {'max_length': '100'})
    61. 

  61.         },
    62. 

  62.         'main.metadata': {
    63. 

  63.             'Meta': {'object_name': 'MetaData'},
    64. 

  64.             'data_file': ('django.db.models.fields.files.FileField', [], {'max_length': '100', 'null': 'True'}),
    65. 

  65.             'data_file_type': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True'}),
    66. 

  66.             'data_type': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
    67. 

  67.             'data_value': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
    68. 

  68.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    69. 

  69.             'xform': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['odk_logger.XForm']"})
    70. 

  70.         },
    71. 

  71.         'main.userprofile': {
    72. 

  72.             'Meta': {'object_name': 'UserProfile'},
    73. 

  73.             'city': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
    74. 

  74.             'country': ('django.db.models.fields.CharField', [], {'max_length': '2', 'blank': 'True'}),
    75. 

  75.             'description': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
    76. 

  76.             'home_page': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
    77. 

  77.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    78. 

  78.             'name': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
    79. 

  79.             'organization': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
    80. 

  80.             'require_auth': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
    81. 

  81.             'twitter': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
    82. 

  82.             'user': ('django.db.models.fields.related.OneToOneField', [], {'related_name': "'profile'", 'unique': 'True', 'to': "orm['auth.User']"})
    83. 

  83.         },
    84. 

  84.         'odk_logger.xform': {
    85. 

  85.             'Meta': {'ordering': "('id_string',)", 'unique_together': "(('user', 'id_string'),)", 'object_name': 'XForm'},
    86. 

  86.             'date_created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
    87. 

  87.             'date_modified': ('django.db.models.fields.DateTimeField', [], {'auto_now': 'True', 'blank': 'True'}),
    88. 

  88.             'description': ('django.db.models.fields.TextField', [], {'default': "u''", 'null': 'True'}),
    89. 

  89.             'downloadable': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
    90. 

  90.             'has_start_time': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
    91. 

  91.             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
    92. 

  92.             'id_string': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'}),
    93. 

  93.             'json': ('django.db.models.fields.TextField', [], {'default': "u''"}),
    94. 

  94.             'shared': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
    95. 

  95.             'shared_data': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
    96. 

  96.             'title': ('django.db.models.fields.CharField', [], {'max_length': '64'}),
    97. 

  97.             'user': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'xforms'", 'null': 'True', 'to': "orm['auth.User']"}),
    98. 

  98.             'xls': ('django.db.models.fields.files.FileField', [], {'max_length': '100', 'null': 'True'}),
    99. 

  99.             'xml': ('django.db.models.fields.TextField', [], {})
    100. 

  100.         }
    101. 

  101.     }
    102. 

  102. 

  103.     complete_apps = ['main']
    104.