datafusion_python/common/
df_schema.rs

1// Licensed to the Apache Software Foundation (ASF) under one
2// or more contributor license agreements.  See the NOTICE file
3// distributed with this work for additional information
4// regarding copyright ownership.  The ASF licenses this file
5// to you under the Apache License, Version 2.0 (the
6// "License"); you may not use this file except in compliance
7// with the License.  You may obtain a copy of the License at
8//
9//   http://www.apache.org/licenses/LICENSE-2.0
10//
11// Unless required by applicable law or agreed to in writing,
12// software distributed under the License is distributed on an
13// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14// KIND, either express or implied.  See the License for the
15// specific language governing permissions and limitations
16// under the License.
17
18use std::sync::Arc;
19
20use datafusion::common::DFSchema;
21use pyo3::prelude::*;
22
23#[derive(Debug, Clone)]
24#[pyclass(name = "DFSchema", module = "datafusion.common", subclass)]
25pub struct PyDFSchema {
26    schema: Arc<DFSchema>,
27}
28
29impl From<PyDFSchema> for DFSchema {
30    fn from(schema: PyDFSchema) -> DFSchema {
31        (*schema.schema).clone()
32    }
33}
34
35impl From<DFSchema> for PyDFSchema {
36    fn from(schema: DFSchema) -> PyDFSchema {
37        PyDFSchema {
38            schema: Arc::new(schema),
39        }
40    }
41}
42
43#[pymethods]
44impl PyDFSchema {
45    #[pyo3(name = "empty")]
46    #[staticmethod]
47    fn py_empty() -> PyResult<Self> {
48        Ok(Self {
49            schema: Arc::new(DFSchema::empty()),
50        })
51    }
52
53    #[pyo3(name = "field_names")]
54    fn py_field_names(&self) -> PyResult<Vec<String>> {
55        Ok(self.schema.field_names())
56    }
57}
datafusion_python/common/df_schema.rs

datafusion_python/common/
df_schema.rs
