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