python 实现简单的按关键字、后缀进行本地文件搜索
上代码:
def LocalFileMatching(source: Union[str, list, tuple, None] = None,
postfix: Union[str, list, tuple, None] = None,
keyword: Union[str, list, tuple] = '',
exact: Union[bool] = False) -> list | str:
occasional_dict = defaultdict(list)
postfix_empty_dict = defaultdict(bool)
postfix_empty_dict['EmptyState'] = False
try:
def item_append(_folder, _item):
if len(folder) < 4:
occasional_path = _folder + _item
else:
occasional_path = _folder + '/' + _item
if isabs(occasional_path):
occasional_dict['MatchingResult'].append(occasional_path)
else:
occasional_dict['MatchingResult'].append(realpath(occasional_path).replace('\\', '/'))
if source and isinstance(source, str):
matching_source_path = [source]
elif source and type(source) in [list, tuple]:
matching_source_path = source
else:
matching_source_path = None
if isinstance(postfix, str) or postfix is None:
if postfix is None:
postfix_empty_dict['EmptyState'] = True
else:
postfix = [postfix]
if postfix:
for item in postfix:
if item is None:
postfix_empty_dict['EmptyState'] = True
else:
occasional_dict['PostFix'].append(sub('\W', '', item))
keyword = '' if keyword is None else keyword
if matching_source_path is None:
for ascii_char in ascii_uppercase:
sys_disk = fr'{ascii_char}:/'
if exists(sys_disk):
for folder, cdir, obj in walk(sys_disk):
matching_for_postfix = comparition(keyword=keyword, compareobj=obj, exact=exact)
for matching_item in matching_for_postfix:
get_item_postfix = splitext(matching_item)[-1].lower()
if occasional_dict['PostFix'] and postfix_empty_dict['EmptyState']:
if get_item_postfix.lower()[1:] in occasional_dict['PostFix'] or not get_item_postfix:
item_append(folder, matching_item)
else:
continue
elif occasional_dict['PostFix'] and not postfix_empty_dict['EmptyState']:
if get_item_postfix.lower()[1:] in occasional_dict['PostFix']:
item_append(folder, matching_item)
else:
continue
else:
item_append(folder, matching_item)
else:
for sourceitem in matching_source_path:
if isfile(sourceitem):
occasional_dict['CombinationSource'].append(split(sourceitem)[0])
elif isdir(sourceitem):
occasional_dict['CombinationSource'].append(sourceitem)
else:
continue
for sourc_path in occasional_dict['CombinationSource']:
for folder, cdir, obj in walk(sourc_path):
folder = folder.replace('\\', '/')
matching_for_postfix = comparition(keyword=keyword, compareobj=obj, exact=exact)
if matching_for_postfix:
for matching_item in matching_for_postfix:
get_item_postfix = splitext(matching_item)[-1]
if occasional_dict['PostFix'] and postfix_empty_dict['EmptyState']:
if get_item_postfix.lower()[1:] in occasional_dict['PostFix'] or not get_item_postfix:
item_append(folder, matching_item)
else:
continue
elif occasional_dict['PostFix'] and not postfix_empty_dict['EmptyState']:
if get_item_postfix.lower()[1:] in occasional_dict['PostFix']:
item_append(folder, matching_item)
else:
continue
else:
item_append(folder, matching_item)
return occasional_dict['MatchingResult']
except BaseException as searchError:
return str(searchError)
def comparition(keyword: any,
compareobj: any,
exact: bool,
similarnumber: float = 0.8):
regular_result = []
if isinstance(compareobj, str):
compareobj = [compareobj]
if keyword:
for item in compareobj:
if isabs(item):
item = split(item)[-1].lower()
if exact:
if isinstance(keyword, list):
for key in keyword:
if key.lower() in item.lower() and item.strip():
regular_result.append(item)
else:
continue
else:
if keyword.lower() in item.lower() and item.strip():
regular_result.append(item)
else:
continue
else:
if isinstance(keyword, str):
keyword = [keyword]
for char in keyword:
equalitycount = 0
char = char.lower()
for ele in char:
if ele in item.lower():
equalitycount += 1
comoare = SequenceMatcher(a=item, b=char).quick_ratio()
if equalitycount == len(char) or comoare >= similarnumber:
if item in regular_result:
continue
else:
regular_result.append(item)
else:
continue
return regular_result
else:
return compareobj直接调用LocalFileMatching就行了， comparition只是辅助，在LocalFileMatching中已经调用了不管它
参数介绍:
source: Union[str, list, tuple, None] 源路径/文件，支持[str, list, tuple, None]类型
postfix: Union[str, list, tuple, None] 后缀，支持[str, list, tuple, None]类型
keyword: Union[str, list, tuple, None] 关键字，支持[str, list, tuple, None]类型
exact: Union[bool] = False 模式: False-模糊匹配或者True-精确匹配
使用:
伍佰粉和不良人粉在此
result1 = LocalFileMatching(source = ('D:\AppMusic', 'G:\BaiduNetdiskDownload', "G:\迅雷下载\画江湖之不良人第四季09.mp4"),
keyword = ['不良人', '伍佰'],
postfix = ['mp4', 'mp3', 'ape', 'flac', 'wav'],
exact = False)
for item in result1:
print(item)结果:
用时0.025179386138916016 秒
全盘查找：
用时 7 秒
精确查找：
为了效率全盘时建议还是加上关键字或者后缀筛选下吧