Sith/core/baker_recipes.py

from datetime import timedelta

from django.conf import settings
from django.utils.timezone import now
from model_bakery import seq
from model_bakery.recipe import Recipe, related

from club.models import Membership
from core.models import User
from subscription.models import Subscription

active_subscription = Recipe(
    Subscription,
    subscription_start=now() - timedelta(days=30),
    subscription_end=now() + timedelta(days=30),
)
ended_subscription = Recipe(
    Subscription,
    subscription_start=now() - timedelta(days=60),
    subscription_end=now() - timedelta(days=30),
)

subscriber_user = Recipe(
    User,
    first_name="subscriber",
    last_name=seq("user "),
    subscriptions=related(active_subscription),
)
"""A user with an active subscription."""

old_subscriber_user = Recipe(
    User,
    first_name="old subscriber",
    last_name=seq("user "),
    subscriptions=related(ended_subscription),
)
"""A user with an ended subscription."""

ae_board_membership = Recipe(
    Membership,
    start_date=now() - timedelta(days=30),
    club_id=settings.SITH_MAIN_CLUB_ID,
    role=settings.SITH_CLUB_ROLES_ID["Board member"],
)

board_user = Recipe(
    User,
    first_name="AE",
    last_name=seq("member "),
    memberships=related(ae_board_membership),
)
"""A user which is in the board of the AE."""