From 3a1c3d40dea13fdbc2ba6ce0635b70e271c62c4f Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Sat, 13 Aug 2022 16:55:52 +0200 Subject: [PATCH] Sanitize large-file filenames --- great_ai/large_file/large_file/large_file_base.py | 3 ++- 1 file changed, 2 insertions(+), 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 index 7671503..005f185 100644 --- 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