deleteprofilefield.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. from __future__ import unicode_literals
  2. from django.contrib.auth import get_user_model
  3. from django.core.management.base import CommandError, BaseCommand
  4. UserModel = get_user_model()
  5. class Command(BaseCommand):
  6. help = (
  7. "Deletes specified profile field from database."
  8. )
  9. def add_arguments(self, parser):
  10. parser.add_argument(
  11. 'fieldname',
  12. help="field to delete",
  13. nargs='?',
  14. )
  15. def handle(self, *args, **options):
  16. fieldname = options['fieldname']
  17. if not fieldname:
  18. self.stderr.write("Specify fieldname to delete.")
  19. return
  20. deleted = 0
  21. queryset = UserModel.objects.filter(
  22. profile_fields__has_keys=[fieldname],
  23. )
  24. for user in queryset.iterator():
  25. if fieldname in user.profile_fields.keys():
  26. user.profile_fields.pop(fieldname)
  27. user.save(update_fields=['profile_fields'])
  28. deleted += 1
  29. self.stdout.write(
  30. '"{}" profile field has been deleted from {} users.'.format(
  31. fieldname, deleted
  32. )
  33. )