russell@unturf.com
b303f497fa
pelican-plugins/ lives in .gitignore so the random_article plugin never reached the build server after cleanup. Moved to lib/ which already serves as a PLUGIN_PATHS entry. Removed the os.path.exists check that caused the plugin to skip all articles when the generator ran before article directories existed in output.
85 lines
1.8 KiB
Python
85 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Random Article Plugin For Pelican
|
|
========================
|
|
|
|
This plugin generates a html file which redirect to a random article
|
|
using javascript's window.location. The generated html file is
|
|
saved at SITEURL.
|
|
"""
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import os.path
|
|
|
|
from logging import info
|
|
from codecs import open
|
|
|
|
from pelican import signals
|
|
|
|
HTML_TOP = """
|
|
<!DOCTYPE html>
|
|
<head>
|
|
<title>random</title>
|
|
<script type="text/javascript">
|
|
function redirect(){
|
|
var urls = [
|
|
"""
|
|
|
|
HTML_BOTTOM = """
|
|
""];
|
|
|
|
var index = Math.floor(Math.random() * (urls.length-1));
|
|
window.location = urls[index];
|
|
}
|
|
</script>
|
|
<body onload="redirect()">
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
ARTICLE_URL = """ "{0}/{1}",
|
|
"""
|
|
|
|
|
|
class RandomArticleGenerator(object):
|
|
"""
|
|
The structure is derived from sitemap plugin
|
|
"""
|
|
|
|
def __init__(self, context, settings, path, theme, output_path, *null):
|
|
|
|
self.output_path = output_path
|
|
self.context = context
|
|
self.siteurl = settings.get('SITEURL')
|
|
self.randomurl = settings.get('RANDOM')
|
|
|
|
def write_url(self, article, fd):
|
|
if getattr(article, 'status', 'published') != 'published':
|
|
return
|
|
|
|
fd.write(ARTICLE_URL.format(self.siteurl, article.url))
|
|
|
|
|
|
def generate_output(self, writer):
|
|
path = os.path.join(self.output_path, self.randomurl)
|
|
articles = self.context['articles']
|
|
info('writing {0}'.format(path))
|
|
|
|
if len(articles) == 0:
|
|
return
|
|
|
|
with open(path, 'w', encoding='utf-8') as fd:
|
|
fd.write(HTML_TOP)
|
|
|
|
for art in articles:
|
|
self.write_url(art, fd)
|
|
|
|
fd.write(HTML_BOTTOM)
|
|
|
|
def get_generators(generators):
|
|
return RandomArticleGenerator
|
|
|
|
|
|
def register():
|
|
signals.get_generators.connect(get_generators)
|