mirror of
https://github.com/ae-utbm/sith.git
synced 2024-10-18 12:58:06 +00:00
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
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."""
|