60 lines
2.6 KiB
Python
60 lines
2.6 KiB
Python
from django.db import transaction
|
|
from django.utils import timezone
|
|
|
|
from statuslog.signals import status_changed
|
|
|
|
from .models import Subscription
|
|
|
|
|
|
def activate_subscription(subscription_id: int):
|
|
"""
|
|
بعد از تایید پرداخت (payment webhook) صدا زده می‌شود؛ اشتراک را active
|
|
می‌کند و expires_at را بر اساس Plan.duration_days محاسبه می‌کند.
|
|
مستقل از caller اتمیک است چون از چند جا (payment webhook، bazaar verify) صدا زده می‌شود.
|
|
"""
|
|
with transaction.atomic():
|
|
try:
|
|
subscription = Subscription.objects.select_for_update().get(pk=subscription_id)
|
|
except Subscription.DoesNotExist:
|
|
return None
|
|
|
|
old_status = subscription.status
|
|
duration_days = subscription.plan.duration_days if subscription.plan else 30
|
|
subscription.status = 'active'
|
|
subscription.expires_at = timezone.now() + timezone.timedelta(days=duration_days)
|
|
subscription.save(update_fields=['status', 'expires_at'])
|
|
|
|
status_changed.send(
|
|
sender='subscription.Subscription', object_id=subscription.id,
|
|
old_status=old_status, new_status='active',
|
|
event='payment_confirmed', user_id=subscription.user_id,
|
|
)
|
|
return subscription
|
|
|
|
|
|
def grant_free_subscription(user_id: int, *, duration_days: int | None = None, event: str = 'offer_applied'):
|
|
"""
|
|
فعال‌سازی اشتراک بدون هیچ Order/پرداختی — برای Offer (آفر رایگان محدود/کامل).
|
|
duration_days=None یعنی نامحدود (free_full)؛ عدد یعنی trial با همان تعداد روز.
|
|
|
|
مستقل از caller اتمیک است. ردیف Subscription قبل از خواندن/تغییر با
|
|
select_for_update قفل می‌شود تا فراخوانی‌های هم‌زمان lost update نسازند.
|
|
"""
|
|
with transaction.atomic():
|
|
Subscription.objects.get_or_create(user_id=user_id)
|
|
subscription = Subscription.objects.select_for_update().get(user_id=user_id)
|
|
old_status = subscription.status
|
|
|
|
subscription.status = 'active'
|
|
subscription.expires_at = (
|
|
timezone.now() + timezone.timedelta(days=duration_days) if duration_days else None
|
|
)
|
|
subscription.save(update_fields=['status', 'expires_at'])
|
|
|
|
status_changed.send(
|
|
sender='subscription.Subscription', object_id=subscription.id,
|
|
old_status=old_status, new_status='active',
|
|
event=event, user_id=user_id,
|
|
)
|
|
return subscription
|