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 | from django.utils.encoding import force_unicode
from django.conf import settings
from django.contrib.comments.forms import CommentForm
from django.contrib.comments.models import Comment
from django.contrib.contenttypes.models import ContentType
from mptt_comments.models import MpttComment
from django import forms
import time
import datetime
class MpttCommentForm(CommentForm):
title = forms.CharField()
parent_pk = forms.IntegerField(widget=forms.HiddenInput, required=False)
def __init__(self, target_object, parent_comment=None, data=None, initial=None):
self.parent_comment = parent_comment
super(MpttCommentForm, self).__init__(target_object, data=data, initial=initial)
self.fields.keyOrder = [
'title',
'name',
'email',
'url',
'comment',
'honeypot',
'content_type',
'object_pk',
'timestamp',
'security_hash',
'parent_pk'
]
def get_comment_object(self):
"""
Return a new (unsaved) comment object based on the information in this
form. Assumes that the form is already validated and will throw a
ValueError if not.
Does not set any of the fields that would come from a Request object
(i.e. ``user`` or ``ip_address``).
"""
if not self.is_valid():
raise ValueError("get_comment_object may only be called on valid forms")
parent_comment = None
parent_pk = self.cleaned_data.get("parent_pk")
if parent_pk:
parent_comment = MpttComment.objects.get(pk=parent_pk)
new = MpttComment(
content_type = ContentType.objects.get_for_model(self.target_object),
object_pk = force_unicode(self.target_object._get_pk_val()),
user_name = self.cleaned_data["name"],
user_email = self.cleaned_data["email"],
user_url = self.cleaned_data["url"],
comment = self.cleaned_data["comment"],
submit_date = datetime.datetime.now(),
site_id = settings.SITE_ID,
is_public = True,
is_removed = False,
title = self.cleaned_data["title"],
parent = parent_comment
)
# Check that this comment isn't duplicate. (Sometimes people post comments
# twice by mistake.) If it is, fail silently by returning the old comment.
possible_duplicates = MpttComment.objects.filter(
content_type = new.content_type,
object_pk = new.object_pk,
user_name = new.user_name,
user_email = new.user_email,
user_url = new.user_url,
parent = parent_comment
)
for old in possible_duplicates:
if old.submit_date.date() == new.submit_date.date() and old.comment == new.comment:
return old
return new
def generate_security_data(self):
"""Generate a dict of security data for "initial" data."""
timestamp = int(time.time())
security_dict = {
'content_type' : str(self.target_object._meta),
'object_pk' : str(self.target_object._get_pk_val()),
'timestamp' : str(timestamp),
'security_hash' : self.initial_security_hash(timestamp),
'parent_pk' : self.parent_comment and str(self.parent_comment.pk) or '',
'title' : self.parent_comment.level == 0 and force_unicode(self.target_object) or
u'%s%s' % ( (self.parent_comment.title[:3] != u'Re:') and 'Re: ' or u'', self.parent_comment.title)
}
return security_dict
|
