B CDb@s,ddlmZddlmZGdddeZdS)) BaseCommand)MyUserc@s eZdZdZddZddZdS)Commandz!restore password same to usernamecCs|jdddddS)Nz--userzinput the restore username.)defaulthelp) add_argument)selfparserr IG:\easytimepro\master/mysite/base\management\commands\restore_password.py add_arguments szCommand.add_argumentscOsl|d}|s"tdtddStjj|d}|r\|||td||fn td|dS)Nuserz(Error: missing "--user" parameterzAExample: python manage.py restore_password --user usernameF)usernamez/Success restore user "%s" password "%s".z&Error: username "%s" not found.)getprintrZobjectsfilterfirstZ set_passwordZsave)r argsZoptionsrrr r r handles  zCommand.handleN)__name__ __module__ __qualname__rr rr r r r r srN)Zdjango.core.management.baserZmysite.accounts.modelsrrr r r r s