import re from docutils import nodes from docutils.parsers.rst import Directive, directives from docutils.writers._html_base import HTMLTranslator class SEONode(nodes.General, nodes.Element): def __init__(self, title=None, description=None, image=None, author=None, author_twitter=None, keywords=None): super(SEONode, self).__init__() self.title = title self.description = description.replace('\n', ' ') self.image = image self.author = author self.author_twitter = author_twitter self.keywords = keywords class RedirectNode(nodes.General, nodes.Element): def __init__(self, url=None): super(RedirectNode, self).__init__() self.url = url def seo_visit(self: HTMLTranslator, node: SEONode): def encode_text(text): special_characters = {ord('&'): '&', ord('<'): '<', ord('"'): '"', ord('>'): '>'} return text.translate(special_characters) def create_content_meta(name, content): if content is None: return self.meta.append('\n'.format(name, encode_text(content))) def create_itemprop_meta(name, content): if content is None: return self.meta.append('\n'.format(name, encode_text(content))) def create_property_meta(name, content): if content is None: return self.meta.append('\n'.format(name, encode_text(content))) # Base create_content_meta("description", node.description) create_content_meta("keywords", node.keywords) # Schema.org create_itemprop_meta("name", node.title) create_itemprop_meta("description", node.description) create_itemprop_meta("image", node.image) # Twitter create_content_meta("twitter:title", node.title) create_content_meta("twitter:image:src", node.image) if node.author: create_content_meta("twitter:card", "summary_large_image") else: create_content_meta("twitter:card", "summary") create_content_meta("twitter:site", "@OttoWinter_") create_content_meta("twitter:creator", node.author_twitter) create_content_meta("twitter:description", node.description) # Open Graph create_property_meta("og:title", node.title) create_property_meta("og:image", node.image) create_property_meta("og:type", "article" if node.author is not None else "website") create_property_meta("og:description", node.description) def redirect_visit(self: HTMLTranslator, node: RedirectNode): self.meta.append(''.format(node.url)) self.body.append(self.starttag(node, 'p', 'Redirecting to {0}'.format(node.url))) def seo_depart(self, _): pass def redirect_depart(self, _): self.body.append('

') class SEODirective(Directive): option_spec = { 'title': directives.unchanged, 'description': directives.unchanged, 'image': directives.path, 'author': directives.unchanged, 'author_twitter': directives.unchanged, 'keywords': directives.unchanged, } def run(self): env = self.state.document.settings.env title_match = re.match(r'.+(.+).+', str(self.state.document)) if title_match is not None and 'title' not in self.options: self.options['title'] = title_match.group(1) image = self.options.get('image') if image is not None: if not image.startswith('/'): image = '/_images/' + image self.options['image'] = env.config.html_baseurl + image return [SEONode(**self.options)] class RedirectDirective(Directive): option_spec = { 'url': directives.unchanged, } def run(self): return [RedirectNode(**self.options)] def setup(app): app.add_directive('seo', SEODirective) app.add_node(SEONode, html=(seo_visit, seo_depart)) app.add_directive('redirect', RedirectDirective) app.add_node(RedirectNode, html=(redirect_visit, redirect_depart)) return {'version': '1.0'}