Sith/core/baker_recipes.py

from datetime import timedelta

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

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

active_subscription = Recipe(
    Subscription,
    subscription_start=localdate() - timedelta(days=30),
    subscription_end=localdate() + timedelta(days=30),
)
ended_subscription = Recipe(
    Subscription,
    subscription_start=localdate() - timedelta(days=60),
    subscription_end=localdate() - 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."""

__inactivity = localdate() - settings.SITH_ACCOUNT_INACTIVITY_DELTA
very_old_subscriber_user = old_subscriber_user.extend(
    subscriptions=related(
        ended_subscription.extend(
            subscription_start=__inactivity - relativedelta(months=6, days=1),
            subscription_end=__inactivity - relativedelta(days=1),
        )
    )
)
"""A user which subscription ended enough time ago to be considered as inactive."""

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),
    groups=lambda: [Group.objects.get(club_board=settings.SITH_MAIN_CLUB_ID)],
)
"""A user which is in the board of the AE."""