You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
6 years ago
|
import os
|
||
|
|
import optparse
|
||
|
|
|
||
|
|
from lib import miscutils, fsutils
|
||
|
|
|
||
|
|
sources = {}
|
||
|
|
|
||
|
|
|
||
|
|
def optparse_init() -> tuple:
|
||
|
|
parser = optparse.OptionParser()
|
||
|
|
parser.add_option('-f', '--file', type='string', dest='s_file', help='Searching lines in the given file.')
|
||
|
|
parser.add_option('-d', '--directory', type='string', dest='s_dir', help='Searching files in a directorys.')
|
||
|
|
parser.add_option('-q', '--query', type='string', dest='query', help='The search term.')
|
||
|
|
parser.add_option('-l', '--loop', action='store_true', default=False, dest='loop', help='Looping over the input?')
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
def read_file_source(fname):
|
||
|
|
finfo = fsutils.FileInfo(fname)
|
||
|
|
if finfo.exist:
|
||
|
|
with finfo.open('r') as f:
|
||
|
|
return list(f.readlines())
|
||
|
|
|
||
|
|
|
||
|
|
def read_directory_source(dirname):
|
||
|
|
""" Reading the contents of the directory """
|
||
|
|
dinfo = fsutils.DirInfo(dirname)
|
||
|
|
if dinfo.exist:
|
||
|
|
return dinfo.get_full_content()
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
options, args = optparse_init()
|
||
|
|
engines = []
|
||
|
|
|
||
|
|
if options.s_file:
|
||
|
|
e_file = miscutils.SearchEngine(read_file_source(options.s_file))
|
||
|
|
engines.append(e_file)
|
||
|
|
|
||
|
|
if options.s_dir:
|
||
|
|
e_dir = miscutils.SearchEngine(read_directory_source(options.s_dir))
|
||
|
|
engines.append(e_dir)
|
||
|
|
|
||
|
|
while True:
|
||
|
|
if options.query and not options.loop:
|
||
|
|
query = options.query
|
||
|
|
else:
|
||
|
|
query = input('Search: ')
|
||
|
|
|
||
|
|
for engine in engines:
|
||
|
|
for res in engine.search(query):
|
||
|
|
print(res)
|
||
|
|
print('- - - - -')
|
||
|
|
if not options.loop:
|
||
|
|
break
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|