diff --git a/test_get_files.py b/test_get_files.py new file mode 100644 index 0000000..cde177c --- /dev/null +++ b/test_get_files.py @@ -0,0 +1,24 @@ +import os + + +def read_files(path, output, filetypes=None): + print(os.path.join(path)) + if filetypes is None: + filetypes = ["pdf", "txt", "png", "jpg", "mp3"] + + for root, dirs, files in os.walk(path): + for file in files: + if (type := file.split(".")[-1]) in filetypes: + output.append((type, os.path.join(path, file))) + + for dir in dirs: + read_files(os.path.join(path, dir), output) + + break + + + +output = [] +read_files("C:\\SoftwareEng", output) + +print(output) \ No newline at end of file