forked from varia/varia.website
90 lines
2.0 KiB
Python
90 lines
2.0 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
|
|
|
|
page_path = os.path.join(self.output_path, article.url)
|
|
if not os.path.exists(page_path):
|
|
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)
|