class AddUser(commands.Subcommand):

    """

    Adds a user to the database.

    """

    name = 'adduser'

    description = "Add a user to the database."

    def add_parser_args(self, parser):

        parser.add_argument('url', metavar='URL',

                            help="Database engine URL")

        parser.add_argument('username',

                            help="Username for the new account.")

        parser.add_argument('-A', '--administrator',

                            action='store_true',

                            help="Add the new user to the Administrator role.")

    def run(self, args):

        from sqlalchemy import create_engine

        from .db import Session

        from .db.model import User

        from getpass import getpass

        from .db.auth import set_user_password, administrator_role

        engine = create_engine(args.url)

        session = Session(bind=engine)

        if session.query(User).filter_by(username=args.username).count():

            session.close()

            print("User '{0}' already exists.".format(args.username))

            return

        passwd = ''

        while not passwd:

            try:

                passwd = getpass("Enter a password for user '{0}': ".format(args.username))

            except KeyboardInterrupt:

                print("\nOperation was canceled.")

                return

        user = User(username=args.username)

        set_user_password(user, passwd)

        if args.administrator:

            user.roles.append(administrator_role(session))

        session.add(user)

        session.commit()

        session.close()

        print("Created user: {0}".format(args.username))

adduser = rattail.commands:AddUser