Add model for storing metadata about transaction uploads

Author: rambo

project/ndaparser/migrations/0001_initial.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import asylum.mixins
+import ndaparser.models
+from django.conf import settings
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='UploadedTransaction',
+            fields=[
+                ('id', models.AutoField(primary_key=True, verbose_name='ID', auto_created=True, serialize=False)),
+                ('file', models.FileField(upload_to=ndaparser.models.datestamped_and_normalized)),
+                ('stamp', models.DateTimeField(auto_now_add=True)),
+                ('last_transaction', models.DateField()),
+                ('user', models.ForeignKey(on_delete=models.SET(ndaparser.models.get_sentinel_user), to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Uploaded transaction',
+                'verbose_name_plural': 'Uploaded transaction',
+                'ordering': ['-stamp'],
+            },
+            bases=(asylum.mixins.AtomicVersionMixin, asylum.mixins.CleanSaveMixin, models.Model),
+        ),
+    ]

project/ndaparser/models.py

@@ -1,6 +1,41 @@
 # -*- coding: utf-8 -*-
+import datetime
+import slugify as unicodeslugify
+
 from django.db import models, transaction
+from django.conf import settings
+from django.contrib.auth import get_user_model
+from django.utils.translation import ugettext_lazy as _
+
 
 from asylum.models import AsylumModel
 
-# Create your models here.
+
+
+def get_sentinel_user():
+    """Gets a "sentinel" user ("deleted") and for assigning as uploader"""
+    return get_user_model().objects.get_or_create(username='deleted')[0]
+
+
+
+def datestamped_and_normalized(instance, filename):
+    """Normalized filename and places in datestamped path"""
+    filename_normalized = unicodeslugify.slugify(
+        filename, only_ascii=True, lower=True,
+        spaces=False, space_replacement='_'
+    )
+    return datetime.datetime.now().strftime("ndaparser/%Y/%m/%d/{}").format(filename_normalized)
+
+
+
+class UploadedTransaction(AsylumModel):
+    """Track uploaded transaction files"""
+    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.SET(get_sentinel_user))
+    file = models.FileField(upload_to=datestamped_and_normalized)
+    stamp = models.DateTimeField(auto_now_add=True, editable=False)
+    last_transaction = models.DateField()
+
+    class Meta:
+        verbose_name = _('Uploaded transaction')
+        verbose_name_plural = _('Uploaded transaction')
+        ordering = [ '-stamp' ]