1 files changed, 33 insertions, 34 deletions

diff --git a/site_builder/feeds.py b/site_builder/feeds.py
index 3279afb..dd343b5 100644
--- a/site_builder/feeds.py
+++ b/site_builder/feeds.py
@@ -6,14 +6,13 @@ Atom Feed Writer
 @date: June 04, 2010
 """
 
-#from io import StringIO
-from StringIO import StringIO
+from io import StringIO
 from xml.sax.saxutils import XMLGenerator
 
 
 class SimpleXMLGenerator(XMLGenerator):
 
-    def __init__(self, encoding='utf-8'):
+    def __init__(self, encoding="utf-8"):
         self.output = StringIO()
         XMLGenerator.__init__(self, out=self.output, encoding=encoding)
 
@@ -53,8 +52,8 @@ class Atom1Feed(object):
 
     def get_feed(self):
         self.handler.startDocument()
-        self.handler.startElement('feed', {
-            'xmlns': 'http://www.w3.org/2005/Atom' })
+        self.handler.startElement("feed", {
+            "xmlns": "http://www.w3.org/2005/Atom" })
 
         self.add_root_elements()
 
@@ -64,47 +63,47 @@ class Atom1Feed(object):
 
             self.add_post(post)
 
-        self.handler.endElement('feed')
+        self.handler.endElement("feed")
 
         return self.handler.get_contents()
 
     def add_root_elements(self):
-        self.handler.addElement('title', self.title)
-        self.handler.addElement('updated', self._format_time(self.updated))
-        self.handler.addElement('id', self.feed_url)
-        self.handler.addElement('link', attrs={
-            'rel': 'alternate',
-            'type': 'text/html',
-            'href': self.blog_url })
-        self.handler.addElement('link', attrs={
-            'rel': 'self',
-            'type': 'application/atom+xml',
-            'href': self.feed_url })
+        self.handler.addElement("title", self.title)
+        self.handler.addElement("updated", self._format_time(self.updated))
+        self.handler.addElement("id", self.feed_url)
+        self.handler.addElement("link", attrs={
+            "rel": "alternate",
+            "type": "text/html",
+            "href": self.blog_url })
+        self.handler.addElement("link", attrs={
+            "rel": "self",
+            "type": "application/atom+xml",
+            "href": self.feed_url })
 
     def add_post(self, post):
         handler = self.handler
 
-        handler.startElement('entry')
+        handler.startElement("entry")
 
-        handler.startElement('author')
-        handler.addElement('name', post.author.name)
-        handler.addElement('email', post.author.email)
-        handler.endElement('author')
+        handler.startElement("author")
+        handler.addElement("name", post.author.name)
+        handler.addElement("email", post.author.email)
+        handler.endElement("author")
 
-        post_href = '{0}/{1}'.format(self.blog_url, post.filename)
+        post_href = "{0}/{1}".format(self.blog_url, post.filename)
 
-        handler.addElement('title', post.title)
-        handler.addElement('link', attrs={
-            'rel': 'alternate',
-            'type': 'text/html',
-            'href': post_href })
-        handler.addElement('id', post_href)
-        handler.addElement('updated', self._format_time(post.post_date))
-        handler.addElement('published', self._format_time(post.post_date))
+        handler.addElement("title", post.title)
+        handler.addElement("link", attrs={
+            "rel": "alternate",
+            "type": "text/html",
+            "href": post_href })
+        handler.addElement("id", post_href)
+        handler.addElement("updated", self._format_time(post.post_date))
+        handler.addElement("published", self._format_time(post.post_date))
 
         for tag in post.tags:
-            handler.addElement('category', attrs={ 'term': tag })
+            handler.addElement("category", attrs={ "term": tag })
 
-        handler.addElement('content', post.contents, attrs={ 'type': 'html' })
+        handler.addElement("content", post.contents, attrs={ "type": "html" })
 
-        handler.endElement('entry')
+        handler.endElement("entry")