1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# -*- coding: utf-8 -*-
import re
from django import forms
from django.forms import widgets
from django.contrib.contenttypes.models import ContentType
from django.utils.translation import ugettext_lazy as _
from wiki.models import Article
from wiki.templatetags.wiki_extras import WIKI_WORD_RE
wikiword_pattern = re.compile('^' + WIKI_WORD_RE + '$')
class ArticleForm(forms.ModelForm):
summary = forms.CharField(widget=forms.Textarea)
comment = forms.CharField(required=False)
user_ip = forms.CharField(widget=forms.HiddenInput)
content_type = forms.ModelChoiceField(
queryset=ContentType.objects.all(),
required=False,
widget=forms.HiddenInput)
object_id = forms.IntegerField(required=False,
widget=forms.HiddenInput)
action = forms.CharField(widget=forms.HiddenInput)
class Meta:
model = Article
exclude = ('creator', 'creator_ip',
'group', 'created_at', 'last_update')
def clean_title(self):
""" Page title must be a WikiWord.
"""
title = self.cleaned_data['title']
if not wikiword_pattern.match(title):
raise forms.ValidationError(_('Must be a WikiWord.'))
return title
def clean(self):
super(ArticleForm, self).clean()
kw = {}
if self.cleaned_data['action'] == 'create':
try:
kw['title'] = self.cleaned_data['title']
kw['content_type'] = self.cleaned_data['content_type']
kw['object_id'] = self.cleaned_data['object_id']
except KeyError:
pass # some error in this fields
else:
if Article.objects.filter(**kw).count():
raise forms.ValidationError(
_("An article with this title already exists."))
return self.cleaned_data
def cache_old_content(self):
if self.instance.id is None:
self.old_title = self.old_content = self.old_markup = ''
self.is_new = True
else:
self.old_title = self.instance.title
self.old_content = self.instance.content
self.old_markup = self.instance.markup
self.is_new = False
def save(self, *args, **kwargs):
# 0 - Extra data
editor_ip = self.cleaned_data['user_ip']
comment = self.cleaned_data['comment']
# 2 - Save the Article
article = super(ArticleForm, self).save(*args, **kwargs)
# 3 - Set creator and group
editor = getattr(self, 'editor', None)
group = getattr(self, 'group', None)
if self.is_new:
article.creator_ip = editor_ip
if editor is not None:
article.creator = editor
article.group = group
article.save(*args, **kwargs)
# 4 - Create new revision
changeset = article.new_revision(
self.old_content, self.old_title, self.old_markup,
comment, editor_ip, editor)
return article, changeset
|