|
|
"""This function implements a simple grep-like function in pure python.
|
|
|
|
|
|
You can enable it by copying it to your ~/.ipython directory and putting
|
|
|
|
|
|
execfile = magic_grepl.py
|
|
|
|
|
|
in your ipythonrc file.
|
|
|
|
|
|
Code contributed by Gever Tulley <gever@helium.com>, minor changes applied.
|
|
|
"""
|
|
|
|
|
|
import glob
|
|
|
import re
|
|
|
import os
|
|
|
|
|
|
def magic_grepl(self, parameter_s=''):
|
|
|
"""Search for a pattern in a list of files.
|
|
|
|
|
|
It prints the names of the files containing the pattern. Similar to 'grep
|
|
|
-l' in Unix-like environments.
|
|
|
|
|
|
Usage: @grepl pattern [files]
|
|
|
|
|
|
- pattern: any regular expression pattern which re.compile() will accept.
|
|
|
- files: list of files to scan. It can contain standard unix wildcards.
|
|
|
"""
|
|
|
|
|
|
# argument processing
|
|
|
params = parameter_s.split()
|
|
|
if len(params) > 1:
|
|
|
target = params[0] # first one is the target
|
|
|
file_patterns = params[1:] # all the rest are filenames or patterns
|
|
|
|
|
|
# build the regular expression
|
|
|
expr = re.compile(target)
|
|
|
|
|
|
for pattern in file_patterns:
|
|
|
flist = [f for f in glob.glob(pattern) if os.path.isfile(f)]
|
|
|
for filename in flist:
|
|
|
# open and read the whole file
|
|
|
f = open(filename,'r')
|
|
|
data = f.read()
|
|
|
f.close()
|
|
|
|
|
|
# see if pattern occurs in the file
|
|
|
if expr.search(data):
|
|
|
print filename
|
|
|
else:
|
|
|
# no parameters given
|
|
|
print("Usage: @grepl pattern [files]");
|
|
|
|
|
|
# Add the new magic function to the class dict:
|
|
|
from IPython.iplib import InteractiveShell
|
|
|
InteractiveShell.magic_grepl = magic_grepl
|
|
|
|
|
|
# And remove the global name to keep global namespace clean. Don't worry, the
|
|
|
# copy bound to IPython stays, we're just removing the global name.
|
|
|
del magic_grepl
|
|
|
|
|
|
#********************** End of file <magic_grepl.py> ***********************
|
|
|
|
