Banktransfer: Big import refactoring

src/pretix/plugins/banktransfer/models.py

@@ -0,0 +1,68 @@
+import hashlib
+import re
+
+from django.db import models
+
+
+class BankImportJob(models.Model):
+    STATE_PENDING = 'pending'
+    STATE_RUNNING = 'running'
+    STATE_ERROR = 'error'
+    STATE_COMPLETED = 'completed'
+    STATES = (
+        (STATE_PENDING, 'pending'),
+        (STATE_RUNNING, 'running'),
+        (STATE_ERROR, 'error'),
+        (STATE_COMPLETED, 'completed'),
+    )
+
+    event = models.ForeignKey('pretixbase.Event')
+    created = models.DateTimeField(auto_now_add=True)
+    state = models.CharField(max_length=32, choices=STATES, default=STATE_PENDING)
+
+
+class BankTransaction(models.Model):
+    STATE_UNCHECKED = 'imported'
+    STATE_NOMATCH = 'nomatch'
+    STATE_INVALID = 'invalid'
+    STATE_ERROR = 'error'
+    STATE_VALID = 'valid'
+    STATE_DISCARDED = 'discarded'
+    STATE_DUPLICATE = 'already'
+
+    STATES = (
+        (STATE_UNCHECKED, 'imported, unchecked'),
+        (STATE_NOMATCH, 'no match'),
+        (STATE_INVALID, 'not valid'),
+        (STATE_ERROR, 'error'),
+        (STATE_VALID, 'valid'),
+        (STATE_DUPLICATE, 'valid, already paid'),
+        (STATE_DISCARDED, 'manually discarded'),
+    )
+
+    event = models.ForeignKey('pretixbase.Event')
+    import_job = models.ForeignKey('BankImportJob', related_name='transactions')
+    state = models.CharField(max_length=32, choices=STATES, default=STATE_UNCHECKED)
+    message = models.TextField()
+    checksum = models.CharField(max_length=190, db_index=True)
+    payer = models.TextField(blank=True)
+    reference = models.TextField(blank=True)
+    amount = models.DecimalField(max_digits=10, decimal_places=2)
+    date = models.DateField()
+    order = models.ForeignKey('pretixbase.Order', null=True, blank=True)
+
+    def calculate_checksum(self):
+        clean = re.compile('[^a-zA-Z0-9.-]')
+        hasher = hashlib.sha1()
+        hasher.update(clean.sub('', self.payer.lower()).encode('utf-8'))
+        hasher.update(clean.sub('', self.reference.lower()).encode('utf-8'))
+        hasher.update(clean.sub('', str(self.amount).lower()).encode('utf-8'))
+        hasher.update(clean.sub('', str(self.date).lower()).encode('utf-8'))
+        return str(hasher.hexdigest())
+
+    def shred_private_data(self):
+        self.payer = ""
+        self.reference = ""
+
+    class Meta:
+        unique_together = ('event', 'checksum')