104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
|
||
|
|
# Python imports
|
||
|
|
import os, traceback, argparse, time, json, base64
|
||
|
|
from setproctitle import setproctitle
|
||
|
|
|
||
|
|
# Lib imports
|
||
|
|
|
||
|
|
# Application imports
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
_files_fifo_file = f"/tmp/search_files_fifo"
|
||
|
|
_grep_fifo_file = f"/tmp/grep_files_fifo"
|
||
|
|
|
||
|
|
filter = (".mkv", ".mp4", ".webm", ".avi", ".mov", ".m4v", ".mpg", ".mpeg", ".wmv", ".flv") + \
|
||
|
|
(".png", ".jpg", ".jpeg", ".gif", ".ico", ".tga", ".webp") + \
|
||
|
|
(".psf", ".mp3", ".ogg", ".flac", ".m4a")
|
||
|
|
|
||
|
|
file_result_set = []
|
||
|
|
|
||
|
|
|
||
|
|
def file_search(fifo, path, query):
|
||
|
|
try:
|
||
|
|
for file in os.listdir(path):
|
||
|
|
target = os.path.join(path, file)
|
||
|
|
if os.path.isdir(target):
|
||
|
|
file_search(fifo, target, query)
|
||
|
|
else:
|
||
|
|
if query.lower() in file.lower():
|
||
|
|
# file_result_set.append([target, file])
|
||
|
|
data = json.dumps([target, file])
|
||
|
|
fifo.write(data)
|
||
|
|
time.sleep(0.01)
|
||
|
|
except Exception as e:
|
||
|
|
print("Couldn't traverse to path. Might be permissions related...")
|
||
|
|
traceback.print_exc()
|
||
|
|
|
||
|
|
def _search_for_string(file, query):
|
||
|
|
try:
|
||
|
|
b64_file = base64.urlsafe_b64encode(file.encode('utf-8')).decode('utf-8')
|
||
|
|
grep_result_set = {}
|
||
|
|
|
||
|
|
with open(file, 'r') as fp:
|
||
|
|
for i, line in enumerate(fp):
|
||
|
|
if query in line:
|
||
|
|
b64_line = base64.urlsafe_b64encode(line.encode('utf-8')).decode('utf-8')
|
||
|
|
|
||
|
|
if f"{b64_file}" in grep_result_set.keys():
|
||
|
|
grep_result_set[f"{b64_file}"][f"{i+1}"] = b64_line
|
||
|
|
else:
|
||
|
|
grep_result_set[f"{b64_file}"] = {}
|
||
|
|
grep_result_set[f"{b64_file}"] = {f"{i+1}": b64_line}
|
||
|
|
|
||
|
|
# NOTE: Push to fifo here after loop
|
||
|
|
with open(_grep_fifo_file, 'w') as fifo:
|
||
|
|
data = json.dumps(grep_result_set)
|
||
|
|
fifo.write(data)
|
||
|
|
time.sleep(0.05)
|
||
|
|
except Exception as e:
|
||
|
|
print("Couldn't read file. Might be binary or other cause...")
|
||
|
|
traceback.print_exc()
|
||
|
|
|
||
|
|
|
||
|
|
def grep_search(path, query):
|
||
|
|
try:
|
||
|
|
for file in os.listdir(path):
|
||
|
|
target = os.path.join(path, file)
|
||
|
|
if os.path.isdir(target):
|
||
|
|
grep_search(target, query)
|
||
|
|
else:
|
||
|
|
if not target.lower().endswith(filter):
|
||
|
|
_search_for_string(target, query)
|
||
|
|
except Exception as e:
|
||
|
|
print("Couldn't traverse to path. Might be permissions related...")
|
||
|
|
traceback.print_exc()
|
||
|
|
|
||
|
|
def search(args):
|
||
|
|
if args.type == "file_search":
|
||
|
|
with open(_files_fifo_file, 'w') as fifo:
|
||
|
|
file_search(fifo, args.dir, args.query)
|
||
|
|
|
||
|
|
if args.type == "grep_search":
|
||
|
|
grep_search(args.dir, args.query)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
try:
|
||
|
|
setproctitle('SolarFM: File Search - Grepy')
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
# Add long and short arguments
|
||
|
|
parser.add_argument("--type", "-t", default=None, help="Type of search to do.")
|
||
|
|
parser.add_argument("--dir", "-d", default=None, help="Directory root for search type.")
|
||
|
|
parser.add_argument("--query", "-q", default=None, help="Query search is working against.")
|
||
|
|
|
||
|
|
# Read arguments (If any...)
|
||
|
|
args = parser.parse_args()
|
||
|
|
search(args)
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
traceback.print_exc()
|