mirror of
https://github.com/DarrenOfficial/dpaste.git
synced 2024-11-15 16:12:51 +11:00
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
from django import forms
|
|
from django.conf import settings
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from dpaste.models import Snippet
|
|
from dpaste.highlight import LEXER_LIST, LEXER_DEFAULT
|
|
import datetime
|
|
|
|
#===============================================================================
|
|
# Snippet Form and Handling
|
|
#===============================================================================
|
|
|
|
EXPIRE_CHOICES = (
|
|
(3600, _(u'In one hour')),
|
|
(3600 * 24 * 7, _(u'In one week')),
|
|
(3600 * 24 * 30, _(u'In one month')),
|
|
)
|
|
|
|
EXPIRE_DEFAULT = 3600 * 24 * 30
|
|
|
|
|
|
class SnippetForm(forms.ModelForm):
|
|
lexer = forms.ChoiceField(
|
|
label=_(u'Lexer'),
|
|
initial=LEXER_DEFAULT,
|
|
widget=forms.TextInput,
|
|
)
|
|
|
|
expire_options = forms.ChoiceField(
|
|
label=_(u'Expires'),
|
|
choices=EXPIRE_CHOICES,
|
|
initial=EXPIRE_DEFAULT,
|
|
)
|
|
|
|
# Honeypot field
|
|
title = forms.CharField(
|
|
label=_(u'Title'),
|
|
required=False,
|
|
widget=forms.TextInput(attrs={'autocomplete': 'off'}),
|
|
)
|
|
|
|
def __init__(self, request, *args, **kwargs):
|
|
super(SnippetForm, self).__init__(*args, **kwargs)
|
|
self.request = request
|
|
self.fields['lexer'].choices = LEXER_LIST
|
|
self.fields['lexer'].widget.attrs = {
|
|
'autocomplete': 'off',
|
|
'data-provide': 'typeahead',
|
|
'data-source': '["%s"]' % '","'.join(dict(LEXER_LIST).keys())
|
|
}
|
|
|
|
def clean_lexer(self):
|
|
lexer = self.cleaned_data.get('lexer')
|
|
if not lexer:
|
|
return LEXER_DEFAULT
|
|
lexer = dict(LEXER_LIST).get(lexer, LEXER_DEFAULT)
|
|
return lexer
|
|
|
|
def clean(self):
|
|
if self.cleaned_data.get('title'):
|
|
raise forms.ValidationError('This snippet was identified as Spam.')
|
|
return self.cleaned_data
|
|
|
|
|
|
def save(self, parent=None, *args, **kwargs):
|
|
|
|
# Set parent snippet
|
|
if parent:
|
|
self.instance.parent = parent
|
|
|
|
# Add expire datestamp
|
|
self.instance.expires = datetime.datetime.now() + \
|
|
datetime.timedelta(seconds=int(self.cleaned_data['expire_options']))
|
|
|
|
# Save snippet in the db
|
|
super(SnippetForm, self).save(*args, **kwargs)
|
|
|
|
# Add the snippet to the user session list
|
|
if self.request.session.get('snippet_list', False):
|
|
if len(self.request.session['snippet_list']) >= getattr(settings, 'MAX_SNIPPETS_PER_USER', 10):
|
|
self.request.session['snippet_list'].pop(0)
|
|
self.request.session['snippet_list'] += [self.instance.pk]
|
|
else:
|
|
self.request.session['snippet_list'] = [self.instance.pk]
|
|
|
|
return self.request, self.instance
|
|
|
|
class Meta:
|
|
model = Snippet
|
|
fields = (
|
|
'content',
|
|
'lexer',
|
|
)
|