github/pelican
1
0
Fork 1
mirror of https://github.com/getpelican/pelican.git synced 2025-10-15 20:28:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 8a5b09d6f7 into 5946313b2e

Browse source
This commit is contained in:
recursify 2015-05-13 02:10:42 +00:00
commit d4ccfa6e71
1 changed files with 30 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

55
pelican/generators.py
View file

@ -554,31 +554,7 @@ class ArticlesGenerator(CachingGenerator):
self.dates.sort(key=attrgetter('date'),
reverse=self.context['NEWEST_FIRST_ARCHIVES'])
# create tag cloud
tag_cloud = defaultdict(int)
for article in self.articles:
for tag in getattr(article, 'tags', []):
tag_cloud[tag] += 1
tag_cloud = sorted(tag_cloud.items(), key=itemgetter(1), reverse=True)
tag_cloud = tag_cloud[:self.settings.get('TAG_CLOUD_MAX_ITEMS')]
tags = list(map(itemgetter(1), tag_cloud))
if tags:
max_count = max(tags)
steps = self.settings.get('TAG_CLOUD_STEPS')
# calculate word sizes
self.tag_cloud = [
(
tag,
int(math.floor(steps - (steps - 1) * math.log(count)
/ (math.log(max_count)or 1)))
)
for tag, count in tag_cloud
]
# put words in chaos
random.shuffle(self.tag_cloud)
self.tag_cloud = self._tag_cloud_from_articles(self.articles)
# and generate the output :)
@ -602,6 +578,35 @@ class ArticlesGenerator(CachingGenerator):
signals.article_writer_finalized.send(self, writer=writer)
def _tag_cloud_from_articles(self, articles):
tag_cloud = defaultdict(int)
for article in articles:
for tag in getattr(article, 'tags', []):
tag_cloud[tag] += 1
tag_cloud = sorted(tag_cloud.items(), key=itemgetter(1), reverse=True)
tag_cloud = tag_cloud[:self.settings.get('TAG_CLOUD_MAX_ITEMS')]
tags = list(map(itemgetter(1), tag_cloud))
if tags:
max_count = max(tags)
steps = self.settings.get('TAG_CLOUD_STEPS')
# calculate word sizes
tag_cloud = [
(
tag,
int(math.floor(steps - (steps - 1) * math.log(count)
/ (math.log(max_count) or 1)))
)
for tag, count in tag_cloud
]
# put words in chaos
random.shuffle(tag_cloud)
return tag_cloud
class PagesGenerator(CachingGenerator):
"""Generate pages"""