diff tools/filter_posts.py @ 334:6805d15cda13

Adding a script I had to write on the fly to filter out posts from the posts csv file that had no parent topics. MyISAM let me get away with that, but InnoDB won't.
author Brian Neal <bgneal@gmail.com>
date Sat, 26 Feb 2011 01:28:22 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tools/filter_posts.py	Sat Feb 26 01:28:22 2011 +0000
@@ -0,0 +1,59 @@
+"""
+filter_posts.py - A script to filter out posts that have no parent topic in the
+new database. 
+
+"""
+from __future__ import with_statement
+import csv
+import optparse
+import sys
+
+
+USAGE = "usage: %prog [options]"
+DESCRIPTION = """Filters out posts that have no parent topic."""
+
+
+def main(argv=None):
+
+    parser = optparse.OptionParser(usage=USAGE, description=DESCRIPTION)
+    parser.set_defaults(
+        topic_file='forums_topic.csv',
+        post_file='forums_post.csv',
+        output_file='forums_post_filtered.csv',
+    )
+    parser.add_option("-s", "--progress", action="store_true",
+        help="Output a . after every 100 posts to show progress [default: %default]")
+    parser.add_option("-t", "--topic-file",
+        help="Name of the topics csv file [default: %default]")
+    parser.add_option("-p", "--post-file",
+        help="Name of the posts csv file [default: %default]")
+    parser.add_option("-o", "--output-file",
+        help="Name of the output posts csv file [default: %default]")
+
+    opts, args = parser.parse_args(args=argv)
+
+    topics = set()
+    with open(opts.topic_file, "rb") as topic_file:
+        reader = csv.reader(topic_file)
+        for row in reader:
+            topics.add(int(row[0]))
+
+    with open(opts.post_file, "rb") as post_file:
+        reader = csv.reader(post_file)
+        # skip first row
+        print reader.next()
+        with open(opts.output_file, "wb") as output_file:
+            writer = csv.writer(output_file)
+            for row in reader:
+                topic = int(row[1])
+                if topic in topics:
+                    writer.writerow(row)
+                    
+
+if __name__ == '__main__':
+    try:
+        main()
+    except IOError, ex:
+        sys.exit("IO Error: %s" % ex)
+    except KeyboardInterrupt:
+        sys.exit("Control-C interrupt")