infra/generate_skels.py: Support Python-style comments

Modifies copies of `generate_skels.py` to support `.py` files. Signed-off-by: NickZaharia308 <[email protected]>
cs-pub-ro · Oct 27, 2024 · 08dcf1d · 08dcf1d
1 parent 5a79c21
commit 08dcf1d
Show file tree

Hide file tree

Showing 10 changed files with 10 additions and 0 deletions.
diff --git a/chapters/data/process-memory/drills/tasks/copy/generate_skels.py b/chapters/data/process-memory/drills/tasks/copy/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/data/working-with-memory/drills/tasks/memory-access/generate_skels.py b/chapters/data/working-with-memory/drills/tasks/memory-access/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/data/working-with-memory/drills/tasks/memory-corruption/generate_skels.py b/chapters/data/working-with-memory/drills/tasks/memory-corruption/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/data/working-with-memory/drills/tasks/memory-protection/generate_skels.py b/chapters/data/working-with-memory/drills/tasks/memory-protection/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/software-stack/applications/drills/tasks/app-investigation/generate_skels.py b/chapters/software-stack/applications/drills/tasks/app-investigation/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/software-stack/high-level-languages/drills/tasks/high-level-lang/generate_skels.py b/chapters/software-stack/high-level-languages/drills/tasks/high-level-lang/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/software-stack/libc/drills/tasks/common-functions/generate_skels.py b/chapters/software-stack/libc/drills/tasks/common-functions/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/software-stack/libc/drills/tasks/libc/generate_skels.py b/chapters/software-stack/libc/drills/tasks/libc/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/software-stack/system-calls/drills/tasks/basic-syscall/generate_skels.py b/chapters/software-stack/system-calls/drills/tasks/basic-syscall/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"

diff --git a/chapters/software-stack/system-calls/drills/tasks/syscall-wrapper/generate_skels.py b/chapters/software-stack/system-calls/drills/tasks/syscall-wrapper/generate_skels.py
@@ -109,6 +109,7 @@ def main():
                 re.match("Makefile.*$", src)
                 or re.match(r".*\.sh$", src)
                 or re.match(r".*\.[sS]$", src)
+                or re.match(r".*\.py$", src)
             ):
                 pattern = r"(^\s*#\s*TODO)( [0-9]*)(:.*)"
                 replace = r"(^\s*#\s*REPLACE)( [0-9]*)"