Keep only exe files in a directory

keep_exe.py

import argparse
import filetype
import os


def main():
    parser = argparse.ArgumentParser(description='Keep all exe files in a directory and remove the rest')
    parser.add_argument('directory')
    args = parser.parse_args()

    directory = args.directory

    for filename in os.listdir(directory):
        filepath = f'{directory}/{filename}'
        kind = filetype.guess(filepath)
        if kind is None or kind.extension != 'exe':
            os.remove(filepath)


if __name__ == '__main__':
    main()