small fix and improvements

Author: Muhammadali-Akbarov

payme/exceptions/webhook.py

@@ -116,10 +116,42 @@ class TransactionAlreadyExists(BasePaymeException):
     }
 
 
+class InvalidFiscalParams(BasePaymeException):
+    """
+    InvalidFiscalParams APIException.
+
+    Raised when the provided fiscal parameters are invalid.
+    """
+    status_code = 200
+    error_code = -32602
+    message = {
+        "uz": "Fiskal parameterlarida kamchiliklar bor",
+        "ru": "Неверные фискальные параметры.",
+        "en": "Invalid fiscal parameters."
+    }
+
+
+class InvalidAccount(BasePaymeException):
+    """
+    InvalidAccount APIException.
+
+    Raised when the provided account is invalid.
+    """
+    status_code = 200
+    error_code = -32400
+    message = {
+        "uz": "Hisob nomida kamchilik bor",
+        "ru": "Неверный номер счета.",
+        "en": "Invalid account."
+    }
+
+
 exception_whitelist = (
     IncorrectAmount,
     MethodNotFound,
     PermissionDenied,
     AccountDoesNotExist,
-    TransactionAlreadyExists
+    TransactionAlreadyExists,
+    InvalidFiscalParams,
+    InvalidAccount
 )

payme/migrations/0002_paymetransactions_fiscal_data.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.2 on 2025-03-14 08:01
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("payme", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="paymetransactions",
+            name="fiscal_data",
+            field=models.JSONField(blank=True, null=True),
+        ),
+    ]

payme/migrations/0003_alter_paymetransactions_fiscal_data.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.2 on 2025-03-14 08:20
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("payme", "0002_paymetransactions_fiscal_data"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="paymetransactions",
+            name="fiscal_data",
+            field=models.JSONField(default=dict),
+        ),
+    ]

payme/models.py

@@ -29,6 +29,7 @@ class PaymeTransactions(models.Model):
     account_id = models.BigIntegerField(null=False)
     amount = models.DecimalField(max_digits=10, decimal_places=2)
     state = models.IntegerField(choices=STATE, default=CREATED)
+    fiscal_data = models.JSONField(default=dict)
     cancel_reason = models.IntegerField(null=True, blank=True)
     created_at = models.DateTimeField(auto_now_add=True, db_index=True)
     updated_at = models.DateTimeField(auto_now=True, db_index=True)

payme/types/response/webhook.py

@@ -134,3 +134,11 @@ class GetStatement(CommonResponse):
     The check perform transactions response
     """
     transactions: List[str]
+
+
+@dataclass
+class SetFiscalData(CommonResponse):
+    """
+    The set fiscal data request
+    """
+    success: bool

payme/views.py

@@ -66,6 +66,7 @@ def post(self, request, *args, **kwargs):
             "CreateTransaction": self.create_transaction,
             "CheckTransaction": self.check_transaction,
             "CheckPerformTransaction": self.check_perform_transaction,
+            "SetFiscalData": self.set_fiscal_data,
         }
 
         try:
@@ -300,6 +301,33 @@ def get_statement(self, params) -> response.GetStatement:
 
         return result.as_resp()
 
+    @handle_exceptions
+    def set_fiscal_data(self, params):
+        """
+        Set fiscal data for the given transaction.
+        """
+        transaction = PaymeTransactions.get_by_transaction_id(transaction_id=params["id"])
+
+        fiscal_data = params.get("fiscal_data")
+        if not fiscal_data:
+            raise exceptions.InvalidFiscalParams(
+                "Missing fiscal_data field in parameters."
+            )
+
+        fiscal_type = params.get("type")
+
+        if fiscal_type not in ("PERFORM", "CANCEL"):
+            raise exceptions.InvalidFiscalParams(
+                f"Invalid fiscal type. Expected 'PERFORM' or 'CANCEL', got: {fiscal_type}"
+            )
+
+        fiscal_data["type"] = fiscal_type
+        transaction.fiscal_data = fiscal_data
+        transaction.save()
+
+        result = response.SetFiscalData(success=True)
+        return result.as_resp()
+
     def _cancel_response(self, transaction):
         """
         Helper method to generate cancel transaction response.

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name='payme-pkg',
-    version='3.0.21',
+    version='3.0.23',
     license='MIT',
     author="Muhammadali Akbarov",
     author_email='muhammadali17abc@gmail.com',