Implement schedule task to collect billing invoice

deployment/docker/pretix.bash

@@ -46,6 +46,11 @@ if [ "$1" == "taskworker" ]; then
     exec celery -A pretix.celery_app worker -l info "$@"
 fi
 
+if [ "$1" == "taskbeat" ]; then
+    shift
+    exec celery -A pretix.celery_app beat -l info "$@"
+fi
+
 if [ "$1" == "upgrade" ]; then
     exec python3 -m pretix updatestyles
 fi

deployment/docker/supervisord/pretixbeat.conf

@@ -0,0 +1,10 @@
+[program:pretixtask]
+command=/usr/local/bin/pretix taskbeat
+autostart=true
+autorestart=true
+priority=5
+user=pretixuser
+stdout_logfile=/dev/fd/1
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/fd/2
+stderr_logfile_maxbytes=0

pyproject.toml

@@ -97,6 +97,7 @@ dependencies = [
         'django-sso==3.0.2',
         'PyJWT~=2.8.0',
         'exhibitors @ git+https://github.com/fossasia/eventyay-tickets-exhibitors.git@master',
+        'pyvat==1.3.18',
 ]
 
 [project.optional-dependencies]

src/pretix/api/urls.py

@@ -6,6 +6,7 @@
 
 from pretix.api.views import cart
 
+from ..eventyay_common.views.billing import BillingInvoicePreview
 from .views import (
     checkin, device, event, exporters, item, oauth, order, organizer, upload,
     user, version, voucher, waitinglist, webhooks,
@@ -98,4 +99,5 @@
     url(r"^upload$", upload.UploadView.as_view(), name="upload"),
     url(r"^me$", user.MeView.as_view(), name="user.me"),
     url(r"^version$", version.VersionView.as_view(), name="version"),
+    url(r"^billing-testing/(?P<task>[^/]+)", BillingInvoicePreview.as_view(), name="billing-testing"),
 ]

src/pretix/base/migrations/0003_alter_cachedcombinedticket_id_alter_cachedticket_id_and_more.py

@@ -0,0 +1,49 @@
+# Generated by Django 5.1.2 on 2024-10-31 09:30
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("pretixbase", "0003_event_is_video_creation_and_more"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="OrganizerBillingModel",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True, primary_key=True, serialize=False
+                    ),
+                ),
+                ("primary_contact_name", models.CharField(max_length=255)),
+                ("primary_contact_email", models.EmailField(max_length=255)),
+                ("company_or_organization_name", models.CharField(max_length=255)),
+                ("address_line_1", models.CharField(max_length=255)),
+                ("address_line_2", models.CharField(max_length=255)),
+                ("city", models.CharField(max_length=255)),
+                ("zip_code", models.CharField(max_length=255)),
+                ("country", models.CharField(max_length=255)),
+                ("preferred_language", models.CharField(max_length=255)),
+                ("tax_id", models.CharField(max_length=255)),
+                ("stripe_customer_id", models.CharField(max_length=255, null=True)),
+                (
+                    "stripe_payment_method_id",
+                    models.CharField(max_length=255, null=True),
+                ),
+                ("stripe_setup_intent_id", models.CharField(max_length=255, null=True)),
+                (
+                    "organizer",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="billing",
+                        to="pretixbase.organizer",
+                    ),
+                ),
+            ],
+        ),
+    ]

src/pretix/base/migrations/0003_create_billing_invoice_table.py

