83 lines
4.3 KiB
Python
83 lines
4.3 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 محاسبه می‌کنه.
|
||
(قبلاً چون Plan فیلد durationای نداشت، این مقدار با یک قاعده‌ی stub
|
||
بر اساس متن title حدس زده می‌شد؛ الان duration_days مستقیماً از مدل
|
||
Plan خونده می‌شه.)
|
||
|
||
⚛️ atomic: خودِ این تابع هم مستقل (بدون اتکا به atomic بودنِ caller)
|
||
اتمیکه، چون به‌عنوان یک سرویسِ مشترک از جاهای مختلف (payment webhook،
|
||
bazaar verify) صدا زده می‌شه؛ save روی Subscription و نوشتنِ
|
||
StatusChangeLog که از signal.send نتیجه می‌شه، باید یک واحد بمونن.
|
||
"""
|
||
with transaction.atomic():
|
||
try:
|
||
# 🔒 select_for_update اضافه شد تا این تابع هم مثل
|
||
# grant_free_subscription، صرف‌نظر از caller، در برابر
|
||
# concurrent update ایمن باشه.
|
||
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 با همون تعداد روز.
|
||
|
||
از discount/services.py (apply_offer) صدا زده می‌شه؛ مثل بقیه‌ی
|
||
ارتباطات بین‌اپیِ این پروژه، یک import مستقیم به تابع سرویس، نه FK
|
||
واقعی و نه HTTP call به خودمون.
|
||
|
||
⚛️ atomic: مثل activate_subscription، مستقل از caller اتمیکه — چون
|
||
get_or_create + save + signal.send (که StatusChangeLog می‌سازه) باید
|
||
یک واحد باشن، چه apply_offer از قبل توی atomic باشه چه نه.
|
||
|
||
⚛️🔒 اصلاح race condition: قبلاً بعد از get_or_create هیچ قفلی گرفته
|
||
نمی‌شد؛ دو فراخوانیِ هم‌زمان (مثلاً دو آفر که تقریباً هم‌زمان اعمال
|
||
می‌شن) می‌تونستن هر دو همون Subscription رو بخونن، هر دو expires_at
|
||
رو بر اساس همون old_status/duration محاسبه کنن و save کنن — یکی از
|
||
دو نتیجه گم می‌شد (lost update). حالا ردیفِ Subscription قبل از
|
||
خوندن/تغییرش با select_for_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
|