Python script to execute a shell command on every sub-folder of current folder

for-each-dir.py

#!/usr/bin/env python3
'''Receives a command as argument and executes it on 
every sub-folder of current folder.

Useful for commands that should be run multiple times,
like doing a git fetch on a bunch of repos.

Usage: ./for-each-dir.py "command" 
Example: ./for-each-dir.py "git fetch"
'''

import os
from subprocess import call

def execute_command(command):
    for folder in os.listdir():
        try:
            os.chdir(folder)
            print("\nTrying", folder + '/' + command)
            retcode = call(command, shell=True)
            os.chdir('..')
        except NotADirectoryError:
            pass

if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument('command', help='shell command to be executed on each folder')
    args = parser.parse_args() 
    
    execute_command(args.command)