X-Git-Url: https://code.communitydata.science/cdsc_reddit.git/blobdiff_plain/6dca79a41fbd5d1e967ba2cbd12da55310e5858f..34185337c9c6f073ce8f8cb0e1b7ea605861b6e8:/comments_2_parquet.py

diff --git a/comments_2_parquet.py b/comments_2_parquet.py
index 3042f58..bd853f7 100755
--- a/comments_2_parquet.py
+++ b/comments_2_parquet.py
@@ -128,5 +128,8 @@ df = df.withColumn("Month",f.month(f.col("CreatedAt")))
 df = df.withColumn("Year",f.year(f.col("CreatedAt")))
 df = df.withColumn("Day",f.dayofmonth(f.col("CreatedAt")))
 df = df.withColumn("subreddit_hash",f.sha2(f.col("subreddit"), 256)[0:3])
-df = df.sort(["subreddit","author","link_id","parent_id","Year","Month","Day"],ascending=True)
-df.write.parquet("/gscratch/comdata/output/reddit_comments.parquet", partitionBy=["Year",'Month'],mode='overwrite')
+df2 = df.sort(["subreddit","author","link_id","parent_id","Year","Month","Day"],ascending=True)
+df2.write.parquet("/gscratch/comdata/output/reddit_comments_by_subreddit.parquet", partitionBy=["Year",'Month'],mode='overwrite')
+
+df3 = df.sort(["author","CreatetdAt","subreddit","link_id","parent_id","Year","Month","Day"],ascending=True)
+df3.write.parquet("/gscratch/comdata/output/reddit_comments_by_author.parquet", partitionBy=["Year",'Month'],mode='overwrite')