import re, os

1. import re, os

2. from os.path import join, isfile

3. from fnmatch import fnmatch

4. def tree_iter(dir):

5. for root, dirs, files in os.walk(dir):

6. for f in files:

7. yield (root,f)

8. def dir_iter(dir):

9. for p in os.listdir(dir):

10. if isfile(join(dir,p)):

11. yield (dir,p)

12. def grep(

13. dir, # string, where to start the search

14. do_subdirs, # boolean, whether or not dwelve into subdirs

15. filemask, # string, a filemask like *.py

16. search_text, # string, the search expression

17. is_regex, # boolean, whether or not search_text is a regular expression

18. update_callback, # a function which takes a string of a filename

19. cancel_callback): # if this parameterless function returns True,

20. # the search is canceled.

21. """Calls update_callback for every file satisfying the search request specified

22. by the arguments."""

23. no_search= (search_text=="")

24. if is_regex:

25. search_pat= re.compile(search_text) # may throw

26. match_func= lambda s: search_pat.search(s) is not None

27. else:

28. match_func= lambda s: s.find(search_text) != -1

29. if do_subdirs:

30. iter= tree_iter

31. else:

32. iter= dir_iter

33. for (d,f) in iter(dir): # (dirname, filename)

34. if fnmatch(f, filemask):

35. if cancel_callback(): return

36. if no_search:

37. update_callback(join(d,f),-1)

38. continue # next f

39. try:

40. fp = open(join(d,f), 'r')

41. except IOError:

42. continue # next f

43. lineno = 0

44. for line in fp:

45. lineno += 1

46. if match_func(line):

47. update_callback(join(d,f), lineno)

48. break

49. fp.close()

50. if __name__=="__main__":