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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
# -*- coding: utf-8 -*-
# Copyright (c) 2012, Brian St. Pierre
# Copyright (c) 2019, Jesús E.
import sys
import collections
import hashlib # libravatar
from pelican import signals
# import pelican.generators as generators
import pelican.readers as readers
from pelican.contents import Content
# Python 3 compatibility mixin from https://stackoverflow.com/a/39166382/807307
PY3 = sys.version_info[0] >= 3
if PY3:
def cmp(a, b):
return (a > b) - (a < b)
# mixin class for Python3 supporting __cmp__
class PY3__cmp__:
def __eq__(self, other):
return self.__cmp__(other) == 0
def __ne__(self, other):
return self.__cmp__(other) != 0
def __gt__(self, other):
return self.__cmp__(other) > 0
def __lt__(self, other):
return self.__cmp__(other) < 0
def __ge__(self, other):
return self.__cmp__(other) >= 0
def __le__(self, other):
return self.__cmp__(other) <= 0
else:
class PY3__cmp__:
pass
class Comment(Content, PY3__cmp__):
mandatory_properties = ('post_id', 'author')
# this is required, but not used
default_template = 'comment'
default_status = 'published'
def __cmp__(self, other):
"""
Comment comparison is by date. This is so that a comment list
can easily be sorted by date. The date attribute will be
automatically set from metadata by Content base class
"""
if self.date == other.date:
return 0
elif self.date < other.date:
return -1
else:
return 1
def __hash__(self):
return hash((self.post_id, self.author, self.date, self._content))
class CommentReader(object):
def __init__(self, generator):
self._comments = collections.defaultdict(list)
reader = readers.Readers(settings=generator.settings)
comments_dir = generator.settings.get('COMMENTS_DIR', 'comments')
comment_filenames = generator.get_files(comments_dir)
for comment_filename in comment_filenames:
comment = reader.read_file(
base_path=generator.settings['PATH'],
path=comment_filename,
content_class=Comment,
context=generator.context)
if 'post_id' not in comment.metadata:
raise Exception("comment %s does not have a post_id" %
(comment_filename, ))
self._comments[comment.metadata['post_id']].append(comment)
# LibreAvatar
librevatar_url = 'https://cdn.libravatar.org/avatar'
# Check the presence of the Email header
if 'email' in comment.metadata:
email = comment.metadata['email'].encode('utf-8')
ehash = hashlib.md5(email.strip().lower()).hexdigest()
comment.avatar = "%s/%s" % (librevatar_url, ehash)
else:
raise Exception("comment %s does not have a email" %
(comment_filename))
for slug, comments in self._comments.items():
comments.sort()
for n, comment in enumerate(comments):
comment.cid = n
def get_comments(self, slug):
return self._comments[slug]
def comment_initialization(generator):
"""
Set up the comment plugin.
"""
generator.plugin_instance = CommentReader(generator)
def add_comments(generator, metadata):
metadata["comments"] = generator.plugin_instance.get_comments(
metadata['slug'])
def register():
signals.article_generator_init.connect(comment_initialization)
signals.article_generator_context.connect(add_comments)
|
