Python - Find public keys matching a given list of patterns

find_public_keys.py

def contents(ssh, files):
    for file in files:
        path = '{}/{}'.format(ssh, file)
        yield open(path).read().replace("\n", "")

def keysFor(args):
    ssh = '{}/.ssh'.format(os.environ['HOME'])
    extensions = [ '.pub' ]
    files = [f for f in os.listdir(ssh) if os.path.splitext(f)[1] in extensions]
    keys = [k for k in contents(ssh, files) if k.split(' ')[2] in args]
    for key in keys:
        yield key

def main(args):
    keys = keysFor(args)
    for key in keys:
        print(key)

if __name__ == '__main__':
    main(sys.argv[1:])  

Call like this:

#!/bin/bash

./find_public_keys.py rgomes@example.com rgomes@localhost rgomes@laptop