mirror of
https://github.com/pretix/pretix.git
synced 2026-05-05 15:14:04 +00:00
afd766999c
* Upgrade to Django 2.0 * more models * i18n foo * Update setup.py * Fix Sentry exception PRETIXEU-JC * Enforce slug uniqueness * Import sorting * Upgrade to Django 2.1 * Travis config * Try to fix PostgreSQL failure * Smaller test matrix * staticfiles→static * Include request in all authenticate() calls
79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
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', null=True, on_delete=models.CASCADE)
|
|
organizer = models.ForeignKey('pretixbase.Organizer', null=True, on_delete=models.CASCADE)
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
state = models.CharField(max_length=32, choices=STATES, default=STATE_PENDING)
|
|
|
|
@property
|
|
def owner_kwargs(self):
|
|
if self.event:
|
|
return {'event': self.event}
|
|
else:
|
|
return {'organizer': self.organizer}
|
|
|
|
|
|
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', null=True, on_delete=models.CASCADE)
|
|
organizer = models.ForeignKey('pretixbase.Organizer', null=True, on_delete=models.CASCADE)
|
|
import_job = models.ForeignKey('BankImportJob', related_name='transactions', on_delete=models.CASCADE)
|
|
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.CharField(max_length=50)
|
|
order = models.ForeignKey('pretixbase.Order', null=True, blank=True, on_delete=models.CASCADE)
|
|
comment = models.TextField(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('', self.date.lower()).encode('utf-8'))
|
|
return str(hasher.hexdigest())
|
|
|
|
def shred_private_data(self):
|
|
self.payer = ""
|
|
self.reference = ""
|
|
|
|
class Meta:
|
|
unique_together = ('event', 'organizer', 'checksum')
|