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):
27 settings.setdefault('TAG_CLOUD_STEPS', 4)
28 settings.setdefault('TAG_CLOUD_MAX_ITEMS', 100)
29 settings.setdefault('TAG_CLOUD_SORTING', 'random')
30 settings.setdefault('TAG_CLOUD_BADGE', False)
33 def init_default_config(pelican):
34 from pelican.settings import DEFAULT_CONFIG
35 set_default_settings(DEFAULT_CONFIG)
37 set_default_settings(pelican.settings)
40 def generate_tag_cloud(generator):
41 tag_cloud = defaultdict(int)
42 for article in generator.articles:
43 for tag in getattr(article, 'tags', []):
46 tag_cloud = sorted(tag_cloud.items(), key=itemgetter(1), reverse=True)
47 tag_cloud = tag_cloud[:generator.settings.get('TAG_CLOUD_MAX_ITEMS')]
49 tags = list(map(itemgetter(1), tag_cloud))
52 steps = generator.settings.get('TAG_CLOUD_STEPS')
54 # calculate word sizes
55 def generate_tag(tag, count):
58 int(math.floor(steps - (steps - 1) * math.log(count)
59 / (math.log(max_count)or 1)))
61 if generator.settings.get('TAG_CLOUD_BADGE'):
66 generate_tag(tag, count)
67 for tag, count in tag_cloud
70 sorting = generator.settings.get('TAG_CLOUD_SORTING')
72 if sorting == 'alphabetically':
73 # `.lower()` to avoid putting all uppercase tags lexicographically
74 # before lowercase tags —ZMD
75 tag_cloud.sort(key=lambda elem: elem[0].name.lower())
76 elif sorting == 'alphabetically-rev':
77 tag_cloud.sort(key=lambda elem: elem[0].name, reverse=True)
78 elif sorting == 'size':
79 tag_cloud.sort(key=lambda elem: elem[1])
80 elif sorting == 'size-rev':
81 tag_cloud.sort(key=lambda elem: elem[1], reverse=True)
82 elif sorting == 'random':
83 random.shuffle(tag_cloud)
85 logger.warning("setting for TAG_CLOUD_SORTING not recognized: %s, "
86 "falling back to 'random'", sorting)
87 random.shuffle(tag_cloud)
89 # make available in context
90 generator.tag_cloud = tag_cloud
91 generator._update_context(['tag_cloud'])
95 signals.initialized.connect(init_default_config)
96 signals.article_generator_finalized.connect(generate_tag_cloud)