import os from os import listdir from sys import argv import json filesList = filter(lambda x: x.endswith(argv[2]), listdir(argv[1])) filesDict = { i : { 'path': i } for i in filesList } print(json.dumps(filesDict))