mirror of
https://github.com/lemeow125/DRF_Template.git
synced 2024-11-17 12:19:24 +08:00
24 lines
759 B
Python
24 lines
759 B
Python
|
from celery import shared_task
|
||
|
|
|
||
|
|
|
||
|
|
@shared_task
|
||
|
|
def get_paying_users():
|
||
|
|
from subscriptions.models import UserSubscription
|
||
|
|
# Get a list of user subscriptions
|
||
|
|
active_subscriptions = UserSubscription.objects.filter(
|
||
|
|
valid=True).distinct('user')
|
||
|
|
|
||
|
|
# Get paying users
|
||
|
|
active_users = []
|
||
|
|
|
||
|
|
# Paying regular users
|
||
|
|
active_users += [
|
||
|
|
subscription.user.id for subscription in active_subscriptions if subscription.user is not None and subscription.user.user_group is None]
|
||
|
|
|
||
|
|
# Paying users within groups
|
||
|
|
active_users += [
|
||
|
|
subscription.user_group.members for subscription in active_subscriptions if subscription.user_group is not None and subscription.user is None]
|
||
|
|
|
||
|
|
# Return paying users
|
||
|
|
return active_users
|