38 lines
765 B
Python
38 lines
765 B
Python
#!/usr/bin/env python3
|
||
"""
|
||
独立命令行导入脚本。
|
||
|
||
用法:
|
||
python run_import.py # 增量导入（MERGE，不删除现有数据）
|
||
python run_import.py --clear # 清空图谱后全量重新导入
|
||
"""
|
||
|
||
import sys
|
||
from dotenv import load_dotenv
|
||
from graph_query import get_driver
|
||
from graph_builder import build_graph
|
||
|
||
load_dotenv()
|
||
|
||
|
||
def main():
|
||
clear = "--clear" in sys.argv
|
||
|
||
print("Connecting to Neo4j...")
|
||
driver = get_driver()
|
||
driver.verify_connectivity()
|
||
print("Connected.\n")
|
||
|
||
build_graph(driver, clear=clear)
|
||
|
||
print("\nGraph stats:")
|
||
from graph_query import get_graph_stats
|
||
for k, v in get_graph_stats().items():
|
||
print(f" {k}: {v}")
|
||
|
||
driver.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|