SGJP / Kuznia

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys


def get_lines(words, reverse):
    first = True
    cut = False
    lines = []
    for line in sys.stdin:
        if first:
            word = line.split('\t')[0].decode('utf-8')
            cut = (word in words) != reverse
            first = False
        if cut:
            lines.append(line)
        if line == '\n':
            first = True
    return lines


if __name__ == '__main__':
    reverse = len(sys.argv) > 2 and sys.argv[2] == '-r'
    with open(sys.argv[1]) as words_file:
        words = [line.strip().decode('utf-8') for line in words_file]
    lines = get_lines(words, reverse)
    sys.stdout.write(''.join(lines))