def testSort(self):
dictFile = "sampledict.txt"
dictFilePath = os.path.join(testTempDir, dictFile)
with open(dictFilePath, "wb") as f:
f.write("%s\t%s\n" % ("abc", "the 3 characters before 'def'"))
f.write("%s\t%s\n" % ("def", "the 3 characters after 'abc'"))
try:
searcher = Searcher(testTempDir)
normalD = searcher.search("def")
self.assertTrue(dictFile in normalD)
self.assertTrue(normalD[dictFile].startswith("abc\t"))
d = searcher.search_raw("def",options=["--show-position"])
d = searcher.sort_result_by_column(d, remove_position_str=True)
sortedD = searcher.decode_result(d)
self.assertTrue(dictFile in sortedD)
self.assertTrue(sortedD[dictFile].startswith("def\t"))
finally:
os.remove(dictFilePath)
def testSort(self):
dictFile = "sampledict.txt"
dictFilePath = os.path.join(testTempDir, dictFile)
with open(dictFilePath, "wb") as f:
f.write("%s\t%s\n" % ("abc", "the 3 characters before 'def'"))
f.write("%s\t%s\n" % ("def", "the 3 characters after 'abc'"))
try:
searcher = Searcher(testTempDir)
normalD = searcher.search("def")
self.assertTrue(dictFile in normalD)
self.assertTrue(normalD[dictFile].startswith("abc\t"))
d = searcher.search_raw("def", options=["--show-position"])
d = searcher.sort_result_by_column(d, remove_position_str=True)
sortedD = searcher.decode_result(d)
self.assertTrue(dictFile in sortedD)
self.assertTrue(sortedD[dictFile].startswith("def\t"))
finally:
os.remove(dictFilePath)
def main():
if len(sys.argv) == 1:
sys.stdout.write("%s\n" % usage)
sys.exit(0)
searcher = Searcher(dataDir, ignoreFiles)
if not searcher.get_data_files():
sys.exit("""
No files in ./data directory.
(Installation is not completed. Put some utf-8 text files
in ./data directory.)
"""[1:-1])
queryOptionSet = set(["-i", "-w"] + ["-%d" % i for i in xrange(0, 10)])
optionSortByColumn = None
showPositionIsAppeared = False
optionHistory = True
opts, args = [], []
for a in sys.argv[1:]:
if a.startswith("-"):
if a in ("-h", "--help"):
sys.stdout.write("%s\n" % usage)
sys.exit(0)
elif a == "--version":
sys.stdout.write("locdic.find %s\nsee http://www.remics.org/locdic/ for more information.\n" % version)
sys.exit(0)
elif a == "--sort-by-column":
optionSortByColumn = True
elif a == "--show-position":
showPositionIsAppeared = True
opts.append(a)
elif a == "--no-history":
optionHistory = False
else:
opts.append(a)
else:
args.append(a)
if len(args) >= 2:
sys.exit("error: too many command-line arguments")
if optionSortByColumn and showPositionIsAppeared:
sys.exit("error: options are mutually exclusive: --sort-by-column, --show-position")
if optionHistory:
d = datetime.datetime.today()
add_to_history(args[0], [opt for opt in opts if opt in queryOptionSet],
d, "local")
if optionSortByColumn:
d = searcher.search_raw(args[0], options=opts + ["--show-position"])
d = searcher.sort_result_by_column(d, remove_position_str=True)
d = searcher.decode_result(d)
else:
d = searcher.search(args[0], options=opts)
output = sys.stdout
for f, r in sorted(d.items()):
output.write("%s:\n" % f)
output.write("%s\n" % r)
