PushshiftDumps/personal/count_by_subreddit.py
2023-01-30 17:05:22 -08:00

24 lines
768 B
Python

import utils
import discord_logging
import os
from collections import defaultdict
log = discord_logging.init_logging()
if __name__ == "__main__":
subreddits = defaultdict(int)
input_file = r"\\MYCLOUDPR4100\Public\pushshift_working\RC_2022-12.zst"
input_file_size = os.stat(input_file).st_size
total_lines = 0
for comment, line, file_bytes_processed in utils.read_obj_zst_meta(input_file):
subreddits[comment['subreddit']] += 1
total_lines += 1
if total_lines % 100000 == 0:
log.info(f"{total_lines:,} lines, {(file_bytes_processed / input_file_size) * 100:.0f}%")
log.info(f"{total_lines:,} lines, 100%")
for subreddit, count in sorted(subreddits.items(), key=lambda item: item[1] * -1):
if count >= 1:
log.info(f"r/{subreddit}: {count:,}")