#342 Create attachment_anonymization management command

chcemvediet/apps/anonymization/management/commands/attachment_anonymization.py

@@ -0,0 +1,65 @@
+from optparse import make_option
+
+import magic
+from django.core.files.base import ContentFile
+from django.core.management.base import BaseCommand, CommandError
+
+from poleno.attachments.models import Attachment
+from chcemvediet.apps.anonymization.models import AttachmentFinalization
+from chcemvediet.apps.inforequests.models import Action
+
+
+class Command(BaseCommand):
+ help = u'Creates AttachmentFinalization for the specified Attachment.'
+
+ option_list = BaseCommand.option_list + (
+ make_option(u'-f', u'--file',
+ dest=u'filename',
+ help=u'define file path'),
+ make_option(u'--content_type',
+ help=u'define content type of file',
+ ),
+ make_option(u'--debug',
+ default=u'',
+ help=u'add debug message'
+ ),
+ make_option(u'--force',
+ action=u'store_true',
+ help=u'overwrite an existing successful AttachmentFinalization'
+ ),
+ )
+
+ def usage(self, subcommand=None):
+ return u'manage.py attachment_anonymization attachment [options] [file]'
+
+ def handle(self, *args, **options):
+ try:
+ filename = options[u'filename'] or args[1]
+ except IndexError:
+ self.stdout.write(u'Usage: {}'.format(self.usage()))
+ self.stdout.write(
+ u'Try "manage.py attachment_anonymization --help" for more information.')
+ return 1
+
+ with open(filename, u'rb') as file:
+ try:
+ attachment = (Attachment.objects
+ .attached_to(Action)
+ .not_normalized()
+ .get(pk=int(args[0])))
+ attachments_finalization = (AttachmentFinalization.objects
+ .filter(attachment=attachment)
+ .successful())
+ if options[u'force'] or not attachments_finalization:
+ attachments_finalization.delete()
+ content = file.read()
+ content_type = magic.from_buffer(content, mime=True)
+ AttachmentFinalization.objects.create(
+ attachment=attachment,
+ successful=True,
+ file=ContentFile(content),
+ content_type=options[u'content_type'] or content_type,
+ debug=options[u'debug'],
+ )
+ except Attachment.DoesNotExist:
+ raise CommandError(u'Attachment {} does not exist'.format(args[0]))

chcemvediet/apps/anonymization/models.py

@@ -322,6 +322,9 @@ def __unicode__(self):
  return format(self.pk)
 
 class AttachmentFinalizationQuerySet(QuerySet):
+ def successful(self):
+ return self.filter(successful=True)
+
  def order_by_pk(self):
  return self.order_by(u'pk')
 

chcemvediet/apps/anonymization/tests/test_management_commands.py

@@ -0,0 +1,71 @@
+import os
+from testfixtures import TempDirectory
+
+from django.core.management import call_command
+from django.test import TestCase
+from django.test.utils import override_settings
+
+from chcemvediet.apps.anonymization.models import AttachmentFinalization
+from chcemvediet.tests import ChcemvedietTestCaseMixin
+
+
+class AttachmentAnonymizationManagementCommandTest(ChcemvedietTestCaseMixin, TestCase):
+
+ def _pre_setup(self):
+ super(AttachmentAnonymizationManagementCommandTest, self)._pre_setup()
+ self.tempdir = TempDirectory()
+ self.settings_override = override_settings(
+ MEDIA_ROOT=self.tempdir.path,
+ )
+ self.settings_override.enable()
+ self.attachment = self._create_attachment()
+
+ def _post_teardown(self):
+ self.settings_override.disable()
+ self.tempdir.cleanup()
+ super(AttachmentAnonymizationManagementCommandTest, self)._post_teardown()
+
+ def _create_attachment(self, **kwargs):
+ return super(AttachmentAnonymizationManagementCommandTest, self)._create_attachment(
+ generic_object=self.action,
+ **kwargs
+ )
+
+
+ def test_attachment_anonymization_command_with_valid_arguments_creates_attachment_finalization(self):
+ with open(os.path.join(self.tempdir.path, u'myfile.txt'), u'w') as file:
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name)
+ AttachmentFinalization.objects.get(attachment=self.attachment)
+
+ def test_attachment_anonymization_command_file_option(self):
+ with open(os.path.join(self.tempdir.path, u'myfile.txt'), u'w') as file:
+ call_command(u'attachment_anonymization', self.attachment.pk, filename=file.name)
+ AttachmentFinalization.objects.get(attachment=self.attachment)
+
+ def test_attachment_anonymization_command_content_type_option(self):
+ with open(os.path.join(self.tempdir.path, u'myfile.txt'), u'w') as file:
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name, content_type=u'type')
+ attachment_finalization = AttachmentFinalization.objects.get(attachment=self.attachment)
+ self.assertEqual(attachment_finalization.content_type, u'type')
+
+ def test_attachment_anonymization_command_debug_option(self):
+ with open(os.path.join(self.tempdir.path, u'myfile.txt'), u'w') as file:
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name, debug=u'debug')
+ attachment_finalization = AttachmentFinalization.objects.get(attachment=self.attachment)
+ self.assertEqual(attachment_finalization.debug, u'debug')
+
+ def test_attachment_anonymization_command_with_existing_attachment_finalization(self):
+ with open(os.path.join(self.tempdir.path, u'myfile.txt'), u'w') as file:
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name)
+ attachment_finalization1 = AttachmentFinalization.objects.get(attachment=self.attachment)
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name)
+ attachment_finalization2 = AttachmentFinalization.objects.get(attachment=self.attachment)
+ self.assertEqual(attachment_finalization1.pk, attachment_finalization2.pk)
+
+ def test_attachment_anonymization_command_force_option(self):
+ with open(os.path.join(self.tempdir.path, u'myfile.txt'), u'w') as file:
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name)
+ attachment_finalization1 = AttachmentFinalization.objects.get(attachment=self.attachment)
+ call_command(u'attachment_anonymization', self.attachment.pk, file.name, force=True)
+ attachment_finalization2 = AttachmentFinalization.objects.get(attachment=self.attachment)
+ self.assertNotEqual(attachment_finalization1.pk, attachment_finalization2.pk)

misc/anonymization.md

@@ -92,3 +92,34 @@ Computed Properties:
 
 
 <sub>*\* Features that are marked ~~strikethrough~~ are not implemented yet.*</sub>
+
+
+## Anonymization
+You can manually anonymize `Attachment` using management command:
+
+```
+manage.py attachment_anonymization attachment [options] [file]
+
+Options:
+ -v VERBOSITY, --verbosity=VERBOSITY
+ Verbosity level; 0=minimal output, 1=normal output,
+ 2=verbose output, 3=very verbose output
+ --settings=SETTINGS The Python path to a settings module, e.g.
+ "myproject.settings.main". If this isn't provided, the
+ DJANGO_SETTINGS_MODULE environment variable will be
+ used.
+ --pythonpath=PYTHONPATH
+ A directory to add to the Python path, e.g.
+ "/home/djangoprojects/myproject".
+ --traceback Raise on exception
+ --no-color Don't colorize the command output.
+ -f FILENAME, --file=FILENAME
+ define file path
+ --content_type=CONTENT_TYPE
+ define content type of file
+ --debug=DEBUG add debug message
+ --force overwrite an existing successful
+ AttachmentFinalization
+ --version show program's version number and exit
+ -h, --help show this help message and exit
+```