Data model for transactional history (#2147)

2021-10-18 17:28:58 +02:00
parent c4e71011ee
commit 8ebba9de86
19 changed files with 699 additions and 10 deletions
--- a/src/pretix/base/management/commands/create_order_transactions.py
+++ b/src/pretix/base/management/commands/create_order_transactions.py
@@ -0,0 +1,84 @@
+#
+# This file is part of pretix (Community Edition).
+#
+# Copyright (C) 2014-2020 Raphael Michel and contributors
+# Copyright (C) 2020-2021 rami.io GmbH and contributors
+#
+# This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General
+# Public License as published by the Free Software Foundation in version 3 of the License.
+#
+# ADDITIONAL TERMS APPLY: Pursuant to Section 7 of the GNU Affero General Public License, additional terms are
+# applicable granting you additional permissions and placing additional restrictions on your usage of this software.
+# Please refer to the pretix LICENSE file to obtain the full terms applicable to this work. If you did not receive
+# this file, see <https://pretix.eu/about/en/license>.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
+# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Affero General Public License along with this program.  If not, see
+# <https://www.gnu.org/licenses/>.
+#
+import time
+
+from django.core.management.base import BaseCommand
+from django.db.models import F, Max, Q
+from django.utils.timezone import now
+from django_scopes import scopes_disabled
+from tqdm import tqdm
+
+from pretix.base.models import Order
+
+
+class Command(BaseCommand):
+    help = "Create missing order transactions"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--slowdown",
+            dest="interval",
+            type=int,
+            default=0,
+            help="Interval for staggered execution. If set to a value different then zero, we will "
+                 "wait this many milliseconds between every order we process.",
+        )
+
+    @scopes_disabled()
+    def handle(self, *args, **options):
+        t = 0
+        qs = Order.objects.annotate(
+            last_transaction=Max('transactions__created')
+        ).filter(
+            Q(last_transaction__isnull=True) | Q(last_modified__gt=F('last_transaction')),
+            require_approval=False,
+        ).prefetch_related(
+            'all_positions', 'all_fees'
+        )
+        for o in tqdm(qs):
+            if o.last_transaction is None:
+                tn = o.create_transactions(
+                    positions=o.all_positions.all(),
+                    fees=o.all_fees.all(),
+                    dt_now=o.datetime,
+                    migrated=True,
+                    is_new=True,
+                    _backfill_before_cancellation=True,
+                )
+                o.create_transactions(
+                    positions=o.all_positions.all(),
+                    fees=o.all_fees.all(),
+                    dt_now=o.cancellation_date or (o.expires if o.status == Order.STATUS_EXPIRED else o.datetime),
+                    migrated=True,
+                )
+            else:
+                tn = o.create_transactions(
+                    positions=o.all_positions.all(),
+                    fees=o.all_fees.all(),
+                    dt_now=now(),
+                    migrated=True,
+                )
+            if tn:
+                t += 1
+            time.sleep(options.get('slowdown', 0) / 1000)
+
+        self.stderr.write(self.style.SUCCESS(f'Created transactions for {t} orders.'))