From 98dd64bab259c27e97f99d36d55d6d3444bc7ec8 Mon Sep 17 00:00:00 2001 From: Andrey Antukh Date: Mon, 29 Sep 2014 17:41:03 +0200 Subject: [PATCH] Minor fixes on sample data related to attachments model changes. --- .../management/commands/sample_data.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/taiga/projects/management/commands/sample_data.py b/taiga/projects/management/commands/sample_data.py index b9288034..929b2a09 100644 --- a/taiga/projects/management/commands/sample_data.py +++ b/taiga/projects/management/commands/sample_data.py @@ -190,19 +190,19 @@ class Command(BaseCommand): project.save() - def create_attachment(self, object, order): - attachment = Attachment.objects.create(project=object.project, - content_type=ContentType.objects.get_for_model(object.__class__), - content_object=object, + def create_attachment(self, obj, order): + attached_file = self.sd.file_from_directory(*ATTACHMENT_SAMPLE_DATA) + membership = self.sd.db_object_from_queryset(obj.project.memberships + .filter(user__isnull=False)) + attachment = Attachment.objects.create(project=obj.project, + name=path.basename(attached_file.name).lower(), + size=attached_file.size, + content_object=obj, order=order, + owner=membership.user, is_deprecated=self.sd.boolean(), description=self.sd.words(3, 12), - object_id=object.id, - owner=self.sd.db_object_from_queryset( - object.project.memberships.filter(user__isnull=False)).user, - attached_file=self.sd.file_from_directory( - *ATTACHMENT_SAMPLE_DATA)) - + attached_file=attached_file) return attachment def create_wiki(self, project, slug):