easyctf-2017/count.py

34 lines
1,016 B
Python
Raw Normal View History

2017-03-03 02:33:16 +00:00
#!/usr/bin/env python
import os
import yaml
import traceback
from collections import Counter
problem_names = os.listdir(os.path.dirname(os.path.abspath(__file__)))
problems = []
for problem_name in problem_names:
try:
metadata_file = os.path.dirname(os.path.abspath(__file__)) + os.sep + problem_name + os.sep + "problem.yml"
with open(metadata_file, "r") as f:
metadata_raw = f.read()
metadata = yaml.load(metadata_raw)
if "category" in metadata:
problems.append(metadata)
except:
pass
# print traceback.format_exc()
2017-03-06 07:58:08 +00:00
problems.sort(key=lambda p: p.get("value"), reverse=True)
2017-03-03 02:33:16 +00:00
print "Grand Total: %d" % len(problems)
print "Category Breakdown:"
c = Counter(map(lambda p: p.get("category", ""), problems))
categories = sorted(c.items(), key=lambda c: c[1], reverse=True)
for category, count in categories:
print " %s: %s" % (category, count)
for problem in problems:
if problem.get("category") != category: continue
2017-03-06 07:58:08 +00:00
print " %s [%s]" % (problem.get("title"), problem.get("value"))