pkraman06 commited on
Commit
060d5df
·
verified ·
1 Parent(s): 8dbf12b

Create utils/file_loader.py

Browse files
Files changed (1) hide show
  1. utils/file_loader.py +16 -0
utils/file_loader.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+
4
+ class FileLoader:
5
+ @staticmethod
6
+ def load_files(directory, extensions=None):
7
+ extensions = extensions or [".py"]
8
+
9
+ files = []
10
+
11
+ for root, _, filenames in os.walk(directory):
12
+ for filename in filenames:
13
+ if any(filename.endswith(ext) for ext in extensions):
14
+ files.append(os.path.join(root, filename))
15
+
16
+ return files