mirror of
https://github.com/adulau/rss-tools.git
synced 2024-05-19 23:02:10 +00:00
153 lines
3.7 KiB
Python
153 lines
3.7 KiB
Python
# rssdir.py
|
|
# a at foo dot be - Alexandre Dulaunoy - http://www.foo.be/cgi-bin/wiki.pl/RssAny
|
|
#
|
|
# rssdir is a simply-and-dirty script to rssify any directory on the filesystem.
|
|
#
|
|
# an example of use on the current directory :
|
|
#
|
|
# python3 /usr/local/bin/rssdir.py --prefix http://www.foo.be/cours/ . >rss.xml
|
|
#
|
|
|
|
import os, fnmatch
|
|
import time
|
|
import sys
|
|
import xml.etree.ElementTree as ET
|
|
from optparse import OptionParser
|
|
|
|
version = "0.2"
|
|
|
|
# recursive list file function from the ASPN cookbook
|
|
def all_files(root, patterns="*", single_level=False, yield_folders=False):
|
|
patterns = patterns.split(";")
|
|
for path, subdirs, files in os.walk(root):
|
|
if yield_folders:
|
|
files.extend(subdirs)
|
|
files.sort()
|
|
for name in files:
|
|
for pattern in patterns:
|
|
if fnmatch.fnmatch(name, pattern):
|
|
yield os.path.join(path, name)
|
|
break
|
|
if single_level:
|
|
break
|
|
|
|
|
|
def date_files(filelist):
|
|
date_filename_list = []
|
|
|
|
for filename in filelist:
|
|
stats = os.stat(filename)
|
|
last_update = stats[8]
|
|
date_filename_tuple = last_update, filename
|
|
date_filename_list.append(date_filename_tuple)
|
|
return date_filename_list
|
|
|
|
|
|
def date_as_rfc(value):
|
|
return time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.localtime(value))
|
|
|
|
|
|
def build_rss(myitem, maxitem):
|
|
|
|
RSSroot = ET.Element("rss", {"version": "2.0"})
|
|
RSSchannel = ET.SubElement(RSSroot, "channel")
|
|
|
|
ET.SubElement(RSSchannel, "title").text = "RSS feed of " + str(title)
|
|
ET.SubElement(RSSchannel, "link").text = link
|
|
ET.SubElement(RSSchannel, "description").text = (
|
|
"A directory RSSified by rssdir.py " + version
|
|
)
|
|
ET.SubElement(RSSchannel, "generator").text = (
|
|
"A directory RSSified by rssdir.py " + version
|
|
)
|
|
ET.SubElement(RSSchannel, "pubDate").text = date_as_rfc(time.time())
|
|
|
|
for bloodyitem in myitem[0:maxitem]:
|
|
|
|
RSSitem = ET.SubElement(RSSchannel, "item")
|
|
ET.SubElement(RSSitem, "title").text = bloodyitem[1]
|
|
ET.SubElement(RSSitem, "pubDate").text = date_as_rfc(bloodyitem[0])
|
|
ET.SubElement(RSSitem, "description").text = prefixurl + bloodyitem[1]
|
|
ET.SubElement(RSSitem, "guid").text = prefixurl + bloodyitem[1]
|
|
|
|
RSSfeed = ET.ElementTree(RSSroot)
|
|
feed = ET.tostring(RSSroot)
|
|
return feed
|
|
|
|
|
|
def complete_feed(myfeed):
|
|
|
|
myheader = '<?xml version="1.0"?>'
|
|
return myheader + str(myfeed)
|
|
|
|
|
|
usage = "usage: %prog [options] directory"
|
|
parser = OptionParser(usage)
|
|
|
|
parser.add_option(
|
|
"-p",
|
|
"--prefix",
|
|
dest="prefix",
|
|
default="",
|
|
help="http prefix to be used for each entry, default none",
|
|
)
|
|
parser.add_option(
|
|
"-t",
|
|
"--title",
|
|
dest="title",
|
|
help="set a title to the rss feed, default using prefix",
|
|
type="string",
|
|
)
|
|
parser.add_option(
|
|
"-l",
|
|
"--link",
|
|
dest="link",
|
|
help="http link set, default is prefix and none if prefix not set",
|
|
)
|
|
parser.add_option(
|
|
"-m",
|
|
"--maxitem",
|
|
dest="maxitem",
|
|
help="maximum item to list in the feed, default 32",
|
|
default=32,
|
|
type="int",
|
|
)
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
if options.prefix is None:
|
|
prefixurl = ""
|
|
else:
|
|
prefixurl = options.prefix
|
|
|
|
if options.link is None:
|
|
link = options.prefix
|
|
else:
|
|
link = options.link
|
|
|
|
if options.title is None:
|
|
title = options.prefix
|
|
else:
|
|
title = options.title
|
|
|
|
if options.maxitem is None:
|
|
maxitem = 32
|
|
else:
|
|
maxitem = options.maxitem
|
|
|
|
if not args:
|
|
print("Missing directory")
|
|
parser.print_help()
|
|
sys.exit(0)
|
|
|
|
file_to_list = []
|
|
for x in all_files(args[0]):
|
|
file_to_list.append(x)
|
|
|
|
mylist = date_files(file_to_list)
|
|
|
|
mylist.sort()
|
|
mylist.reverse()
|
|
|
|
print(complete_feed(build_rss(mylist, maxitem)))
|