FilesExpand file tree

 main

/

case.rs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (50 loc) · 1.63 KB

 main

/

case.rs

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (50 loc) · 1.63 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements. See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership. The ASF licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License. You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied. See the License for the

// specific language governing permissions and limitations

// under the License.

use crate::expr::PyExpr;

use datafusion::logical_expr::Case;

use pyo3::prelude::*;

#[pyclass(name = "Case", module = "datafusion.expr", subclass)]

#[derive(Clone)]

pub struct PyCase {

case: Case,

}

impl From<PyCase> for Case {

fn from(case: PyCase) -> Self {

case.case

}

}

impl From<Case> for PyCase {

fn from(case: Case) -> PyCase {

PyCase { case }

}

}

#[pymethods]

impl PyCase {

fn expr(&self) -> Option<PyExpr> {

self.case.expr.as_ref().map(|e| (**e).clone().into())

}

fn when_then_expr(&self) -> Vec<(PyExpr, PyExpr)> {

self.case

.when_then_expr

.iter()

.map(|e| ((*e.0).clone().into(), (*e.1).clone().into()))

.collect()

}

fn else_expr(&self) -> Option<PyExpr> {

self.case.else_expr.as_ref().map(|e| (**e).clone().into())

}

}