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()
|