from sqlalchemy.sql.functions import current_user
from wtforms.validators import ValidationError
from flask_babel import lazy_gettext as _
from flask_login import current_user as u
from buyercall.blueprints.user.models import User


def ensure_identity_exists(form, field):
    """
    Ensure an identity exists.

    :param form: wtforms Instance
    :param field: Field being passed in.
    :return: None
    """
    user = User.find_by_identity(field.data.lower())

    if not user:
        raise ValidationError(_('Unable to locate account.'))


def ensure_existing_password_matches(form, field):
    """
    Ensure that the current password matches their existing password.

    :param form: wtforms Instance
    :param field: Field being passed in.
    :return: None
    """
    try:
        user = User.query.get(form._obj.id)
    except:
        user = User.query.get(u.id)

    if not user.authenticated(password=field.data):
        raise ValidationError(_('Does not match.'))