|
1
by Martin Mahner
Initial structure and data |
1 |
from django import forms |
2 |
from django.utils.translation import ugettext_lazy as _ |
|
3 |
from django.contrib.contenttypes.models import ContentType |
|
4 |
from attachments.models import Attachment |
|
5 |
||
6 |
class AttachmentForm(forms.ModelForm): |
|
7 |
attachment_file = forms.FileField(label=_('Upload attachment')) |
|
8 |
||
9 |
class Meta: |
|
10 |
model = Attachment |
|
11 |
fields = ('attachment_file',) |
|
12 |
||
13 |
def save(self, request, obj, *args, **kwargs): |
|
14 |
self.instance.creator = request.user |
|
15 |
self.instance.content_type = ContentType.objects.get_for_model(obj) |
|
16 |
self.instance.object_id = obj.id |
|
17 |
super(AttachmentForm, self).save(*args, **kwargs) |