1 # from https://github.com/getpelican/pelican-plugins as of f4d717ff
3 # but with a fix to the alphabetical sorting
7 ===================================
9 This plugin generates a tag cloud from available tags
11 from __future__ import unicode_literals
13 from collections import defaultdict
14 from operator import itemgetter
20 from pelican import signals
22 logger = logging.getLogger(__name__)
25 def set_default_settings(settings):
26 settings.setdefault('TAG_CLOUD_STEPS', 4)
27 settings.setdefault('TAG_CLOUD_MAX_ITEMS', 100)
28 settings.setdefault('TAG_CLOUD_SORTING', 'random')
29 settings.setdefault('TAG_CLOUD_BADGE', False)
32 def init_default_config(pelican):
33 from pelican.settings import DEFAULT_CONFIG
34 set_default_settings(DEFAULT_CONFIG)
36 set_default_settings(pelican.settings)
39 def generate_tag_cloud(generator):
40 tag_cloud = defaultdict(int)
41 for article in generator.articles:
42 for tag in getattr(article, 'tags', []):
45 tag_cloud = sorted(tag_cloud.items(), key=itemgetter(1), reverse=True)
46 tag_cloud = tag_cloud[:generator.settings.get('TAG_CLOUD_MAX_ITEMS')]
48 tags = list(map(itemgetter(1), tag_cloud))
51 steps = generator.settings.get('TAG_CLOUD_STEPS')
53 # calculate word sizes
54 def generate_tag(tag, count):
57 int(math.floor(steps - (steps - 1) * math.log(count)
58 / (math.log(max_count)or 1)))
60 if generator.settings.get('TAG_CLOUD_BADGE'):
65 generate_tag(tag, count)
66 for tag, count in tag_cloud
69 sorting = generator.settings.get('TAG_CLOUD_SORTING')
71 if sorting == 'alphabetically':
72 # `.lower()` to avoid putting all uppercase tags lexicographically
73 # before lowercase tags —ZMD
74 tag_cloud.sort(key=lambda elem: elem[0].name.lower())
75 elif sorting == 'alphabetically-rev':
76 tag_cloud.sort(key=lambda elem: elem[0].name, reverse=True)
77 elif sorting == 'size':
78 tag_cloud.sort(key=lambda elem: elem[1])
79 elif sorting == 'size-rev':
80 tag_cloud.sort(key=lambda elem: elem[1], reverse=True)
81 elif sorting == 'random':
82 random.shuffle(tag_cloud)
84 logger.warning("setting for TAG_CLOUD_SORTING not recognized: %s, "
85 "falling back to 'random'", sorting)
86 random.shuffle(tag_cloud)
88 # make available in context
89 generator.tag_cloud = tag_cloud
90 generator._update_context(['tag_cloud'])
94 signals.initialized.connect(init_default_config)
95 signals.article_generator_finalized.connect(generate_tag_cloud)