CAEA support

.gitignore

@@ -18,3 +18,7 @@ django_afip/version.py
 testapp/test_ticket.yaml
 test.csr
 test2.csr
+
+#Branch files
+Pipfile
+Pipfile.lock

django_afip/admin.py

@@ -498,3 +498,4 @@ def successful(self, obj):
 admin.site.register(models.VatType)
 admin.site.register(models.TaxType)
 admin.site.register(models.Observation)
+admin.site.register(models.Caea)

django_afip/exceptions.py

@@ -63,3 +63,7 @@ class CannotValidateTogether(DjangoAfipException):
 
 class ValidationError(DjangoAfipException):
     """Raised when a single Receipt failed to validate with AFIP's WS."""
+
+
+class CaeaCountError(DjangoAfipException):
+    """Raised when query the caea to obtain the number but 0 or 2 or more."""

django_afip/factories.py

@@ -1,7 +1,9 @@
+import calendar
 from datetime import date
 from datetime import datetime
 from pathlib import Path
 
+import pytest
 from django.contrib.auth.models import User
 from django.utils.timezone import make_aware
 from factory import LazyFunction
@@ -22,6 +24,42 @@ def get_test_file(filename: str, mode="r") -> Path:
     return path
 
 
+def get_order_of_date() -> int:
+    """
+    Helper method to detect if  the day of the month
+    corresponds to the first quarter (1) or the second (2)
+    """
+    today = datetime.now()
+    if today.day > 15:
+        return 2
+    return 1
+
+
+def valid_since_caea():
+    """
+    Helper method to assign the valid_since field from Caea model
+    to the correspondent year,month,day in the quarter
+    """
+    order = get_order_of_date()
+    valid_since = datetime(datetime.now().year, datetime.now().month, 1)
+    if order == 2:
+        valid_since = datetime(datetime.now().year, datetime.now().month, 16)
+    return valid_since
+
+
+def expires_caea():
+    """
+    Helper method to assign the expires field from Caea model
+    to the correspondent year,month,day in the quarter
+    """
+    order = get_order_of_date()
+    expires = datetime(datetime.now().year, datetime.now().month, 15)
+    if order == 2:
+        final = calendar.monthrange(datetime.now().year, datetime.now().month)[1]
+        expires = datetime(datetime.now().year, datetime.now().month, final)
+    return expires
+
+
 class UserFactory(DjangoModelFactory):
     class Meta:
         model = User
@@ -120,6 +158,12 @@ class Meta:
     sales_terms = "Credit Card"
 
 
+class PointOfSalesFactoryCaea(PointOfSalesFactory):
+
+    number = 4
+    issuance_type = "CAEA"
+
+
 class ReceiptFactory(DjangoModelFactory):
     class Meta:
         model = models.Receipt
@@ -149,6 +193,15 @@ def post(obj: models.Receipt, create, extracted, **kwargs):
         TaxFactory(tax_type__code=3, receipt=obj)
 
 
+class ReceiptWithVatAndTaxFactoryCaea(ReceiptFactory):
+    """Receipt with a valid Vat and Tax, ready to validate."""
+
+    @post_generation
+    def post(obj: models.Receipt, create, extracted, **kwargs):
+        VatFactory(vat_type__code=5, receipt=obj)
+        TaxFactory(tax_type__code=3, receipt=obj)
+
+
 class ReceiptWithInconsistentVatAndTaxFactory(ReceiptWithVatAndTaxFactory):
     """Receipt with a valid Vat and Tax, ready to validate."""
 
@@ -243,6 +296,20 @@ class Meta:
     tax_type = SubFactory(TaxTypeFactory)
 
 
+class CaeaFactory(DjangoModelFactory):
+    class Meta:
+        model = models.Caea
+
+    caea_code = "12345678974125"
+    period = datetime.today().strftime("%Y%m")
+    order = LazyFunction(get_order_of_date)
+    valid_since = LazyFunction(valid_since_caea)
+    expires = LazyFunction(expires_caea)
+    generated = make_aware(datetime(2022, 5, 30, 21, 6, 4))
+    report_deadline = make_aware(datetime(2022, 6, 20))
+    taxpayer = SubFactory(TaxPayerFactory)
+
+
 class ReceiptEntryFactory(DjangoModelFactory):
     class Meta:
         model = models.ReceiptEntry

django_afip/migrations/0011_caea.py

