view python/lsexts.py @ 756:12f0be02e9cc

http://www.howtogeek.com/168147/add-public-ssh-key-to-remote-server-in-a-single-command/
author Jeff Hammel <k0scist@gmail.com>
date Wed, 05 Aug 2015 08:44:23 -0700
parents 5f186e65da09
children
line wrap: on
line source

#!/usr/bin/env python

import os

def extensions(*filenames):
    return set([os.path.splitext(f)[-1] for f in filenames
                if '.' in f])

def find_extensions(*directories):
    _extensions = set()
    for directory in directories:
        for _, _, fnames in os.walk(directory):
            _extensions.update(extensions(*fnames))
    return _extensions

if __name__ == '__main__':
    import sys
    for ext in sorted(find_extensions(*(sys.argv[1:] or [os.getcwd()]))):
        print ext