@@ -0,0 +1,49 @@
+# Generated by Django 5.1.2 on 2024-10-29 04:33
+
+import datetime
+
+import django.contrib.postgres.fields
+import django.db.models.deletion
+from django.db import migrations, models
+
+import pretix.base.models.base
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('pretixbase', '0003_alter_cachedcombinedticket_id_alter_cachedticket_id_and_more'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='BillingInvoice',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False)),
+                ('status', models.CharField(default='n', max_length=1)),
+                ('amount', models.DecimalField(decimal_places=2, max_digits=10)),
+                ('currency', models.CharField(max_length=3)),
+                ('ticket_fee', models.DecimalField(decimal_places=2, max_digits=10)),
+                ('payment_method', models.CharField(max_length=20, null=True)),
+                ('paid_datetime', models.DateTimeField(blank=True, null=True)),
+                ('note', models.TextField(null=True)),
+                ('monthly_bill', models.DateField(default=datetime.date.today)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('created_by', models.CharField(max_length=50)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('updated_by', models.CharField(max_length=50)),
+                ('last_reminder_datetime', models.DateTimeField(blank=True, null=True)),
+                ('next_reminder_datetime', models.DateTimeField(blank=True, null=True)),
+                ('reminder_schedule', django.contrib.postgres.fields.ArrayField(base_field=models.IntegerField(), default=list, size=None)),
+                ('reminder_enabled', models.BooleanField(default=True)),
+                ('event', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='pretixbase.event')),
+                ('organizer', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='pretixbase.organizer')),
+            ],
+            options={
+                'verbose_name': 'Billing Invoice',
+                'verbose_name_plural': 'Billing Invoices',
+                'ordering': ('-created_at',),
+            },
+            bases=(models.Model, pretix.base.models.base.LoggingMixin),
+        ),
+    ]

src/pretix/base/migrations/0004_billinginvoice_stripe_payment_intent_id_and_more.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.2 on 2024-11-05 04:08
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("pretixbase", "0003_create_billing_invoice_table"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="billinginvoice",
+            name="stripe_payment_intent_id",
+            field=models.CharField(max_length=50, null=True),
+        )
+    ]

src/pretix/base/models/__init__.py

@@ -1,6 +1,7 @@
 from ..settings import GlobalSettingsObject_SettingsStore
 from .auth import U2FDevice, User, WebAuthnDevice
 from .base import CachedFile, LoggedModel, cachedfile_name
