python-feedgen/feedgen/__main__.py

# -*- coding: utf-8 -*-
'''
	feedgen
	~~~~~~~

	:copyright: 2013, Lars Kiesow <lkiesow@uos.de>

	:license: FreeBSD and LGPL, see license.* for more details.
'''

from feedgen.feed import FeedGenerator
import sys

def print_enc(s):
	'''Print function compatible with both python2 and python3 accepting strings
	and byte arrays.
	'''
	if sys.version_info[0] >= 3:
		print(s.decode('utf-8') if type(s) == type(b'') else s)
	else:
		print(s)



if __name__ == '__main__':
	if len(sys.argv) != 2 or not (
			sys.argv[1].endswith('rss') \
					or sys.argv[1].endswith('atom') \
                                        or sys.argv[1].endswith('torrent') \
					or sys.argv[1].endswith('podcast') ):
		print_enc ('Usage: %s ( <file>.atom | atom | <file>.rss | rss | podcast | torrent )' % \
				'python -m feedgen')
		print_enc ('')
		print_enc ('  atom             -- Generate ATOM test output and print it to stdout.')
		print_enc ('  rss              -- Generate RSS test output and print it to stdout.')
		print_enc ('  <file>.atom      -- Generate ATOM test feed and write it to file.atom.')
		print_enc ('  <file>.rss       -- Generate RSS test teed and write it to file.rss.')
		print_enc ('  podcast          -- Generate Podcast test output and print it to stdout.')
		print_enc ('  dc.atom          -- Generate DC extension test output (atom format) and print it to stdout.')
		print_enc ('  dc.rss           -- Generate DC extension test output (rss format) and print it to stdout.')
		print_enc ('  syndication.atom -- Generate DC extension test output (atom format) and print it to stdout.')
		print_enc ('  syndication.rss  -- Generate DC extension test output (rss format) and print it to stdout.')
		print_enc ('  torrent          -- Generate Torrent test output and print it to stdout.')
		print_enc ('')
		exit()

	arg = sys.argv[1]

	fg = FeedGenerator()
	fg.id('http://lernfunk.de/_MEDIAID_123')
	fg.title('Testfeed')
	fg.author( {'name':'Lars Kiesow','email':'lkiesow@uos.de'} )
	fg.link( href='http://example.com', rel='alternate' )
	fg.category(term='test')
	fg.contributor( name='Lars Kiesow', email='lkiesow@uos.de' )
	fg.contributor( name='John Doe', email='jdoe@example.com' )
	fg.icon('http://ex.com/icon.jpg')
	fg.logo('http://ex.com/logo.jpg')
	fg.rights('cc-by')
	fg.subtitle('This is a cool feed!')
	fg.link( href='http://larskiesow.de/test.atom', rel='self' )
	fg.language('de')
	fe = fg.add_entry()
	fe.id('http://lernfunk.de/_MEDIAID_123#1')
	fe.title('First Element')
	fe.content('''Lorem ipsum dolor sit amet, consectetur adipiscing elit. Tamen
			aberramus a proposito, et, ne longius, prorsus, inquam, Piso, si ista
			mala sunt, placet. Aut etiam, ut vestitum, sic sententiam habeas aliam
			domesticam, aliam forensem, ut in fronte ostentatio sit, intus veritas
			occultetur? Cum id fugiunt, re eadem defendunt, quae Peripatetici,
			verba.''')
	fe.summary(u'Lorem ipsum dolor sit amet, consectetur adipiscing elit…')
	fe.link( href='http://example.com', rel='alternate' )
	fe.author( name='Lars Kiesow', email='lkiesow@uos.de' )

	if arg == 'atom':
		print_enc (fg.atom_str(pretty=True))
	elif arg == 'rss':
		print_enc (fg.rss_str(pretty=True))
	elif arg == 'podcast':
		# Load the podcast extension. It will automatically be loaded for all
		# entries in the feed, too. Thus also for our “fe”.
		fg.load_extension('podcast')
		fg.podcast.itunes_author('Lars Kiesow')
		fg.podcast.itunes_category('Technology', 'Podcasting')
		fg.podcast.itunes_explicit('no')
		fg.podcast.itunes_complete('no')
		fg.podcast.itunes_new_feed_url('http://example.com/new-feed.rss')
		fg.podcast.itunes_owner('John Doe', 'john@example.com')
		fg.podcast.itunes_summary('Lorem ipsum dolor sit amet, ' + \
				'consectetur adipiscing elit. ' + \
				'Verba tu fingas et ea dicas, quae non sentias?')
		fe.podcast.itunes_author('Lars Kiesow')
		print_enc (fg.rss_str(pretty=True))
	
	elif arg == 'torrent':
		fg.load_extension('torrent')
		fe.link( href='http://somewhere.behind.the.sea/torrent/debian-8.4.0-i386-netint.iso.torrent', rel='alternate', type='application/x-bittorrent, length=1000' )
		fe.torrent.filename('debian-8.4.0-i386-netint.iso.torrent')
		fe.torrent.infohash('7661229811ef32014879ceedcdf4a48f256c88ba')
		fe.torrent.contentlength('331350016')
		fe.torrent.seeds('789')
		fe.torrent.peers('456')
		fe.torrent.verified('123')
		print_enc (fg.rss_str(pretty=True))

	elif arg.startswith('dc.'):
		fg.load_extension('dc')
		fg.dc.dc_contributor('Lars Kiesow')
		if arg.endswith('.atom'):
			print_enc (fg.atom_str(pretty=True))
		else:
			print_enc (fg.rss_str(pretty=True))

	elif arg.startswith('syndication'):
		fg.load_extension('syndication')
		fg.syndication.update_period('daily')
		fg.syndication.update_frequency(2)
		fg.syndication.update_base('2000-01-01T12:00+00:00')
		if arg.endswith('.rss'):
			print_enc (fg.rss_str(pretty=True))
		else:
			print_enc (fg.atom_str(pretty=True))

	elif arg.endswith('atom'):
		fg.atom_file(arg)

	elif arg.endswith('rss'):
		fg.rss_file(arg)