3_downstream_scalars.py

from chalk import online
from chalk.client import ChalkClient
from chalk.features import features


@features
class User:
    id: int
    email: str
    email_domain: str
    banned_email: bool


@online
def get_email_domain(email: User.email) -> User.email_domain:
    return email.split("@")[1].lower()


@online
def is_banned_email(domain: User.email_domain) -> User.banned_email:
    return domain in {
        "pergi.id",
        "convoitucpa.com",
        "vshgl.com",
        "nieise.com",
        "bookue.site",
        "umaasa.com",
    }


if __name__ == "__main__":
    client = ChalkClient()
    assert not client.query(
        input={User.email: "katherine.johnson@nasa.gov"},
        # Requesting User.banned_email requires running
        # `get_email_domain` and then `is_banned_email`
        output=[User.banned_email],
    ).get_feature_value(User.banned_email)

    assert client.query(
        input={User.email: "attacker@vshgl.com"},
        output=[User.banned_email],
    ).get_feature_value(User.banned_email)