46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
# search_file.py
|
|
# searching for target file in a given dir
|
|
# prompt user if multiple are found
|
|
|
|
import os
|
|
from fnmatch import fnmatch
|
|
|
|
def search_all_match(filename, path):
|
|
# search part shamelessly copied from stack overflow...
|
|
# real author: Nadia Alramli
|
|
results = []
|
|
for root, dirs, files in os.walk(path):
|
|
for name in files:
|
|
if fnmatch(filename, name):
|
|
results.append(os.path.join(root, name))
|
|
return results
|
|
|
|
def print_num_list(paths):
|
|
n = 1
|
|
for p in paths:
|
|
print(f"{n}. {p}")
|
|
n += 1
|
|
return 0
|
|
|
|
def select_path(paths):
|
|
l = len(paths)
|
|
if l == 0 :
|
|
raise Exception("Error select_path - empty list")
|
|
elif l == 1 :
|
|
return paths[0]
|
|
c = '0'
|
|
while not (c.isdigit() and 0 < int(c) <= l):
|
|
print("Multiple matches found.")
|
|
print("Please select target:")
|
|
print_num_list(paths)
|
|
c = input(f"Please select file to link to: (1-{l})")
|
|
return paths[int(c)-1]
|
|
|
|
def search_file(filename, path):
|
|
paths = search_all_match(filename, path)
|
|
if len(paths) == 0 :
|
|
print(f"No match for \"{filename}\". Link not modified.")
|
|
raise Exception(FileNotFoundError)
|
|
p = select_path(paths)
|
|
return p
|