pasteque/paste/forms.py

24 lines
692 B
Python

from django.forms import ModelForm
from .models import Paste, Language
class PasteForm(ModelForm):
"""Paste model form."""
class Meta:
model = Paste
fields = ['language', 'title', 'password', 'content', 'lifetime',
'lifecount', 'private']
def save(self, commit=True):
"""Overwrites save method."""
paste = super(PasteForm, self).save(commit=False)
paste.compute_size()
if not self.cleaned_data['title']:
paste.title = 'no title'
if self.cleaned_data['password']:
paste.set_password(self.cleaned_data['password'])
if commit:
paste.save()
return paste