Skip to content

Commit f175f48

Browse files
committed
Add a more strict check for the managed folder id
1 parent 4af4d77 commit f175f48

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dataikuapi/dss_plugin_mlflow/artifact_repository.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,16 @@
22
import posixpath
33
import tempfile
44
import urllib
5+
import re
56
from dataikuapi import DSSClient
67

78

89
def parse_dss_managed_folder_uri(uri):
910
parsed = urllib.parse.urlparse(uri)
1011
if parsed.scheme != "dss-managed-folder":
1112
raise Exception("Not a DSS Managed Folder URI: %s" % uri)
12-
if not parsed.netloc or parsed.netloc == '.':
13+
pattern = re.compile("^(\w+\.)?\w{8}")
14+
if not parsed.netloc or not pattern.match(parsed.netloc):
1315
raise Exception("Could not find a managed folder id in URI: %s" % uri)
1416
return parsed
1517

0 commit comments

Comments
 (0)