We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9cd0c91 commit 5f3bbebCopy full SHA for 5f3bbeb
python/paddle/dataset/common.py
@@ -18,6 +18,7 @@
18
import importlib
19
import os
20
import pickle
21
+import re
22
import shutil
23
import sys
24
import tempfile
@@ -71,6 +72,11 @@ def md5file(fname):
71
72
73
74
def download(url, module_name, md5sum, save_name=None):
75
+ module_name = re.match("^[a-zA-Z0-9_/\\-]+$", module_name).group()
76
+ if isinstance(save_name, str):
77
+ save_name = re.match(
78
+ "^(?:(?!\\.\\.)[a-zA-Z0-9_/\\.-])+$", save_name
79
+ ).group()
80
dirname = os.path.join(DATA_HOME, module_name)
81
if not os.path.exists(dirname):
82
os.makedirs(dirname)
0 commit comments