fix: Avoid crash when file in repo contains non-unicode characters (#59)

This fix prevents a crash when using `--clean` on a repo where a file contains non-unicode characters. Fix seems benign since any file with non-unicode characters likely wasnt created by this tool. Closes #58 --------- Co-authored-by: Bradley Dice <[email protected]>
difyrrwrzd · Dec 13, 2023 · f178845 · f178845
1 parent ae0fc61
commit f178845
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/rapids_dependency_file_generator/rapids_dependency_file_generator.py b/src/rapids_dependency_file_generator/rapids_dependency_file_generator.py
@@ -36,8 +36,11 @@ def delete_existing_files(root="."):
             lambda fn: fn.endswith(".txt") or fn.endswith(".yaml"), filenames
         ):
             with open(file_path := os.path.join(dirpath, fn)) as f:
-                if HEADER in f.read():
-                    os.remove(file_path)
+                try:
+                    if HEADER in f.read():
+                        os.remove(file_path)
+                except UnicodeDecodeError:
+                    pass
 
 
 def dedupe(dependencies):