view master/src/wecu/sac_mapper.py @ 60:5fdca5baa4e9

refactor a bit, add support for sac with bespoke mapper
author Henry S. Thompson <ht@markup.co.uk>
date Thu, 28 May 2020 12:55:03 +0000
parents a3edba8dab11
children cfaf5223b071
line wrap: on
line source

#!/usr/bin/python3
import os
import sys
import re

print('args',sys.argv)

is_regex = sys.argv[1] == 'true'
search_terms = sys.argv[2:]
search_terms_counters = dict()

# Initialise
for term in search_terms:
    search_terms_counters[term] = 0

if not is_regex:
    for line in sys.stdin:
        for term in search_terms_counters:
            search_terms_counters[term] += line.count(term)
else:
    for line in sys.stdin:
        for term_regex in search_terms_counters:
            search_terms_counters[term_regex] += len(re.findall(term_regex, line))


for term in search_terms_counters:
    print("{}\t{}".format(term, search_terms_counters[term]))