Sanitize large-file filenames

schmelczer · Aug 13, 2022 · 3a1c3d4 · radl97 · Aug 13, 2022 · 3a1c3d4
1 parent 2f750c2
commit 3a1c3d4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/great_ai/large_file/large_file/large_file_base.py b/great_ai/large_file/large_file/large_file_base.py
@@ -1,4 +1,5 @@
 import os
+import re
 import shutil
 import sys
 import tempfile
@@ -63,7 +64,7 @@ def __init__(
         keep_last_n: Optional[int] = None,
         cache_only_mode: bool = False,
     ):
-        self._name = name
+        self._name = re.sub(r"[^a-zA-Z0-9._-]+", "", name)
         self._version = version
         self._mode = mode
         self._keep_last_n = keep_last_n