regex match on process_analysis rather than os.basepath
Taylor committed
Jun 28, 2024 at 10:31 UTC
ac779c2eecf082210b896fe842336baa97f24eae
1 file changed
+3
-2
backend/app/threat_intel/schema/socfortress.py
+3
-2
@@ -1,4 +1,4 @@
1
-import os
1
+import re
2
from typing import List
3
from typing import Optional
4
@@ -58,7 +58,8 @@ class SocfortressProcessNameAnalysisRequest(BaseModel):
58
59
@validator("process_name", pre=True)
60
def extract_filename(cls, v):
61
- return os.path.basename(v)
61
+ match = re.search(r"[^\\]+$", v)
62
+ return match.group() if match else v
63
64
65
class Path(BaseModel):