Method load_edge_metadata

raphtory/src/python/graph/graph.rs:1094–1173

(
        &self,
        data: &Bound<PyAny>,
        src: &str,
        dst: &str,
        metadata: Option<Vec<PyBackedStr>>,
        shared_metadata: Option<HashMap<String, Prop>>,
        layer: O

        signature = (data, src, dst, metadata = None, shared_metadata = None, layer = None, layer_col = None, schema = None, csv_options = None)
    )]
    fn load_edge_metadata(
        &self,
        data: &Bound<PyAny>,
        src: &str,
        dst: &str,
        metadata: Option<Vec<PyBackedStr>>,
        shared_metadata: Option<HashMap<String, Prop>>,
        layer: Option<&str>,
        layer_col: Option<&str>,
        schema: Option<Bound<PyAny>>,
        csv_options: Option<CsvReadOptions>,
    ) -> Result<(), GraphError> {
        let metadata = convert_py_prop_args(metadata.as_deref()).unwrap_or_default();
        let column_schema = convert_py_schema(schema)?;
        if data.hasattr("__arrow_c_stream__")? {
            load_edge_metadata_from_arrow_c_stream(
                &self.graph,
                data,
                src,
                dst,
                &metadata,
                shared_metadata.as_ref(),
                layer,
                layer_col,
                column_schema,
            )
        } else if let Ok(path) = data.extract::<PathBuf>() {
            // extracting PathBuf handles Strings too
            let is_parquet = is_parquet_path(&path)?;
            let is_csv = is_csv_path(&path)?;

            // fail before loading anything at all to avoid loading partial data
            if !is_csv && csv_options.is_some() {
                return Err(GraphError::from(PyValueError::new_err(format!(
                    "CSV options were passed but no CSV files were detected at {}.",
                    path.display()
                ))));
            }

            // wrap in Arc to avoid cloning the entire schema for Parquet, CSV, and inner loops in CSV path
            let arced_schema = column_schema.map(Arc::new);

            // if-if instead of if-else to support directories with mixed parquet and CSV files
            if is_parquet {
                load_edge_metadata_from_parquet(
                    &self.graph,
                    path.as_path(),
                    src,
                    dst,
                    &metadata,
                    shared_metadata.as_ref(),
                    layer,
                    layer_col,
                    None,
                    arced_schema.clone(),
                    true,
                )?;
            }