mail: make it possible to specify a category when feeding mails

welco/sources/mail/management/commands/feed_mail.py

@@ -14,6 +14,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+from optparse import make_option
 import os
 
 from django.core.files import File
@@ -22,7 +23,13 @@ from django.core.management.base import BaseCommand, CommandError
 from ...models import Mail
 
 class Command(BaseCommand):
-    args = ['...']
+    args = ['category', '...']
+
+    option_list = BaseCommand.option_list + (
+            make_option('--category',
+                        dest='category',
+                        default=None),
+            )
 
     def handle(self, *args, **kwargs):
         count = 0
@@ -30,6 +37,7 @@ class Command(BaseCommand):
             if not os.path.exists(filepath):
                 continue
             mail = Mail(content=File(open(filepath)))
+            mail.scanner_category = kwargs.get('category')
             mail.save()
             count += 1
         if count == 0:

welco/sources/mail/migrations/0009_mail_scanner_category.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mail', '0008_remove_mail_mail_number'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='mail',
+            name='scanner_category',
+            field=models.CharField(max_length=100, null=True, blank=True),
+        ),
+    ]

welco/sources/mail/models.py

@@ -40,6 +40,8 @@ class Mail(models.Model):
             null=True, max_length=50)
     note = models.TextField(_('Note'), null=True)
 
+    scanner_category = models.CharField(max_length=100, blank=True, null=True)
+
     # common to all source types:
     status = models.CharField(_('Status'), blank=True, max_length=50)
     contact_id = models.CharField(max_length=50, null=True)