Exemple #1
0
def register_account(uow: UnitOfWork, email: str, password: str,
                     username: str) -> Account:
    whitelisted_emails = getenv('WHITELISTED_EMAILS')

    if whitelisted_emails is not None and email not in whitelisted_emails.split(
            ','):
        raise AccountRegistrationError('Email not whitelisted')

    with uow:
        if uow.account.get(Account, email=email) is not None:
            raise AccountRegistrationError(
                f"Account with the email '{email}' already exists")

        if uow.user.get(User, username=username):
            raise AccountRegistrationError(
                f"Account with the username '{username}' already exists")

        password_hash = Authentication().generate_password_hash(password)
        user = User(username)
        account = Account(email, password_hash, user)

        account_record = uow.account.add(account)
        uow.commit()

        account_record = uow.account.get(Account, email=email)
        verification_token = account.generate_verification_token
        account.send_registration_email(verification_token)

        return account_record
def create_article(uow: UnitOfWork, user_id: int, title: str,
                   body: str) -> Article:
    with uow:
        user_record = uow.user.get(User, id=user_id)

        if user_record is None:
            raise UserError('User does not exist')

        new_article = Article(title, body)
        new_article.convert_body_to_html

        user_record.articles.append(new_article)
        uow.commit()

        article_record = uow.articles.get(Article)

        return article_record
def create_project(uow: UnitOfWork, user_id: int, title: str,
                   body: str) -> Project:
    with uow:
        user_record = uow.user.get(User, id=user_id)

        if user_record is None:
            raise UserError('User does not exist')

        new_project = Project(title, body)
        new_project.convert_body_to_html

        user_record.projects.append(new_project)
        uow.commit()

        project_record = uow.projects.get(Project)

        return project_record
Exemple #4
0
def verify_account(uow: UnitOfWork, token: str) -> Account:
    registration_details = Account.validate_verification_token(token)
    email = registration_details['email']

    with uow:
        account_record = uow.account.get(Account, email=email)

        if account_record is None:
            raise AccountError(
                f"No registered account with the email '{email}'")

        if account_record.verified:
            raise AccountVerificationError('Account already verified')

        setattr(account_record, 'verified', True)
        uow.commit()

        account_record = uow.account.get(Account, email=email)

        return account_record