Dynastie/generators/atom.py

75 lines
3.1 KiB
Python
Raw Normal View History

import os
import datetime
import xml
from dynastie.generators.generator import DynastieGenerator
from dynastie.generators.rss import RSS
from xml.dom.minidom import getDOMImplementation
from django.db import models
class Atom(RSS):
def generate(self, blog, src, output):
2012-08-28 09:09:14 +02:00
from dynastie.models import Post, Blog
now = datetime.datetime.now()
impl = getDOMImplementation()
dom = impl.createDocument(None, "feed", None)
root = dom.documentElement
root.setAttributeNS('xml', 'xml:lang', 'en-gb')
root.setAttribute('xmlns', 'http://www.w3.org/2005/Atom')
self.appendElement(dom, root, 'title', blog.title, {'type':'text'})
self.appendElement(dom, root, 'subtitle', blog.description, {'type':'text'})
address = 'http://' + blog.name
self.appendElement(dom, root, 'link', '', {'rel':'alternate', 'type':'text/html', 'href' : address})
self.appendElement(dom, root, 'id', address)
builddate = now.strftime('%Y-%m-%dT%H:%M:%SZ')
self.appendElement(dom, root, 'updated', builddate)
self.appendElement(dom, root, 'generator', 'The Dynastie project', {'uri':'http://indefero.soutade.fr/p/dynastie', 'version':'0.1'})
self.appendElement(dom, root, 'link', '', {'rel':'self', 'type':'application/atom+xml', 'href':address + '/atom.xml'})
2012-09-08 12:35:52 +02:00
posts = Post.objects.filter(published=True).order_by('-modification_date')[:10]
2012-08-28 09:09:14 +02:00
for post in posts:
item = dom.createElement('entry')
2012-08-28 09:09:14 +02:00
self.appendElement(dom, item, 'title', post.title)
path = 'http://' + blog.name + post.getPath()
self.appendElement(dom, item, 'link', '', {'rel':'alternate', 'type':'text/html', 'href':path})
2012-08-28 09:09:14 +02:00
creationDate = post.creation_date.strftime('%Y-%m-%dT%H:%M:%SZ')
self.appendElement(dom, item, 'published', creationDate)
2012-09-08 12:35:52 +02:00
modificationDate = post.modification_date.strftime('%Y-%m-%dT%H:%M:%SZ')
self.appendElement(dom, item, 'updated', modificationDate)
self.appendElement(dom, item, 'id', path)
author = dom.createElement('author')
2012-08-28 09:09:14 +02:00
self.appendElement(dom, author, 'name', post.author.first_name + ' ' + post.author.last_name)
self.appendElement(dom, author, 'email', post.author.email)
item.appendChild(author)
2012-08-28 09:09:14 +02:00
filename = blog.src_path + '/_post/' + str(post.id)
if not os.path.exists(filename):
self.addError('File does not exists ' + filename)
return
f = open(filename, 'rb')
2012-08-28 09:09:14 +02:00
post_content = '<![CDATA[' + f.read() + ']]>'
f.close()
2012-08-28 09:09:14 +02:00
self.appendElement(dom, item, 'summary', post_content, {'type':'html'})
self.appendElement(dom, item, 'content', post_content, {'type':'html'})
root.appendChild(item)
filename = 'atom.xml'
self.writeIfNotTheSame(output + '/' + filename, root)
if not self.somethingWrote:
self.addReport('Nothing changed')
return self.report