数据导出

本节介绍用于导出点、边的方法。

export()

导出图中的点或边。

参数

• exportRequest: ExportRequest：导出请求的配置，包括dbType、schema、selectProperties和graph。
• cb: Callable[[List[Node], List[Edge]], None]：导出数据时执行的回调函数。
• config: RequestConfig（可选）：请求配置。

返回值

• 无

# Exports 'account' nodes in the graph 'miniCircle'

batch_counter = 0 # Global counter for batch tracking

exportRequest = ExportRequest(
    dbType=DBType.DBNODE,
    schema="account",
    selectProperties=["_id", "name", "year"],
    graph="miniCircle"
)

def handle_export(nodes: List[Node], edges: List[Edge]): # Defines the callback function
    global batch_counter
    batch_counter += 1

    try:
        schema = exportRequest.schema
        if nodes:
            df_nodes = pd.DataFrame([node.__dict__ for node in nodes]) # Converts to dictionary
            df_nodes.to_csv(f"{schema}_nodes.csv", mode="a", index=False)
            print(f"Batch {batch_counter} exported")
        if edges:
            df_edges = pd.DataFrame([edge.__dict__ for edge in edges]) # Converts to dictionary
            df_edges.to_csv(f"{schema}_edges.csv", mode="a", index=False)
            print(f"Batch {batch_counter} exported")
    except Exception as e:
        print(f"Batch {batch_counter} export failed: {e}")

Conn.export(exportRequest, handle_export)

Batch 1 exported
Batch 2 exported
Batch 3 exported

导出文件account_nodes.csv将保存到与执行文件相同的目录中。

完整示例

from typing import List
import pandas as pd
from ultipa import UltipaConfig, Connection, ExportRequest, DBType, Node, Edge

ultipaConfig = UltipaConfig()
# URI example: ultipaConfig.hosts = ["mqj4zouys.us-east-1.cloud.ultipa.com:60010"]
ultipaConfig.hosts = ["192.168.1.85:60061", "192.168.1.87:60061", "192.168.1.88:60061"]
ultipaConfig.username = "<username>"
ultipaConfig.password = "<password>"

Conn = Connection.NewConnection(defaultConfig=ultipaConfig)

# Exports 'account' nodes in the graph 'miniCircle'

batch_counter = 0 # Global counter for batch tracking

exportRequest = ExportRequest(
    dbType=DBType.DBNODE,
    schema="account",
    selectProperties=["_id", "name", "year"],
    graph="miniCircle"
)

def handle_export(nodes: List[Node], edges: List[Edge]): # Defines the callback function
    global batch_counter
    batch_counter += 1

    try:
        schema = exportRequest.schema
        if nodes:
            df_nodes = pd.DataFrame([node.__dict__ for node in nodes]) # Converts to dictionary
            df_nodes.to_csv(f"{schema}_nodes.csv", mode="a", index=False)
            print(f"Batch {batch_counter} exported")
        if edges:
            df_edges = pd.DataFrame([edge.__dict__ for edge in edges]) # Converts to dictionary
            df_edges.to_csv(f"{schema}_edges.csv", mode="a", index=False)
            print(f"Batch {batch_counter} exported")
    except Exception as e:
        print(f"Batch {batch_counter} export failed: {e}")

Conn.export(exportRequest, handle_export)