Klubis#1131 - Zaračunavanje računov

Author: Brontes

django_project_base/base/event.py

@@ -2,12 +2,12 @@
 import datetime
 
 from abc import ABC, abstractmethod
-from gettext import gettext
 
 import swapper
 
 from django.conf import settings
 from django.shortcuts import get_object_or_404
+from django.utils.translation import gettext_lazy as _
 from rest_registration.settings import registration_settings
 
 from django_project_base.constants import EMAIL_SENDER_ID_SETTING_NAME, SMS_SENDER_ID_SETTING_NAME
@@ -186,8 +186,9 @@ def trigger(self, payload=None, **kwargs):
         if to := getattr(settings, "ADMINS", getattr(settings, "MANAGERS", [])):
             SystemEMailNotificationWithListOfEmails(
                 message=DjangoProjectBaseMessage(
-                    subject=gettext("Project settings action required"),
-                    body=f"{gettext('Action required for setting')} {payload.name} in project {payload.project.name}",
+                    subject=_("Project settings action required"),
+                    body=_("Action required for setting %(sett_name)s in project %(project_name).")
+                    % {"sett_name": payload.name, "project_name": payload.project.name},
                     footer="",
                     content_type=DjangoProjectBaseMessage.HTML,
                 ),

django_project_base/licensing/license_type_data.py

@@ -0,0 +1,13 @@
+from enum import Enum
+
+
+class LicenseType(Enum):
+    EMAIL = "email"
+    SMS = "sms"
+
+
+class LicenseTypeData:
+    price: float
+
+    def __init__(self, price: float) -> None:
+        self.price = price

django_project_base/licensing/logic.py

@@ -1,5 +1,7 @@
-from typing import List, Optional
+from enum import Enum
+from typing import List, Optional, Union
 
+from django.conf import settings
 from django.contrib.contenttypes.models import ContentType
 from django.db import connections
 from django.db.models import Model, Sum
@@ -8,7 +10,16 @@
 from dynamicforms.serializers import Serializer
 from rest_framework.exceptions import PermissionDenied
 
+from django_project_base.licensing.license_type_data import LicenseType
 from django_project_base.licensing.models import LicenseAccessUse
+from django_project_base.notifications.base.channels.channel import Channel
+
+
+def license_price(license_type: Union[LicenseType, Enum]):
+    license_type_data = settings.DJANGO_PROJECT_BASE_LICENSE_TYPE_DATA.get(license_type)
+    if not license_type_data:
+        raise f"No data for license type {license_type.value}"
+    return license_type_data.price
 
 
 class LicenseUsageReport(Serializer):
@@ -87,7 +98,7 @@ def log(
         content_type = ContentType.objects.get_for_model(model=record._meta.model)
         used = (
             LicenseAccessUse.objects.using(self.db)
-            .filter(user_id=str(user_profile_pk), content_type=content_type)
+            .filter(user_id=str(user_profile_pk))
             .aggregate(Sum("amount"))
             .get("amount__sum", None)
             or 0
@@ -98,7 +109,11 @@ def log(
 
             chl_prices = {i.name: i.notification_price for i in ChannelIdentifier.supported_channels()}
             for used_channel in list(set(list(items.keys()))):
-                used += chl_prices.get(used_channel, 0) * items.get(used_channel, 0)
+                if isinstance(used_channel, Channel):
+                    used_channel_name = used_channel.name
+                else:
+                    used_channel_name = used_channel
+                used += chl_prices.get(used_channel_name, 0) * items.get(used_channel, 0)
 
         if not kwargs.get("is_system_notification") and used >= 0:
             raise PermissionDenied(gettext("Your license is consumed. Please contact support."))