Víðir Valberg Guðmundsson
64ba5cc6f4
All checks were successful
continuous-integration/drone/push Build is passing
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
import contextlib
|
|
|
|
from django.db.models import Exists
|
|
from django.db.models import OuterRef
|
|
from django.utils import timezone
|
|
|
|
from membership.models import Member
|
|
from membership.models import Membership
|
|
from membership.models import SubscriptionPeriod
|
|
|
|
|
|
def get_subscription_periods(member: Member | None = None) -> list[SubscriptionPeriod]:
|
|
subscription_periods = SubscriptionPeriod.objects.prefetch_related(
|
|
"membership_set",
|
|
"membership_set__user",
|
|
).all()
|
|
|
|
if member:
|
|
subscription_periods = subscription_periods.annotate(
|
|
membership_exists=Exists(
|
|
Membership.objects.filter(
|
|
user=member,
|
|
period=OuterRef("pk"),
|
|
),
|
|
),
|
|
)
|
|
|
|
return list(subscription_periods)
|
|
|
|
|
|
def get_current_subscription_period() -> SubscriptionPeriod | None:
|
|
with contextlib.suppress(SubscriptionPeriod.DoesNotExist):
|
|
return SubscriptionPeriod.objects.prefetch_related(
|
|
"membership_set",
|
|
"membership_set__user",
|
|
).get(period__contains=timezone.now())
|
|
|
|
|
|
def get_memberships(
|
|
*,
|
|
member: Member | None = None,
|
|
period: SubscriptionPeriod | None = None,
|
|
) -> Membership.QuerySet:
|
|
memberships = Membership.objects.select_related("membership_type").all()
|
|
|
|
if member:
|
|
memberships = memberships.for_member(member=member)
|
|
|
|
if period:
|
|
memberships = memberships.filter(period=period)
|
|
|
|
return memberships
|
|
|
|
|
|
def get_members():
|
|
return Member.objects.all().annotate_membership().order_by("username")
|
|
|
|
|
|
def get_member(*, member_id: int) -> Member:
|
|
return get_members().get(id=member_id)
|