| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- from __future__ import unicode_literals
- import getpass
- from optparse import make_option
- from django.contrib.auth import get_user_model
- from django.core.management.base import BaseCommand, CommandError
- from django.db import DEFAULT_DB_ALIAS
- from django.utils.encoding import force_str
- class Command(BaseCommand):
- option_list = BaseCommand.option_list + (
- make_option('--database', action='store', dest='database',
- default=DEFAULT_DB_ALIAS, help='Specifies the database to use. Default is "default".'),
- )
- help = "Change a user's password for django.contrib.auth."
- requires_system_checks = False
- def _get_pass(self, prompt="Password: "):
- p = getpass.getpass(prompt=force_str(prompt))
- if not p:
- raise CommandError("aborted")
- return p
- def handle(self, *args, **options):
- if len(args) > 1:
- raise CommandError("need exactly one or zero arguments for username")
- if args:
- username, = args
- else:
- username = getpass.getuser()
- UserModel = get_user_model()
- try:
- u = UserModel._default_manager.using(options.get('database')).get(**{
- UserModel.USERNAME_FIELD: username
- })
- except UserModel.DoesNotExist:
- raise CommandError("user '%s' does not exist" % username)
- self.stdout.write("Changing password for user '%s'\n" % u)
- MAX_TRIES = 3
- count = 0
- p1, p2 = 1, 2 # To make them initially mismatch.
- while p1 != p2 and count < MAX_TRIES:
- p1 = self._get_pass()
- p2 = self._get_pass("Password (again): ")
- if p1 != p2:
- self.stdout.write("Passwords do not match. Please try again.\n")
- count = count + 1
- if count == MAX_TRIES:
- raise CommandError("Aborting password change for user '%s' after %s attempts" % (u, count))
- u.set_password(p1)
- u.save()
- return "Password changed successfully for user '%s'" % u
|
