mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-22 14:13:21 +00:00
33 lines
816 B
Python
33 lines
816 B
Python
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from django.utils.timezone import now
|
||
|
|
from model_bakery import seq
|
||
|
|
from model_bakery.recipe import Recipe, related
|
||
|
|
|
||
|
|
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),
|
||
|
|
)
|
||
|
|
old_subscriber_user = Recipe(
|
||
|
|
User,
|
||
|
|
first_name="old subscriber",
|
||
|
|
last_name=seq("user "),
|
||
|
|
subscriptions=related(ended_subscription),
|
||
|
|
)
|