from flask import current_app
from buyercall.extensions import ses_client


class AddEmailIdentity:
    def add(self, partnership, username):
        domain = partnership.partner_url
        if current_app.config.get('TEST_SERVER_DOMAIN', None) and current_app.config.get('DEBUG', False):
            domain = current_app.config.get('TEST_SERVER_DOMAIN', None)
        return ses_client.verify_email_identity(f"{username}@{domain}")