Skip to content

Commit aa9290e

Browse files
committed
MachineScraper now uses regex escape function for robustness
1 parent 6c8c97d commit aa9290e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/scrape_images.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import re
23

34
from statemachine.contrib.diagram import DotGraphMachine
@@ -13,8 +14,8 @@ class MachineScraper:
1314

1415
def __init__(self, project_root):
1516
self.project_root = project_root
16-
sanitized_path = self.project_root.replace("\\", "\\\\")
17-
self.re_machine_module_name = re.compile(f"{sanitized_path}/(.*).py$")
17+
sanitized_path = re.escape(os.path.abspath(self.project_root))
18+
self.re_machine_module_name = re.compile(f"{sanitized_path}[\\/](.*)\\.py$")
1819
self.seen = set()
1920

2021
def __repr__(self):

0 commit comments

Comments
 (0)