#!/usr/bin/env python3 # -*- encoding: utf-8 -*- ## news.py - Generates news. ## ## Copyright © 2012 Ben Longbons ## ## This file is part of The Mana World ## ## This program is free software: you can redistribute it and/or modify ## it under the terms of the GNU General Public License as published by ## the Free Software Foundation, either version 2 of the License, or ## (at your option) any later version. ## ## This program is distributed in the hope that it will be useful, ## but WITHOUT ANY WARRANTY; without even the implied warranty of ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ## GNU General Public License for more details. ## ## You should have received a copy of the GNU General Public License ## along with this program. If not, see . from __future__ import print_function import sys import os from abc import ABCMeta, abstractmethod try: # py3 from urllib.parse import quote_plus except ImportError: # py2 from urllib import quote_plus try: # py3 from html import escape as html_escape except ImportError: # py2 from cgi import escape as html_escape import _news_colors as colors def url_escape(s): return quote_plus(s) class BasicWriter(object): __slots__ = ('stream', 'isFirst') __metaclass__ = ABCMeta def __init__(self, outfile): self.stream = open(outfile, 'w') self.isFirst = True @abstractmethod def start(self): pass @abstractmethod def put(self, entry, path): pass @abstractmethod def finish(self): pass class HtmlWriter(BasicWriter): __slots__ = () def start(self): self.stream.write('\n') #self.stream.write('
\n')
        pass

    def put(self, entry, path):
        self.stream.write('
\n
\n') entry = entry.replace('\n\n', '\n
\n') entry = entry.format(**colors.make_html_colors_dict()) self.stream.write(entry) self.stream.write('
\n') def finish(self): #self.stream.write('
\n') pass class NOPFmt: def __format__(self, s): return s class RSSWriter(BasicWriter): __slots__ = () def start(self): self.stream.write('\nThe Mana World NewsTMW Newshttps://www.themanaworld.org/news-feed.php\n') def put(self, entry, path): self.stream.write('') title, entry = entry.lstrip('\n').split('\n', 1) self.stream.write(html_escape(title.format(**{'title':NOPFmt()}))) # FIXME: Assumes title is the first nonblank line self.stream.write('\n\n') entry = entry.replace('\n\n', '\n
\n') entry = entry.format(**colors.make_html_colors_dict()) entry = html_escape(entry) self.stream.write(entry) self.stream.write('
\nhttps://www.themanaworld.org/news-feed.php#' + url_escape(path) + '\n') self.stream.write(path) self.stream.write('\n
\n') def finish(self): self.stream.write('
\n
\n') class JSONWriter(BasicWriter): __slots__ = () def start(self): self.stream.write('[') def put(self, entry, path): if not self.isFirst: self.stream.write(',\n') self.isFirst = False self.stream.write('{') self.stream.write('"id":"%s",' % path) title, entry = entry.lstrip('\n').split('\n', 1) date, entry = entry.split('\n', 1) entry, author = entry.rstrip('\n').rsplit('\n', 1) self.stream.write('"title":"%s",' % title.format(**{'title':NOPFmt()})) # FIXME: assumes the first line is always the title self.stream.write('"date":"%s",' % date.format(**{'date':NOPFmt()})) # FIXME: assumes the second line is always the date self.stream.write('"author":"%s",' % author.format(**{'author':NOPFmt()})) # FIXME: assumes the second line is always the date self.stream.write('"html":"') entry = entry.lstrip('\n') entry = entry.replace('\n\n', '
') entry = entry.format(**colors.make_html_colors_dict()) entry = entry.replace('"', '\\"') entry = entry.replace('\n', ' ') entry = entry.rstrip('\n').rstrip(' ') self.stream.write(entry) self.stream.write('"') self.stream.write('}') def finish(self): self.stream.write(']\n') class TxtWriter(BasicWriter): __slots__ = () def start(self): pass def put(self, entry, path): entry = entry.replace('\n\n', '\n \n') entry = entry.format(**colors.make_txt_colors_dict()) self.stream.write(entry) self.stream.write(' \n \n') def finish(self): # DO NOT REMOVE #self.stream.write('Did you really read down this far?\n') pass class ForumWriter(BasicWriter): # unlike the other writers, the forum writer only generates the latest __slots__ = ('done') def start(self): self.done = False def put(self, entry, path): if self.done: return entry = entry.format(**colors.make_forum_colors_dict()) self.stream.write(entry) self.done = True def finish(self): pass def create_writers(outdir): yield TxtWriter(os.path.join(outdir, 'news.txt')) yield HtmlWriter(os.path.join(outdir, 'news.html')) yield RSSWriter(os.path.join(outdir, 'news.rss')) yield JSONWriter(os.path.join(outdir, 'news.json')) yield ForumWriter(os.path.join(outdir, 'news.phpbb.txt')) def main(outdir, indir=None): if indir is None: indir = os.path.join(outdir, 'news.d') out = list(create_writers(outdir)) for s in out: s.start() for entry in sorted(os.listdir(indir), reverse=True): suffix = '.txt' if not entry.endswith(suffix): continue e = open(os.path.join(indir, entry)).read() for s in out: s.put(e, entry[:-len(suffix)]) for s in out: s.finish() if __name__ == '__main__': main(*sys.argv[1:])