From: Brett Parker Date: Wed, 19 Dec 2007 20:09:30 +0000 (+0000) Subject: * Parsing of the RSS feed using feedparser X-Git-Url: https://git.sommitrealweird.co.uk/rss2maildir.git/commitdiff_plain/d3fb153e25a909bee388d63448063fc0f7650732?hp=4abf90071a0d3d39f6a5829f79c980a195ed4c3b * Parsing of the RSS feed using feedparser * Creation of files for the maildir --- diff --git a/CHANGES b/CHANGES index 45e7187..db0c47d 100644 --- a/CHANGES +++ b/CHANGES @@ -3,3 +3,5 @@ * Initial starting points, config parser and option parsers * Creation of directories for maildir usage +* Initial creation of file per message (no collision detection or use of the + state directory yet) diff --git a/rss2maildir.py b/rss2maildir.py index 41654dd..a735d3a 100755 --- a/rss2maildir.py +++ b/rss2maildir.py @@ -6,9 +6,40 @@ import os import stat import urllib +import feedparser + +import email + +import datetime + from optparse import OptionParser from ConfigParser import SafeConfigParser + +def parse_and_deliver(maildir, url, statedir): + md = mailbox.Maildir(maildir) + fp = feedparser.parse(url) + for item in fp["items"]: + # things that we need in the message + msg = email.message_from_string("") + msg.add_header("Subject", item["title"]) + msg.set_unixfrom("Brett Parker ") + msg.add_header("Date", datetime.datetime(*item["created_parsed"][0:6]).strftime("%a, %e %b %Y %T -0000")) + msg.add_header("To", url) + msg.set_payload(item["content"][0]["value"]) + msg.set_charset("utf8") + msg.set_default_type("text/plain") + + # open a temporary file in the maildir + fn = os.tempnam(os.path.join(maildir, "tmp")) + fh = open(fn, "w") + fh.write(msg.as_string()) + fh.close() + # now move it in to the new directory + newfn = os.tempnam(os.path.join(maildir, "new")) + os.link(fn, newfn) + os.unlink(fn) + # first off, parse the command line arguments oparser = OptionParser() @@ -16,6 +47,10 @@ oparser.add_option( "-c", "--conf", dest="conf", help="location of config file" ) +oparser.add_option( + "-s", "--statedir", dest="statedir", + help="location of directory to store state in" + ) (options, args) = oparser.parse_args() @@ -52,6 +87,37 @@ scp = SafeConfigParser() scp.read(configfile) maildir_root = "RSSMaildir" +state_dir = "state" + +if options.statedir != None: + state_dir = options.statedir + try: + mode = os.stat(state_dir)[stat.ST_MODE] + if not stat.S_ISDIR(mode): + sys.stderr.write("State directory (%s) is not a directory\n" %(state_dir)) + sys.exit(1) + except: + # try to make the directory + try: + os.mkdir(state_dir) + except: + sys.stderr.write("Couldn't create statedir %s" %(state_dir)) + sys.exit(1) +elif scp.has_option("general", "state_dir"): + new_state_dir = scp.get("general", "state_dir") + try: + mode = os.stat(state_dir)[stat.ST_MODE] + if not stat.S_ISDIR(mode): + sys.stderr.write("State directory (%s) is not a directory\n" %(state_dir)) + sys.exit(1) + except: + # try to create it + try: + os.mkdir(new_state_dir) + state_dir = new_state_dir + except: + sys.stderr.write("Couldn't create state directory %s\n" %(new_state_dir)) + sys.exit(1) if scp.has_option("general", "maildir_root"): maildir_root = scp.get("general", "maildir_root") @@ -94,23 +160,36 @@ for section in feeds: except: os.mkdir(os.path.join(maildir, "cur")) if not stat.S_ISDIR(mode): - sys.stderr.write("Broken maildir: %s" %(maildir)) + sys.stderr.write("Broken maildir: %s\n" %(maildir)) try: mode = os.stat(os.path.join(maildir, "tmp"))[stat.ST_MODE] except: os.mkdir(os.path.join(maildir, "tmp")) if not stat.S_ISDIR(mode): - sys.stderr.write("Broken maildir: %s" %(maildir)) + sys.stderr.write("Broken maildir: %s\n" %(maildir)) try: mode = os.stat(os.path.join(maildir, "new"))[stat.ST_MODE] if not stat.S_ISDIR(mode): - sys.stderr.write("Broken maildir: %s" %(maildir)) + sys.stderr.write("Broken maildir: %s\n" %(maildir)) except: os.mkdir(os.path.join(maildir, "new")) else: - sys.stderr.write("Broken maildir: %s" %(maildir)) + sys.stderr.write("Broken maildir: %s\n" %(maildir)) except: - os.mkdir(maildir) - os.mkdir(os.path.join(maildir, "new")) - os.mkdir(os.path.join(maildir, "cur")) - os.mkdir(os.path.join(maildir, "tmp")) + try: + os.mkdir(maildir) + except: + sys.stderr.write("Couldn't create root maildir %s\n" %(maildir)) + sys.exit(1) + try: + os.mkdir(os.path.join(maildir, "new")) + os.mkdir(os.path.join(maildir, "cur")) + os.mkdir(os.path.join(maildir, "tmp")) + except: + sys.stderr.write("Couldn't create required maildir directories for %s\n" %(section,)) + sys.exit(1) + + # right - we've got the directories, we've got the section, we know the + # url... lets play! + + parse_and_deliver(maildir, section, state_dir)