INNER CODE UNIT · Python

main

kangarooking/cangjie-skill · scripts/build_index.py:76

def main() -> int:
    ap = argparse.ArgumentParser(description=__doc__)
    ap.add_argument("chunks")
    ap.add_argument("--db", default=None)
    ap.add_argument("--query", default=None)
    ap.add_argument("--limit", type=int, default=5)
    ap.add_argument("--neighbors", type=int, default=1)
    args = ap.parse_args()

    chunks_path = Path(args.chunks)
    db_path = Path(args.db) if args.db else chunks_path.parent.parent / "index" / "lexical.sqlite"
    if args.query:
        return query(db_path, args.query, args.limit, args.neighbors)
    return build(chunks_path, db_path)


if __name__ == "__main__":
    raise SystemExit(main())

View source record →