refactor: only iterate header files.

2025-01-22 18:43:36 +08:00
parent 5a2577c3c6
commit c17dba1823
1 changed files with 5 additions and 1 deletions
--- a/DeThunk/src/main.py
+++ b/DeThunk/src/main.py
@@ -78,13 +78,17 @@ def remove_thunks(path_to_file: str):


 def iterate_headers(target_path: str):
+    def is_header(path: str):
+        return path.endswith('.h') or path.endswith('.hpp')
+
    if not os.path.isdir(target_path):
        print('invalid path.')
        return

    for root, dirs, files in os.walk(target_path):
        for file in files:
-            remove_thunks(os.path.join(root, file))
+            if is_header(file):
+                remove_thunks(os.path.join(root, file))


 def main():