adafruit-mirror/Adafruit_Python_Shell
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Allow finding multiple items

Browse source
This commit is contained in:
Melissa LeBlanc-Williams 2025-08-06 14:58:27 -07:00
parent 0a414a2342
commit 01be1e437d
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
adafruit_shell.py
View file

@ -384,7 +384,10 @@ class Shell:
# Not found; append (silently) # Not found; append (silently)
self.write_text_file(file, replacement, append=True) self.write_text_file(file, replacement, append=True)
def pattern_search(self, location, pattern, multi_line=False, return_match=False): # pylint: disable=too-many-arguments
def pattern_search(
self, location, pattern, multi_line=False, return_match=False, find_all=False
):
""" """
Similar to grep, but uses pure python Similar to grep, but uses pure python
multi_line will search the entire file as a large text glob, multi_line will search the entire file as a large text glob,
@ -394,16 +397,17 @@ class Shell:
""" """
location = self.path(location) location = self.path(location)
found = False found = False
search_function = re.findall if find_all else re.search
if self.exists(location) and not self.isdir(location): if self.exists(location) and not self.isdir(location):
if multi_line: if multi_line:
with open(location, "r+", encoding="utf-8") as file: with open(location, "r+", encoding="utf-8") as file:
match = re.search(pattern, file.read(), flags=re.DOTALL) match = search_function(pattern, file.read(), flags=re.DOTALL)
if match: if match:
found = True found = True
else: else:
for line in fileinput.FileInput(location): for line in fileinput.FileInput(location):
match = re.search(pattern, line) match = search_function(pattern, line)
if match: if match:
found = True found = True
break break