Thelie/python-feedgen
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

File output:

Browse source 
Added parameter for pretty print. Pass the file parameter directly to lxml.
This commit is contained in:
wltb 2014-04-21 22:37:05 +02:00
parent 8e06d8e41f
commit c2197dd368
1 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
feedgen/feed.py
View file

@ -204,16 +204,15 @@ class FeedGenerator(object):
return etree.tostring(feed, pretty_print=pretty) return etree.tostring(feed, pretty_print=pretty)
def atom_file(self, filename, extensions=True): def atom_file(self, filename, extensions=True, pretty=False):
'''Generates an ATOM feed and write the resulting XML to a file. '''Generates an ATOM feed and write the resulting XML to a file.
:param filename: Name of file to write. :param filename: Name of file to write, or a file-like object, or a URL.
:param extensions: Enable or disable the loaded extensions for the xml :param extensions: Enable or disable the loaded extensions for the xml
generation (default: enabled). generation (default: enabled).
''' '''
feed, doc = self._create_atom(extensions=extensions) feed, doc = self._create_atom(extensions=extensions)
with open(filename, 'w') as f: doc.write(filename, pretty_print=pretty)
doc.write(f)
def _create_rss(self, extensions=True): def _create_rss(self, extensions=True):
@ -357,16 +356,15 @@ class FeedGenerator(object):
return etree.tostring(feed, pretty_print=pretty) return etree.tostring(feed, pretty_print=pretty)
def rss_file(self, filename, extensions=True): def rss_file(self, filename, extensions=True, pretty=False):
'''Generates an RSS feed and write the resulting XML to a file. '''Generates an RSS feed and write the resulting XML to a file.
:param filename: Name of file to write. :param filename: Name of file to write, or a file-like object, or a URL.
:param extensions: Enable or disable the loaded extensions for the xml :param extensions: Enable or disable the loaded extensions for the xml
generation (default: enabled). generation (default: enabled).
''' '''
feed, doc = self._create_rss(extensions=extensions) feed, doc = self._create_rss(extensions=extensions)
with open(filename, 'w') as f: doc.write(filename, pretty_print=pretty)
doc.write(f)
def title(self, title=None): def title(self, title=None):