From 591478a2c008eef9a8777568867afe7c5fd416f1 Mon Sep 17 00:00:00 2001 From: Aleksey Lobanov Date: Sun, 21 Feb 2016 13:06:52 +0300 Subject: [PATCH] We build json with freqs of ids --- all_good_ids.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/all_good_ids.py b/all_good_ids.py index ddcca1f..0f937ff 100644 --- a/all_good_ids.py +++ b/all_good_ids.py @@ -2,17 +2,21 @@ # -*- coding: utf-8 -*- import sys +import json import os from os.path import join -all_ids = set() +all_ids = {} for dirpath, dirnames, filenames in os.walk(sys.argv[1]): for f in filenames: fp = os.path.join(dirpath, f) for uid in open(fp): - all_ids.add(int(uid)) + uid = int(uid) + if uid in all_ids: + all_ids[uid] += 1 + else: + all_ids[uid] = 1 f_out = open(sys.argv[2],'w') -for uid in all_ids: - f_out.write(str(uid) + '\n') +f_out.write(json.dumps(all_ids))