18 lines
522 B
Python
18 lines
522 B
Python
from django.forms import ModelForm
|
|
|
|
from . import models
|
|
|
|
class UploadForm(ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
self.user = kwargs.pop('user')
|
|
super(UploadForm, self).__init__(*args, **kwargs)
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.instance.user = self.user
|
|
self.instance.document_filename = self.files['document_file'].name
|
|
return super(UploadForm, self).save(*args, **kwargs)
|
|
|
|
class Meta:
|
|
model = models.Document
|
|
fields = ['document_file']
|