@@ -0,0 +1,78 @@
+# Generated by Django 4.1.1 on 2022-09-06 15:02
+
+import django.core.validators
+import django.db.models.deletion
+from django.db import migrations
+from django.db import models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0010_alter_authticket_service"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Caea",
+            fields=[
+                (
+                    "id",
+                    models.AutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                (
+                    "caea_code",
+                    models.BigIntegerField(
+                        help_text="CAEA code to operate offline AFIP",
+                        unique=True,
+                        validators=[
+                            django.core.validators.RegexValidator(regex="[0-9]{14}")
+                        ],
+                    ),
+                ),
+                (
+                    "period",
+                    models.IntegerField(
+                        help_text="Period to send in the CAEA request (yyyymm)"
+                    ),
+                ),
+                (
+                    "order",
+                    models.IntegerField(
+                        choices=[(1, "1"), (2, "2")],
+                        help_text="Month is divided in 1st quarter or 2nd quarter",
+                    ),
+                ),
+                ("valid_since", models.DateTimeField(verbose_name="valid_to")),
+                ("expires", models.DateTimeField(verbose_name="expires")),
+                ("generated", models.DateTimeField(verbose_name="generated")),
+                (
+                    "final_date_inform",
+                    models.DateTimeField(verbose_name="final_date_inform"),
+                ),
+                (
+                    "service",
+                    models.CharField(
+                        help_text="Service for which this ticket has been authorized.",
+                        max_length=34,
+                        verbose_name="service",
+                    ),
+                ),
+                ("active", models.BooleanField(default=False)),
+                (
+                    "taxpayer",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="caea_tickets",
+                        to="afip.taxpayer",
+                        verbose_name="taxpayer",
+                    ),
+                ),
+            ],
+        ),
+    ]

django_afip/migrations/0012_alter_caea_caea_code.py

@@ -0,0 +1,28 @@
+# Generated by Django 4.1.1 on 2022-09-06 15:15
+
+import django.core.validators
+from django.db import migrations
+from django.db import models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0011_caea"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="caea",
+            name="caea_code",
+            field=models.BigIntegerField(
+                help_text="CAEA code to operate offline AFIP",
+                unique=True,
+                validators=[
+                    django.core.validators.RegexValidator(regex="[0-9]{14}"),
+                    django.core.validators.MinLengthValidator(14),
+                    django.core.validators.MaxLengthValidator(14),
+                ],
+            ),
+        ),
+    ]

django_afip/migrations/0013_alter_caea_caea_code.py

@@ -0,0 +1,27 @@
+# Generated by Django 4.1.1 on 2022-09-06 15:24
+
+import django.core.validators
+from django.db import migrations
+from django.db import models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0012_alter_caea_caea_code"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="caea",
+            name="caea_code",
+            field=models.PositiveBigIntegerField(
+                help_text="CAEA code to operate offline AFIP",
+                unique=True,
+                validators=[
+                    django.core.validators.RegexValidator(regex="[0-9]{14}"),
+                    django.core.validators.MaxValueValidator(99999999999999),
+                ],
+            ),
+        ),
+    ]

django_afip/migrations/0014_remove_caea_service.py

@@ -0,0 +1,17 @@
+# Generated by Django 4.1.1 on 2022-09-06 21:34
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0013_alter_caea_caea_code"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="caea",
+            name="service",
+        ),
+    ]

django_afip/migrations/0015_alter_caea_expires_alter_caea_final_date_inform_and_more.py

@@ -0,0 +1,29 @@
+# Generated by Django 4.1.1 on 2022-09-06 21:43
+
+from django.db import migrations
+from django.db import models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0014_remove_caea_service"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="caea",
+            name="expires",
+            field=models.DateField(verbose_name="expires"),
+        ),
+        migrations.AlterField(
+            model_name="caea",
+            name="final_date_inform",
+            field=models.DateField(verbose_name="final_date_inform"),
+        ),
+        migrations.AlterField(
+            model_name="caea",
+            name="valid_since",
+            field=models.DateField(verbose_name="valid_to"),
+        ),
+    ]

django_afip/migrations/0016_receipt_generated.py

@@ -0,0 +1,25 @@
+# Generated by Django 4.1.1 on 2022-09-07 13:01
+
+import django.utils.timezone
+from django.db import migrations
+from django.db import models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0015_alter_caea_expires_alter_caea_final_date_inform_and_more"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="receipt",
+            name="generated",
+            field=models.DateTimeField(
+                auto_now_add=True,
+                default=django.utils.timezone.now,
+                verbose_name="Time when the receipt was created",
+            ),
+            preserve_default=False,
+        ),
+    ]

django_afip/migrations/0017_receipt_caea_receiptvalidation_caea.py

@@ -0,0 +1,35 @@
+# Generated by Django 4.1.1 on 2022-09-08 19:18
+
+import django.db.models.deletion
+from django.db import migrations
+from django.db import models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("afip", "0016_receipt_generated"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="receipt",
+            name="caea",
+            field=models.ForeignKey(
+                blank=True,
+                help_text="CAEA in case that the receipt must contain it",
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="afip.caea",
+            ),
+        ),
+        migrations.AddField(
+            model_name="receiptvalidation",
+            name="caea",
+            field=models.BooleanField(
+                default=False,
+                help_text="Indicate if the validation was from a CAEA receipt, in that case the field cae contains the CAEA number",
+                verbose_name="is_caea",
+            ),
+        ),
+    ]