sherlock/tests/sherlock_interactives.py
Paul Pfeister 2b24cca340
Add tests
2024-05-31 14:46:46 -04:00

31 lines
1.2 KiB
Python

import os
import re
import subprocess
class Interactives:
def run_cli(args: str = "") -> str:
"""Pass arguments to Sherlock as a normal user on the command line"""
command = [f"sherlock {args}"]
proc_out: str = ""
try:
proc_out = subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT)
return proc_out.decode()
except subprocess.CalledProcessError as e:
raise InteractivesSubprocessError(e.output.decode())
def walk_sherlock_for_files_with(pattern: str) -> list[str]:
"""Check all files within the Sherlock package for matching patterns"""
pattern: re.Pattern = re.compile(pattern)
matching_files: list[str] = []
for root, dirs, files in os.walk("sherlock"):
for file in files:
file_path = os.path.join(root,file)
if "__pycache__" in file_path:
continue
with open(file_path, 'r', errors='ignore') as f:
if pattern.search(f.read()):
matching_files.append(file_path)
return matching_files
class InteractivesSubprocessError(Exception):
pass