Dynastie/generators/post.py

243 lines
8.7 KiB
Python
Raw Normal View History

2012-09-11 22:25:58 +02:00
import datetime
2012-08-28 09:09:14 +02:00
import os
from xml.dom.minidom import parse, parseString
2012-09-11 22:25:58 +02:00
from dynastie.generators.generator import DynastieGenerator, StrictUTF8Writer
2012-08-28 09:09:14 +02:00
from dynastie.generators.index import Index
from django.db import models
2012-10-20 19:05:29 +02:00
from dynastie.tree import TreeNode
2012-08-28 09:09:14 +02:00
class Post(Index):
cur_comment = None
comment_index = {}
def createReplace(self, post, dom, root, replace_elem):
if not replace_elem.hasAttribute('div_name'):
self.addError('No attribute div_name for a replace tag')
return
div_element = replace_elem.cloneNode(True)
div_element.tagName = replace_elem.getAttribute('div_name')
div_element.removeAttribute('div_name')
for key,value in replace_elem.attributes.items():
if key == 'div_name': continue
value = value.replace('dyn:post_id', str(post.id))
if self.cur_comment is None:
2012-10-20 19:05:29 +02:00
value = value.replace('dyn:comment_index', '0')
else:
2012-10-20 19:05:29 +02:00
value = value.replace('dyn:comment_index', str(self.comment_index[self.cur_comment.id]))
if self.cur_comment is None:
value = value.replace('dyn:comment_id', '0')
else:
2012-10-20 19:05:29 +02:00
value = value.replace('dyn:comment_id', str(self.cur_comment.id))
2012-12-10 19:30:25 +01:00
value = value.replace('dyn:blog_id', str(self.blog.id))
div_element.setAttribute(key, value)
root.replaceChild(div_element, replace_elem)
return div_element
def createComment(self, comment, dom, comment_elem, root):
values = {}
2012-10-20 19:05:29 +02:00
values['comment_index'] = str(self.comment_index[comment.id])
values['comment_author'] = comment.author
values['comment_date'] = comment.date.strftime('%d %B %Y %H:%m')
values['comment_content'] = comment.the_comment
self.simpleTransform(values, dom, comment_elem, root)
2012-10-20 19:05:29 +02:00
def _createComments(self, rootNode, post, dom, root_comment, root):
self.cur_comment = rootNode.value
comment_element = self.createElement(dom, 'comment')
self.createComment(self.cur_comment, dom, comment_element, root)
root_comment.appendChild(comment_element)
# Parse inner HTML
self._parse(self.hooks, post, dom, comment_element)
for commentNode in rootNode.childs:
self._createComments(commentNode, post, dom, comment_element, root)
def createComments(self, post, dom, post_elem, root):
from dynastie.models import Post, Blog, Comment
comments = Comment.objects.filter(post=post).order_by('date')
cur_comment = None
comment_index = {}
index = 1
2012-10-20 19:05:29 +02:00
rootNode = TreeNode('', '')
for comment in comments:
self.comment_index[comment.id] = index
index = index + 1
2012-10-20 19:05:29 +02:00
tnode = TreeNode(comment.id, comment)
if comment.parent is None:
rootNode.addChildNode(tnode)
else:
temp = rootNode.find(comment.parent.id)
if temp is None:
self.addWarning('Error with comments chain')
rootNode.addChildNode(tnode)
else:
2012-10-20 19:05:29 +02:00
temp.addChildNode(tnode)
initial_root_comment = root_comment = self.createElement(dom, 'comments')
2012-10-20 19:05:29 +02:00
for tnode in rootNode.childs:
self._createComments(tnode, post, dom, root_comment, root)
# Empty tag seems to crap rendering
2012-10-20 19:05:29 +02:00
if len(rootNode.childs) == 0:
post_elem.removeChild(root)
return None
else:
post_elem.replaceChild(root_comment, root)
return root_comment
def createMetas(self, post, dom, meta_elem, root):
name = root.getAttribute('name')
if name is None:
self.addError('Missing name attribute in dyn:meta')
return
new_elem = None
if name == 'keywords':
new_elem = self.createMeta(dom, name, post.keywords)
elif name == 'title':
new_elem = self.createMeta(dom, name, post.title)
elif name == 'description':
new_elem = self.createMeta(dom, name, post.description)
elif name == 'author':
new_elem = self.createMeta(dom, name, post.author.first_name + ' ' + post.author.last_name)
if not new_elem is None:
root.parentNode.replaceChild(new_elem, root)
return new_elem
else:
self.addError('name attribute \'' + name + '\' unknown for dyn:meta' )
return None
2012-08-28 09:09:14 +02:00
def _createPost(self, post, dom, post_elem, root):
import sys, traceback
2012-08-28 09:09:14 +02:00
self.createPost(post, dom, post_elem, root)
# Post are appended by index. Remove template
post_nodes = dom.getElementsByTagNameNS(self.URI, 'post')
2012-08-28 09:09:14 +02:00
post_elem = post_nodes[0]
post_elem.parentNode.removeChild(post_elem)
title_nodes = dom.getElementsByTagName('title')
2012-08-28 10:42:33 +02:00
# Set title to be title's post
for node in title_nodes:
if node.hasChildNodes():
node.removeChild(node.childNodes[0])
node.appendChild(dom.createTextNode(post.title))
return node
2012-10-09 20:47:12 +02:00
def _generate(self, blog, src, output, posts):
import xml
2012-10-20 19:05:29 +02:00
self.hooks = {'post' : self._createPost,
'meta' : self.createMetas,
'comments' : self.createComments,
'replace' : self.createReplace,
'tags' : self.createTags}
2012-08-28 09:09:14 +02:00
if not os.path.exists(src + '/_post.html'):
self.addError('No _post.html found, exiting')
return self.report
2012-10-04 21:49:33 +02:00
if not os.path.exists(output + '/post'):
os.mkdir(output + '/post')
2012-08-28 09:09:14 +02:00
try:
dom = parse(src + '/_post.html')
except xml.dom.DOMException as e:
self.addError('Error parsing _post.html : ' + e)
return self.report
for post in posts:
#print 'Generate ' + filename
nodes = dom.getElementsByTagName("*")
2012-10-20 19:05:29 +02:00
nodes[0] = self.parse(src, self.hooks, post, dom, nodes[0])
2012-08-28 09:09:14 +02:00
filename = output + '/post/'
filename = filename + post.creation_date.strftime("%Y") + '/' + post.creation_date.strftime("%m") + '/'
if not os.path.exists(filename):
os.makedirs(filename)
filename = filename + post.title_slug + '.html'
self.writeIfNotTheSame(filename, nodes[0])
dom = parse(src + '/_post.html')
if not self.somethingWrote:
self.addReport('Nothing changed')
return self.report
2012-10-04 21:49:33 +02:00
def generate(self, blog, src, output):
from dynastie.models import Post, Blog
2012-12-10 19:30:25 +01:00
self.blog = blog
2012-10-04 21:49:33 +02:00
posts = Post.objects.all()
2012-10-09 20:47:12 +02:00
return self._generate(blog, src, output, posts)
2012-10-04 21:49:33 +02:00
2012-09-11 22:25:58 +02:00
def createPreview(self, values, dom, root, node):
now = datetime.datetime.now()
v = {}
v['title'] = self.createLinkElem(dom, '/preview.html', values['title'])
v['author'] = values['author']
v['date'] = now.strftime("%A, %d %B %Y %H:%m")
v['post_content'] = ''
self.simpleTransform(v, dom, root, node)
content_nodes = root.getElementsByTagName("div")
post_transform = ('post_content')
for content_node in content_nodes:
the_class = content_node.getAttribute('class')
if not the_class in post_transform:
continue
if the_class == 'post_content':
new_node = dom.createTextNode(values['content'])
content_node.appendChild(new_node)
post_nodes = dom.getElementsByTagNameNS(self.URI, "post")
post_elem = post_nodes[0]
post_elem.parentNode.removeChild(post_elem)
return post_elem
2012-09-11 22:25:58 +02:00
def preview(self, src, values):
from dynastie.models import Blog
self.hooks = {'post' : self.createPreview,
'tags' : self.createTags}
2012-09-11 22:25:58 +02:00
if not os.path.exists(src + '/_post.html'):
self.addError('No _post.html found, exiting')
return self.report
try:
dom = parse(src + '/_post.html')
except xml.dom.DOMException as e:
self.addError('Error parsing _post.html : ' + e)
return self.report
post_nodes = dom.getElementsByTagNameNS(self.URI, "posts")
if post_nodes is None:
self.addError('No tag dyn:posts found')
return self.report
nodes = dom.getElementsByTagName("*")
2012-10-20 19:05:29 +02:00
nodes[0] = self.parse(src, self.hooks, values, dom, nodes[0])
2012-09-11 22:25:58 +02:00
writer = StrictUTF8Writer()
nodes[0].writexml(writer)
return writer.getvalue().encode('utf-8')