klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2026-09-22 12:14:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

better subscription generation in populate_more

Browse Source
This commit is contained in:
imperosol
2026-09-22 11:04:18 +02:00
parent e937189b80
commit 130afc471c
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/management/commands/populate_more.py
+14 -2
View File
@@ -138,10 +138,22 @@ class Command(BaseCommand):
)
def create_subscriptions(self, users: list[User]):
subscription_types = [
"un-semestre",
"deux-semestres",
"cursus-tronc-commun",
"cursus-branche",
]
def prepare_subscription(_user: User, start_date: date) -> Subscription:
payment_method = random.choice(settings.SITH_SUBSCRIPTION_PAYMENT_METHOD)[0]
duration = random.randint(1, 4)
s = Subscription(member=_user, payment_method=payment_method)
subscription_type = random.choice(subscription_types)
s = Subscription(
member=_user,
payment_method=payment_method,
subscription_type=subscription_type,
)
duration = settings.SITH_SUBSCRIPTIONS[subscription_type]["duration"]
s.subscription_start = s.compute_start(d=start_date, duration=duration)
s.subscription_end = s.compute_end(duration)
return s