X-Git-Url: https://git.sommitrealweird.co.uk/rss2maildir.git/blobdiff_plain/63dc58ea5dc82d4ab0984101798fd8a813187450..e4e1e5427c6afcedb1876b15694e5f473272e164:/rss2maildir.py diff --git a/rss2maildir.py b/rss2maildir.py index cece696..968c647 100755 --- a/rss2maildir.py +++ b/rss2maildir.py @@ -3,10 +3,55 @@ import mailbox import sys import os +import stat +import urllib + +import feedparser + +import email + +import datetime +import random +import string + +import socket from optparse import OptionParser from ConfigParser import SafeConfigParser +from base64 import b64encode + + +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.set_unixfrom("\"%s\" " %(url)) + msg.add_header("From", "\"%s\" " %(item["author"])) + msg.add_header("To", "\"%s\" " %(url)) + msg.add_header("Date", datetime.datetime(*item["created_parsed"][0:6]).strftime("%a, %e %b %Y %T -0000")) + msg.add_header("Subject", item["title"]) + msg.set_charset("utf8") + msg.set_default_type(item["content"][0]["type"]) + msg.set_type(item["content"][0]["type"]) + msg.set_payload(b64encode(item["content"][0]["value"])) + + msg.add_header("Message-ID", "<" + datetime.datetime.now().strftime("%Y%m%d%H%M") + "." + "".join([random.choice(string.ascii_letters + string.digits) for a in range(0,6)]) + "@" + socket.gethostname() + ">") + + # start by working out the filename we should be writting to, we do + # this following the normal maildir style rules + fname = str(os.getpid()) + "." + socket.gethostname() + "." + "".join([random.choice(string.ascii_letters + string.digits) for a in range(0,10)]) + "." + datetime.datetime.now().strftime('%s') + fn = os.path.join(maildir, "tmp", fname) + fh = open(fn, "w") + fh.write(msg.as_string()) + fh.close() + # now move it in to the new directory + newfn = os.path.join(maildir, "new", fname) + os.link(fn, newfn) + os.unlink(fn) + # first off, parse the command line arguments oparser = OptionParser() @@ -14,6 +59,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() @@ -50,10 +99,53 @@ 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") +try: + mode = os.stat(maildir_root)[stat.ST_MODE] + if not stat.S_ISDIR(mode): + sys.stderr.write("Maildir Root %s is not a directory\n" %(maildir_root)) + sys.exit(1) +except: + try: + os.mkdir(maildir_root) + except: + sys.stderr.write("Couldn't create Maildir Root %s\n" %(maildir_root)) + sys.exit(1) + feeds = scp.sections() try: feeds.remove("general") @@ -61,6 +153,55 @@ except: pass for section in feeds: - print section - print "-" * len(section) - print scp.items(section) + # check if the directory exists + maildir = None + try: + maildir = scp.get(section, "maildir") + except: + maildir = section + + maildir = urllib.urlencode(((section, maildir),)).split("=")[1] + maildir = os.path.join(maildir_root, maildir) + + try: + exists = os.stat(maildir) + if stat.S_ISDIR(exists[stat.ST_MODE]): + # check if there's a new, cur and tmp directory + try: + mode = os.stat(os.path.join(maildir, "cur"))[stat.ST_MODE] + except: + os.mkdir(os.path.join(maildir, "cur")) + if not stat.S_ISDIR(mode): + 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\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\n" %(maildir)) + except: + os.mkdir(os.path.join(maildir, "new")) + else: + sys.stderr.write("Broken maildir: %s\n" %(maildir)) + except: + 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)