+from .billing import BillingInvoice
 from .checkin import Checkin, CheckinList
 from .devices import Device, Gate
 from .event import (

src/pretix/base/models/billing.py

@@ -0,0 +1,59 @@
+from datetime import date
+
+from django.contrib.postgres.fields import ArrayField
+from django.db import models
+from django.utils.translation import gettext_lazy as _
+from django_scopes import ScopedManager
+
+from pretix.base.models import LoggedModel
+
+
+class BillingInvoice(LoggedModel):
+    STATUS_PENDING = "n"
+    STATUS_PAID = "p"
+    STATUS_EXPIRED = "e"
+    STATUS_CANCELED = "c"
+
+    STATUS_CHOICES = [
+        (STATUS_PENDING, _("pending")),
+        (STATUS_PAID, _("paid")),
+        (STATUS_EXPIRED, _("expired")),
+        (STATUS_CANCELED, _("canceled")),
+    ]
+
+    organizer = models.ForeignKey('Organizer', on_delete=models.CASCADE)
+    # organizer_billing = models.ForeignKey('OrganizerBilling', on_delete=models.CASCADE)
+    event = models.ForeignKey('Event', on_delete=models.CASCADE)
+
+    status = models.CharField(max_length=1, choices=STATUS_CHOICES, default=STATUS_PENDING)
+    amount = models.DecimalField(max_digits=10, decimal_places=2)
+    currency = models.CharField(max_length=3)
+
+    ticket_fee = models.DecimalField(max_digits=10, decimal_places=2)
+    payment_method = models.CharField(max_length=20, null=True, blank=True)
+    paid_datetime = models.DateTimeField(null=True, blank=True)
+    note = models.TextField(null=True, blank=True)
+
+    monthly_bill = models.DateField(default=date.today)
+    created_at = models.DateTimeField(auto_now_add=True)
+    created_by = models.CharField(max_length=50)
+    updated_at = models.DateTimeField(auto_now=True)
+    updated_by = models.CharField(max_length=50)
+
+    last_reminder_datetime = models.DateTimeField(null=True, blank=True)
+    next_reminder_datetime = models.DateTimeField(null=True, blank=True)
+    reminder_schedule = ArrayField(
+        models.IntegerField(),
+        default=list,  # Sets the default to an empty list
+        blank=True,
+        help_text="Days after creation for reminders, e.g., [14, 28]"
+    )
+    reminder_enabled = models.BooleanField(default=True)
+    stripe_payment_intent_id = models.CharField(max_length=50, null=True, blank=True)
+
+    objects = ScopedManager(organizer='organizer')
+
+    class Meta:
+        verbose_name = "Billing Invoice"
+        verbose_name_plural = "Billing Invoices"
+        ordering = ("-created_at",)

src/pretix/base/models/organizer.py

@@ -395,3 +395,92 @@ def get_events_with_permission(self, permission, request=None):
             return self.get_events_with_any_permission()
         else:
             return self.team.organizer.events.none()
+
+
+class OrganizerBillingModel(models.Model):
+    """
+    Billing model - support billing information for organizer
+    """
+
+    organizer = models.ForeignKey(
+        "Organizer", on_delete=models.CASCADE, related_name="billing"
+    )
+
+    primary_contact_name = models.CharField(
+        max_length=255,
+        verbose_name=_("Primary Contact Name"),
+    )
+
+    primary_contact_email = models.EmailField(
+        max_length=255,
+        verbose_name=_("Primary Contact Email"),
+    )
+
+    company_or_organization_name = models.CharField(
+        max_length=255,
+        verbose_name=_("Company or Organization Name"),
+    )
+
+    address_line_1 = models.CharField(
+        max_length=255,
+        verbose_name=_("Address Line 1"),
+    )
+
+    address_line_2 = models.CharField(
+        max_length=255,
+        verbose_name=_("Address Line 2"),
+    )
+
+    city = models.CharField(
+        max_length=255,
+        verbose_name=_("City"),
+    )
+
+    zip_code = models.CharField(
+        max_length=255,
+        verbose_name=_("Zip Code"),
+    )
+
+    country = models.CharField(
+        max_length=255,
+        verbose_name=_("Country"),
+    )
+
+    preferred_language = models.CharField(
+        max_length=255,
+        verbose_name=_("Preferred Language"),
+    )
+
+    tax_id = models.CharField(
+        max_length=255,
+        verbose_name=_("Tax ID"),
+    )
+
+    stripe_customer_id = models.CharField(
+        max_length=255,
+        verbose_name=_("Stripe Customer ID"),
+        blank=True,
+        null=True,
+    )
+
+    stripe_payment_method_id = models.CharField(
+        max_length=255,
+        verbose_name=_("Payment Method"),
+        blank=True,
+        null=True,
+    )
+
+    stripe_setup_intent_id = models.CharField(
+        max_length=255,
+        verbose_name=_("Setup Intent ID"),
+        blank=True,
+        null=True,
+    )
+
+    def delete(self, *args, **kwargs):
+        super().delete(*args, **kwargs)
+        self.organizer.cache.clear()
+
+    def save(self, *args, **kwargs):
+        super().save(*args, **kwargs)
+        self.organizer.cache.clear()

src/pretix/base/services/mail.py

@@ -1,3 +1,4 @@
+import base64
 import inspect
 import logging
 import os
@@ -277,7 +278,8 @@ def _create_mime_attachment(self, content, mimetype):
 def mail_send_task(self, *args, to: List[str], subject: str, body: str, html: str, sender: str,
                    event: int = None, position: int = None, headers: dict = None, bcc: List[str] = None,
                    invoices: List[int] = None, order: int = None, attach_tickets=False, user=None,
-                   attach_ical=False, attach_cached_files: List[int] = None) -> bool:
+                   attach_ical=False, attach_cached_files: List[int] = None, attach_file_base64: str = None,
+                   attach_file_name: str = None) -> bool:
     email = CustomEmail(subject, body, sender, to=to, bcc=bcc, headers=headers)
     if html is not None:
         html_message = SafeMIMEMultipart(_subtype='related', encoding=settings.DEFAULT_CHARSET)
@@ -385,6 +387,9 @@ def cm(): return scopes_disabled()  # noqa
                         pass
 
         email = global_email_filter.send_chained(event, 'message', message=email, user=user, order=order)
+        if attach_file_base64:
+            attach_file_content = base64.b64decode(attach_file_base64)
+            email.attach(attach_file_name, attach_file_content, "application/pdf")
 
         try:
             backend.send_messages([email])