25 lines
616 B
Python
25 lines
616 B
Python
from fugashi_parser import *
|
|
|
|
# ---------- CLI ----------
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--vocab-file", required=True)
|
|
parser.add_argument("text", nargs="*")
|
|
args = parser.parse_args()
|
|
|
|
if args.text:
|
|
text = " ".join(args.text)
|
|
else:
|
|
text = sys.stdin.read()
|
|
|
|
vocab = load_vocab(args.vocab_file)
|
|
result = analyze(text, vocab)
|
|
save_vocab(args.vocab_file, vocab)
|
|
|
|
result["vocab_file"] = args.vocab_file
|
|
print(json.dumps(result, ensure_ascii=False, indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |