32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
from django.utils.timezone import now
|
||
from datetime import timedelta
|
||
from django.core.exceptions import ObjectDoesNotExist
|
||
from sitemanagement.models import RoutePromotionLog, Pricing
|
||
from api.models import UserProfile
|
||
|
||
def check_monthly_limit(user, route, action_type):
|
||
try:
|
||
month_ago = now() - timedelta(days=30)
|
||
|
||
# получаем профиль пользователя и его тарифный план
|
||
user_profile = UserProfile.objects.get(user=user)
|
||
pricing_plan = Pricing.objects.get(plan=user_profile.account_type)
|
||
|
||
# определяем лимит в зависимости от типа действия и тарифного плана
|
||
if action_type == 'highlight':
|
||
action_limit = pricing_plan.highlight_limit
|
||
else: # rising
|
||
action_limit = pricing_plan.rising_limit
|
||
|
||
# проверяем количество действий за последний месяц
|
||
actions_count = RoutePromotionLog.objects.filter(
|
||
user=user,
|
||
route=route,
|
||
action_type=action_type,
|
||
created_at__gte=month_ago
|
||
).count()
|
||
|
||
return actions_count < action_limit
|
||
except ObjectDoesNotExist:
|
||
# если профиль или тарифный план не найден, запрещаем действие
|
||
return False |