-
Notifications
You must be signed in to change notification settings - Fork 0
/
filesearch.py
91 lines (73 loc) · 2.09 KB
/
filesearch.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#!/usr/local/bin/python3
import sys
import os
SearchDir = "/"
SearchExt = ".mp3" # Default extension.
numargs = len(sys.argv)
if numargs > 3:
print("Error - too many arguments supplied.")
sys.exit()
elif numargs == 2:
if sys.argv[1][0] == ".":
SearchExt = sys.argv[1]
else:
if os.path.isdir(sys.argv[1]):
SearchDir = sys.argv[1]
else:
print("Error - name is not a valid directory")
sys.exit()
elif numargs == 3:
if sys.argv[1][0] == ".":
SearchExt = sys.argv[1]
if os.path.isdir(sys.argv[2]):
SearchDir = sys.argv[2]
else :
print("Error - name is not a valid directory")
sys.exit()
else :
if os.path.isdir(sys.argv[1]):
SearchDir = sys.argv[1]
else :
print("Error - name is not a valid directory")
sys.exit()
if sys.argv[2][0] == ".":
SearchExt = sys.argv[2]
else :
print("Error - search extension does not contain a period.")
sys.exit()
def findfile (searchdir, searchext):
print("Searchdir = ", searchdir, "Searchext = ", searchext)
foundfiles = []
contents = os.listdir(searchdir)
# print(contents)
for name in contents :
if os.path.isfile(os.path.join(searchdir, name)):
# print("found a file", name)
if name.endswith(searchext):
foundfiles.append(os.path.join(searchdir, name))
elif os.path.isdir(os.path.join(searchdir, name)):
# print("found a directory", name)
tempfiles = findfile(os.path.join(searchdir, name), searchext)
foundfiles.extend(tempfiles)
return foundfiles
def findfile2(searchdir, searchext):
os.chdir(searchdir)
print("Searchdir = ", os.getcwd(), "Searchext = ", searchext)
foundfiles = []
contents = os.listdir(".")
# print(contents)
for name in contents:
if os.path.isfile(name):
# print("found a file", name)
if name.endswith(searchext):
foundfiles.append(os.path.join(os.getcwd(), name))
elif os.path.isdir(name):
# print("found a directory", name)
tempfiles = findfile2(name, searchext)
foundfiles.extend(tempfiles)
os.chdir("..")
return foundfiles
# allfiles = findfile(SearchDir, SearchExt)
allfiles = findfile2(SearchDir, SearchExt)
for name in allfiles:
print(name)