def by_email(email): """Retrieve User by email.""" click.secho('Getting user by email: %s' % email, bold=True) try: user = User.get_by_email(email) click.secho('* Found: %s' % user, fg='green') except NotFoundError: click.secho('* Does not exist!', fg='red')
def jwt(email, renewable): """Generate JWT for User.""" try: user = User.get_by_email(email) except NotFoundError: click.secho('* Does not exist!', fg='red') else: click.echo(create_jwt(user, renewable=renewable))
def create_user(email, customer_names): """Create an User.""" click.secho('Creating user: %s' % email, bold=True) for customer_name in customer_names: try: Customer.get_by_name(customer_name) except NotFoundError: click.secho('* customer does not exist: %s' % customer_name, fg='red') raise click.Abort() try: user = User.get_by_email(email) click.secho('* user exists already: %s' % user, fg='red') except NotFoundError: user = User(email=email, customers=customer_names) user.save() click.secho('* user created: %s' % user, fg='green')
def test_cannot_resolve_ambiguous_email(self, user_index, user): User(email=user.email).save(index=user_index._name, refresh=True) with pytest.raises(ConflictError): User.get_by_email(user.email)
def test_cannot_resolve_nonexistent_email(self, user_index): with pytest.raises(NotFoundError): User.get_by_email('*****@*****.**')
def test_can_resolve_unique_emails(self, user_index, user, another_user): match = User.get_by_email(user.email) assert match == user