filter file by keys

filter_file_by_keys.py

#!/usr/bin/env python
# coding=utf-8
import sys


def go(key_filename, filename):
    k = open(key_filename)
    keys = list()
    while 1:
        lines = k.readlines(100000)
        if not lines:
            break
        for line in lines:
            val = line.strip()
            if len(val) != 36:
                continue
            keys.append(val)
    k.close()
    #print('loaded keys: {0}'.format(len(keys)))

    f = open(filename)
    while 1:
        lines = f.readlines(100000)
        if not lines:
            break
        for line in lines:
            val = line.strip()
            if val in keys:
                print(val)
    f.close()


def main():
    if len(sys.argv) < 3:
        print('usage: ' + sys.argv[0] + ' [keys filename] [filename]')
        return
    go(sys.argv[1], sys.argv[2])

if __name__ == '__main__':
    main()

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4