27 lines
736 B
Python
27 lines
736 B
Python
from django.contrib.auth.models import User
|
|
|
|
from membership.models import Membership
|
|
from membership.models import SubscriptionPeriod
|
|
|
|
|
|
def get_subscription_periods() -> list[SubscriptionPeriod]:
|
|
subscription_periods = SubscriptionPeriod.objects.prefetch_related(
|
|
"membership_set",
|
|
"membership_set__user",
|
|
).all()
|
|
return list(subscription_periods)
|
|
|
|
|
|
def get_memberships(
|
|
*, user: User | None = None, period: SubscriptionPeriod | None = None
|
|
) -> Membership.QuerySet:
|
|
memberships = Membership.objects.select_related("membership_type").all()
|
|
|
|
if user:
|
|
memberships = memberships.for_user(user)
|
|
|
|
if period:
|
|
memberships = memberships.filter(period=period)
|
|
|
|
return memberships
|