mirror of
https://github.com/ae-utbm/sith.git
synced 2024-12-23 00:01:16 +00:00
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
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."""
|