#!/usr/bin/env python3 # Copyright (C) 2018 Lubuntu Team # Authored by: Simon Quigley # # 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 3 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 . import sys import yaml from os import path from babel import Locale from markdown import markdown from wordpress_xmlrpc import Client from wordpress_xmlrpc.methods import posts from wordpress_xmlrpc.methods import taxonomies def getdirectories(args): directories = [] for directory in args: if path.isdir(directory) and path.exists(directory+"/info.yaml") and path.exists(directory+"/post.md"): directories.append(directory) return directories def loadwp(): user = open("user.yaml", "r") userinfo = yaml.load(user, Loader=yaml.CSafeLoader) wp = Client(url=userinfo["url"], username=userinfo["username"], password=userinfo["password"]) user.close() return wp def loadposts(): wp = loadwp() offset = 0 increment = 10 wpposts = [] while True: wppostlist = wp.call(posts.GetPosts({'number': increment, 'offset': offset})) if len(wppostlist) == 0: break for post in wppostlist: wpposts.append(post) offset = offset + increment return wpposts def updatepost(infoyaml, md, post=None, title=None, slug=None): wp = loadwp() if not post: post = WordPressPost() post.id = wp.call(posts.NewPost(post)) if not title: title = infoyaml["title"] if not slug: slug = infoyaml["slug"] title = title.replace("POUND", "#") if "l10n" in infoyaml: l10nstr = "" for lang in infoyaml["l10n"]: try: locale = Locale(lang) except: break langname = locale.display_name if l10nstr != "": l10nstr = l10nstr + ", " else: l10nstr = "Translated into: " l10nstr = l10nstr + "" + langname + "" message = "\n\n" + l10nstr else: message = "" md = md.replace("NOTICE", message) md = md.replace("

"+message+"

", message) post.title = title post.content = md post.post_status = "publish" tags = [] for category in wp.call(taxonomies.GetTerms('category')): for posttype in infoyaml["type"]: if category.name == posttype: tags.append(category) post.terms = tags wp.call(posts.EditPost(post.id, post)) def main(): posts = loadposts() for directory in getdirectories(sys.argv[1:]): info = open(directory+"/info.yaml", "r", encoding="utf-8") infoyaml = yaml.load(info, Loader=yaml.CSafeLoader) info.close() mdf = open(directory+"/post.md", "r", encoding="utf-8") mdt = mdf.read() md = markdown(text=mdt) mdf.close() post = None for post in posts: if post.title == infoyaml["title"].replace("POUND", "#"): updatepost(post=post, title=post.title, infoyaml=infoyaml, md=md) post = True break elif post.slug == infoyaml["slug"]: updatepost(post=post, slug=post.slug, infoyaml=infoyaml, md=md) post = True break if not post: updatepost(infoyaml=infoyaml, md=md) if __name__ == "__main__": main()