Sanitize large-file filenames

2022-08-13 16:55:52 +02:00 · 2022-08-13 16:55:52 +02:00 · 3a1c3d40de
commit 3a1c3d40de
parent 2f750c2478
1 changed files with 2 additions and 1 deletions
--- 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 @@ class LargeFileBase(ABC):
        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