chg: [rssdir] Make rssdir work in the modern world - move to Python 3

This commit is contained in:
Alexandre Dulaunoy 2024-02-11 13:15:23 +01:00
parent 2240a563d2
commit 93ced8398e
Signed by: adulau
GPG key ID: 09E2CD4944E6CBCD

View file

@ -2,12 +2,11 @@
# 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 :
#
# python2.5 /usr/local/bin/rssdir.py --prefix http://www.foo.be/cours/ . >rss.xml
# python3 /usr/local/bin/rssdir.py --prefix http://www.foo.be/cours/ . >rss.xml
#
# Don't really need python2.5 except for ElementTree but you are free to install it.
import os, fnmatch
import time
@ -15,103 +14,138 @@ import sys
import xml.etree.ElementTree as ET
from optparse import OptionParser
version = "0.1"
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 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 = []
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)
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
return date_filename_list
def date_as_rfc(value):
return time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.localtime(value))
return time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.localtime(value))
def build_rss(myitem,maxitem):
def build_rss(myitem, maxitem):
RSSroot = ET.Element( 'rss', {'version':'2.0'} )
RSSchannel = ET.SubElement( RSSroot, 'channel' )
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())
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]:
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
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)
myheader = '<?xml version="1.0"?>'
return myheader + str(myfeed)
usage = "usage: %prog [options] directory"
parser = OptionParser(usage)
parser.add_option("-p","--prefix",dest="prefix",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",type="int")
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
prefixurl = ""
else:
prefixurl = options.prefix
if options.link is None:
link = options.prefix
else :
link = options.link
link = options.prefix
else:
link = options.link
if options.title is None:
title = options.prefix
else :
title = options.title
title = options.prefix
else:
title = options.title
if options.maxitem is None:
maxitem = 32
else :
maxitem = options.maxitem
maxitem = 32
else:
maxitem = options.maxitem
if not args:
parser.print_help()
sys.exit(0)
mylist = date_files(all_files(args[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))
print(complete_feed(build_rss(mylist, maxitem)))